diff --git a/tests/__init__.py b/tests/__init__.py index 1e8d09c..8247abc 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -74,6 +74,7 @@ class Modeltests(unittest.TestCase): self.session = None self.path = tempfile.mkdtemp(prefix='progit-tests') self.gitrepo = None + self.gitrepos = None # pylint: disable=C0103 def setUp(self): @@ -134,6 +135,11 @@ class Modeltests(unittest.TestCase): if self.gitrepo and os.path.exists(self.gitrepo): shutil.rmtree(self.gitrepo) + if self.gitrepos: + for repo in self.gitrepo: + if os.path.exists(repo): + shutil.rmtree(repo) + class FakeGroup(object): """ Fake object used to make the FakeUser object closer to the