debiantestgitrepo: Allow to inject repo class name

This allows us to use the helpers for other repo types as well

1 files changed, 5 insertions, 2 deletions

diff --git a/tests/testutils/debiangittestrepo.py b/tests/testutils/debiangittestrepo.py
index 61456f1e..09bf8a70 100644
--- a/tests/testutils/debiangittestrepo.py
+++ b/tests/testutils/debiangittestrepo.py
@@ -10,11 +10,14 @@ import gbp.deb.git
 class DebianGitTestRepo(unittest.TestCase):
     """Scratch repo for a single unit test"""
 
-    def setUp(self):
+    def setUp(self, repo_cls=None):
         self.tmpdir = context.new_tmpdir(__name__)
 
+        if repo_cls is None:
+            repo_cls = gbp.deb.git.DebianGitRepository
+
         repodir = self.tmpdir.join('test_repo')
-        self.repo = gbp.deb.git.DebianGitRepository.create(repodir)
+        self.repo = repo_cls.create(repodir)
 
     def tearDown(self):
         context.teardown()