tests/testutils/repo module: New module for Source support scaffolding

Currently comes with a Repo() abstract class and a git implementation

1 files changed, 42 insertions, 0 deletions

diff --git a/tests/testutils/repo/bzr.py b/tests/testutils/repo/bzr.py
new file mode 100644
index 000000000..848cff85f
--- /dev/null
+++ b/tests/testutils/repo/bzr.py
@@ -0,0 +1,42 @@
+import os
+import shutil
+import subprocess
+from .repo import Repo
+
+BZR_ENV = {
+    "BZR_EMAIL": "Testy McTesterson <testy.mctesterson@example.com>"
+}
+
+
+class Bzr(Repo):
+
+    def create(self, directory):
+        branch_dir = os.path.join(self.repo, 'trunk')
+
+        subprocess.call(['bzr', 'init-repo', self.repo], env=BZR_ENV)
+        subprocess.call(['bzr', 'init', branch_dir], env=BZR_ENV)
+        self.copy_directory(directory, branch_dir)
+        subprocess.call(['bzr', 'add', '.'], env=BZR_ENV, cwd=branch_dir)
+        subprocess.call(['bzr', 'commit', '--message="Initial commit"'],
+                        env=BZR_ENV, cwd=branch_dir)
+
+        return self.latest_commit()
+
+    def source_config(self, ref=None):
+        config = {
+            'kind': 'bzr',
+            'url': 'file://' + self.repo,
+            'track': 'trunk'
+        }
+        if ref is not None:
+            config['ref'] = ref
+
+        return config
+
+    def latest_commit(self):
+        output = subprocess.check_output([
+            'bzr', 'version-info',
+            '--custom', '--template={revno}',
+            os.path.join(self.repo, 'trunk')
+        ], env=BZR_ENV)
+        return output.decode('UTF-8').strip()