test(git-abort): add its unit test

Author: vanpipy

tests/conftest.py

@@ -0,0 +1,24 @@
+# Sharing fixtures
+# Ref: https://docs.pytest.org/en/6.2.x/fixture.html#scope-sharing-fixtures-across-classes-modules-packages-or-session
+
+import os
+import subprocess
+import shlex
+import tempfile
+import pytest
+import git
+
+git_extras_cwd = os.getcwd()
+
+@pytest.fixture(scope="module")
+def git_repo():
+    tmp_dir = tempfile.mkdtemp()
+    tmp_file_a = tempfile.mkstemp(dir=tmp_dir)
+    tmp_file_b = tempfile.mkstemp(dir=tmp_dir)
+    repo = git.Repo.init(tmp_dir)
+    os.chdir(tmp_dir)
+    repo.git.add(".")
+    repo.git.config("--local", "user.name", "test")
+    repo.git.config("--local", "user.email", "test@git-extras.com")
+    repo.git.commit("-m", "chore: initial commit")
+    return [repo, tmp_dir, tmp_file_a, tmp_file_b]

tests/helper.py

@@ -0,0 +1,7 @@
+import os
+import subprocess
+
+def invoke_git_extras_command(name):
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    git_extras_bin = os.path.join(current_dir, "..", "bin")
+    return subprocess.run(os.path.join(git_extras_bin, name), capture_output=True)

tests/test_git_abort.py

@@ -0,0 +1,72 @@
+import os
+import subprocess
+import shlex
+import git
+from helper import invoke_git_extras_command
+
+class TestGitAbort:
+    def test_init(self, git_repo):
+        repo, tmp_dir, tmp_a, tmp_b = git_repo
+        repo.git.branch("A")
+        repo.git.branch("B")
+        repo.git.checkout("A")
+        with open(tmp_a[1], "w", encoding="utf-8") as file_a_on_A:
+            file_a_on_A.write("a")
+            file_a_on_A.close()
+        repo.git.add(".")
+        repo.git.commit("-m", "A")
+        repo.git.checkout("B")
+        with open(tmp_a[1], "w", encoding="utf-8") as file_a_on_B:
+            file_a_on_B.write("b")
+            file_a_on_B.close()
+        repo.git.add(".")
+        repo.git.commit("-m", "B")
+        repo.git.status()
+
+    def test_cherry_pick(self, git_repo):
+        repo = git_repo[0]
+        try:
+            repo.git.cherry_pick("A")
+        except git.GitCommandError:
+            pass
+        result = repo.git.status()
+        assert "Unmerged path" in result
+        invoke_git_extras_command("git-abort")
+        result = repo.git.status()
+        assert "nothing to commit, working tree clean" in result
+
+    def test_merge(self, git_repo):
+        repo = git_repo[0]
+        try:
+            repo.git.merge("A")
+        except git.GitCommandError:
+            pass
+        result = repo.git.status()
+        assert "Unmerged path" in result
+        invoke_git_extras_command("git-abort")
+        result = repo.git.status()
+        assert "nothing to commit, working tree clean" in result
+
+    def test_rebase(self, git_repo):
+        repo = git_repo[0]
+        try:
+            repo.git.rebase("A")
+        except git.GitCommandError:
+            pass
+        result = repo.git.status()
+        assert "Unmerged path" in result
+        invoke_git_extras_command("git-abort")
+        result = repo.git.status()
+        assert "nothing to commit, working tree clean" in result
+
+    def test_revert(self, git_repo):
+        repo = git_repo[0]
+        try:
+            repo.git.revert("A")
+        except git.GitCommandError:
+            pass
+        result = repo.git.status()
+        assert "Unmerged path" in result
+        invoke_git_extras_command("git-abort")
+        result = repo.git.status()
+        assert "nothing to commit, working tree clean" in result