From 936ad4fdd233f2be8e3b92ad2ee060e027fc6eac Mon Sep 17 00:00:00 2001 From: Jon Bernard Date: Thu, 17 Jul 2014 17:37:30 -0400 Subject: Experiments with using pytest for tests --- tests/utils.py | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 tests/utils.py (limited to 'tests/utils.py') diff --git a/tests/utils.py b/tests/utils.py new file mode 100644 index 0000000..ecdeb84 --- /dev/null +++ b/tests/utils.py @@ -0,0 +1,55 @@ +import os +from dotfiles.utils import compare_path as samefile + + +def _touch(fname, times=None): + with open(fname, 'a'): + os.utime(fname, times) + + +class HomeDirectory(object): + + + def __init__(self, path, repo, contents): + self.path = path + self.repo = repo + if contents: + self.setup(contents) + + + def setup(self, contents): + repo = os.path.join(self.path, self.repo) + os.mkdir(repo) + + for link, link_should_exist in contents.items(): + + target = os.path.join('%s/%s' % (repo, link[1:])) + _touch(target) + + if link_should_exist: + os.symlink(target, os.path.join(self.path, link)) + + self.verify(contents) + + + def verify(self, contents): + __tracebackhide__ = True + + for link, link_should_exist in contents.items(): + + target = os.path.join(self.path, '%s/%s' % (self.repo, link[1:])) + + if not os.path.exists(target): + pytest.fail("missing expected repo file \"%s\"" % target) + + link = os.path.join(self.path, link) + link_exists = os.path.exists(link) + + if link_should_exist: + if not link_exists: + pytest.fail("missing expected symlink \"%s\"" % link) + if not samefile(link, target): + pytest.fail("\"%s\" does not link to \"%s\"" % (link, target)) + + elif link_exists: + pytest.fail("found unexpected symlink \"%s\"" % link) -- cgit v1.2.3