update tests

test/util.py

@@ -0,0 +1,64 @@
+import shlex
+import shutil
+import tempfile
+import pathlib2 as pl
+import subprocess as sp
+
+
+class Shell:
+    def __init__(self, cwd):
+        self.cwd = cwd
+
+    def __call__(self, cmd):
+        if isinstance(cmd, str):
+            cmd_parts = shlex.split(cmd)
+        else:
+            cmd_parts = cmd
+        output = sp.check_output(cmd_parts, cwd=self.cwd)
+        return output.decode("utf-8")
+
+
+_MODULE_PATH = pl.Path(__file__)
+FIXTURES_DIR = _MODULE_PATH.parent / "fixtures"
+
+
+class Project:
+    def __init__(self, project_prefix="a"):
+        if not project_prefix.endswith("_"):
+            project_prefix += "_"
+
+        tmpdir      = pl.Path(tempfile.mkdtemp(prefix="pytest_"))
+        self.tmpdir = tmpdir
+
+        _project_dir = tmpdir / "pycalver_project"
+        _project_dir.mkdir()
+
+        for fpath in FIXTURES_DIR.glob(project_prefix + "*"):
+            fname = fpath.name[len(project_prefix) :]
+            shutil.copy(fpath, _project_dir / fname)
+
+        self.dir = _project_dir
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *exc):
+        shutil.rmtree(str(self.tmpdir))
+        return False
+
+    def sh(self, cmd):
+        shell = Shell(str(self.dir))
+        return shell(cmd)
+
+    def git_init(self):
+        self.sh("git init")
+        self.sh("git add pycalver.toml")
+        self.sh("git add README.md")
+        self.sh("git commit -m 'initial commit'")
+
+    def hg_init(self):
+        self.sh = Shell(str(self.dir))
+        self.sh("hg init")
+        self.sh("hg add pycalver.toml")
+        self.sh("hg add README.md")
+        self.sh("hg commit -m 'initial commit'")