Commit 3f3175db authored by Jim Fulton's avatar Jim Fulton

Added doctest setup helper functions.

parent 02a18650
...@@ -16,7 +16,11 @@ ...@@ -16,7 +16,11 @@
$Id$ $Id$
""" """
import os
import shutil
import tempfile
import time import time
import persistent import persistent
import transaction import transaction
from ZODB.MappingStorage import MappingStorage from ZODB.MappingStorage import MappingStorage
...@@ -38,3 +42,18 @@ class P(persistent.Persistent): ...@@ -38,3 +42,18 @@ class P(persistent.Persistent):
def __repr__(self): def __repr__(self):
return 'P(%s)' % self.name return 'P(%s)' % self.name
def setUp(test):
test.globs['__teardown_stack__'] = []
tmp = tempfile.mkdtemp('test')
registerTearDown(test, lambda : shutil.rmtree(tmp))
here = os.getcwd()
registerTearDown(test, lambda : os.chdir(here))
os.chdir(tmp)
def registerTearDown(test, func):
test.globs['__teardown_stack__'].append(func)
def tearDown(test):
for f in test.globs['__teardown_stack__']:
f()
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment