diff options
author | Guido Günther <agx@sigxcpu.org> | 2014-04-18 15:05:13 +0200 |
---|---|---|
committer | Guido Günther <agx@sigxcpu.org> | 2014-04-18 15:05:13 +0200 |
commit | 9a952f68c6dc5916c2f8d4fb686e7f216fa6736f (patch) | |
tree | 7e76b381ec327b97cf9b2ef562eab383da8f1432 /tests/context.py | |
parent | 5946d6d1079969822b750af12bb5c95b188485ae (diff) | |
parent | 98bdda0e2e0fe182857ebb7f56b86193c2a9e7d3 (diff) |
Merge tag 'v0.0.5' into debian/master
whatmaps 0.0.5
Diffstat (limited to 'tests/context.py')
-rw-r--r-- | tests/context.py | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/tests/context.py b/tests/context.py new file mode 100644 index 0000000..d859c2f --- /dev/null +++ b/tests/context.py @@ -0,0 +1,53 @@ +# this context.py should be included by all tests +# idea from http://kennethreitz.com/repository-structure-and-python.html + +import os +import shutil +import sys +import tempfile + +import whatmaps + +sys.path.insert(0, os.path.abspath('..')) + +# the top or root dir of the git-buildpackage source tree to be used by tests +projectdir = os.path.dirname(os.path.dirname(os.path.abspath(whatmaps.__file__))) + +_chdir_backup = None +_tmpdirs = [] + +def chdir(dir): + global _chdir_backup + if not _chdir_backup: + _chdir_backup = os.path.abspath(os.curdir) + os.chdir(str(dir)) + +def new_tmpdir(name): + global _tmpdirs + prefix='whatmaps_%s_' % name + tmpdir = TmpDir(prefix) + _tmpdirs.append(tmpdir) + return tmpdir + +def teardown(): + if _chdir_backup: + os.chdir(_chdir_backup) + for tmpdir in _tmpdirs: + tmpdir.rmdir() + del _tmpdirs[:] + +class TmpDir(object): + + def __init__(self, suffix='', prefix='tmp'): + self.path = tempfile.mkdtemp(suffix=suffix, prefix=prefix) + + def rmdir(self): + if self.path and not os.getenv("WHATMAPS_TESTS_NOCLEAN"): + shutil.rmtree(self.path) + self.path = None + + def __repr__(self): + return self.path + + def join(self, *args): + return os.path.join(self.path, *args) |