Fix new repo-smith unit tests

Author: woojiahao

tests/unit/helpers/git_helper/test_git_helper.py

@@ -1,5 +1,5 @@
 from subprocess import CompletedProcess
-from unittest.mock import patch
+from unittest.mock import patch, MagicMock
 
 import pytest
 from repo_smith.command_result import CommandResult
@@ -8,32 +8,35 @@
 
 
 def test_tag_with_annotate():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
         return_value=CommandResult(
             CompletedProcess(["git", "tag", "v0.1.0", "-a"], returncode=0)
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.tag("v0.1.0", annotate=True)
         mock_helper.assert_called_with(["git", "tag", "v0.1.0", "-a"])
 
 
 def test_tag_with_force():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
         return_value=CommandResult(
             CompletedProcess(["git", "tag", "v0.1.0", "-f"], returncode=0)
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.tag("v0.1.0", force=True)
         mock_helper.assert_called_with(["git", "tag", "v0.1.0", "-f"])
 
 
 def test_tag_with_message():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -43,14 +46,15 @@ def test_tag_with_message():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.tag("v0.1.0", message="This is a message.")
         mock_helper.assert_called_with(
             ["git", "tag", "v0.1.0", "-m", "This is a message."]
         )
 
 
 def test_tag_with_multiple_flags():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -61,30 +65,29 @@ def test_tag_with_multiple_flags():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.tag("v0.1.0", message="This is a message.", force=True, annotate=True)
         mock_helper.assert_called_with(
             ["git", "tag", "v0.1.0", "-f", "-m", "This is a message.", "-a"]
         )
 
 
 def test_add_all():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
         return_value=CommandResult(
-            CompletedProcess(
-                ["git", "add", "-a"],
-                returncode=0,
-            )
+            CompletedProcess(["git", "add", "-A"], returncode=0),
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.add(all=True)
-        mock_helper.assert_called_with(["git", "add", "-a"])
+        mock_helper.assert_called_with(["git", "add", "-A"])
 
 
 def test_commit_all():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -95,12 +98,13 @@ def test_commit_all():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.commit(all=True, message="Test commit")
         mock_helper.assert_called_with(["git", "commit", "-a", "-m", "Test commit"])
 
 
 def test_commit_with_pathspec():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -111,14 +115,15 @@ def test_commit_with_pathspec():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.commit(pathspec="file*", all=True, message="Test commit")
         mock_helper.assert_called_with(
             ["git", "commit", "-a", "-m", "Test commit", "file*"]
         )
 
 
 def test_checkout_with_branch_and_start_point():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -129,7 +134,7 @@ def test_checkout_with_branch_and_start_point():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.checkout(
             branch_name="test",
             start_point="HEAD",
@@ -139,6 +144,7 @@ def test_checkout_with_branch_and_start_point():
 
 
 def test_checkout_with_files():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -149,7 +155,7 @@ def test_checkout_with_files():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.checkout(
             branch_name="test",
             paths=["filea.txt", "fileb.txt"],
@@ -160,6 +166,7 @@ def test_checkout_with_files():
 
 
 def test_restore_staged_worktree():
+    repo = MagicMock()
     with patch.object(
         Helper,
         "run",
@@ -170,12 +177,13 @@ def test_restore_staged_worktree():
             )
         ),
     ) as mock_helper:
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.restore(pathspec=".", staged=True, worktree=True)
         mock_helper.assert_called_with(["git", "restore", "-W", "-S", "."])
 
 
 def test_restore_ours():
+    repo = MagicMock()
     with (
         patch.object(
             Helper,
@@ -189,5 +197,5 @@ def test_restore_ours():
         ),
         pytest.raises(ValueError),
     ):
-        gh = GitHelper(False)
+        gh = GitHelper(repo, False)
         gh.restore(pathspec=".", ours=True, source=".", staged=True, worktree=True)