test(git): add test for get_default_branch

bearomorphism · bearomorphism · commit 9627306099a8 · 2025-06-10T13:35:49.000+08:00
diff --git a/tests/test_git.py b/tests/test_git.py
@@ -480,3 +480,22 @@ def test_create_commit_cmd_string(mocker, os_name, committer_date, expected_cmd)
     mocker.patch("os.name", os_name)
     result = git._create_commit_cmd_string("", committer_date, "temp.txt")
     assert result == expected_cmd
+
+
+def test_get_default_branch(mocker: MockFixture):
+    # Test successful case
+    mocker.patch(
+        "commitizen.cmd.run", return_value=FakeCommand(out="refs/remotes/origin/main\n")
+    )
+    assert git.get_default_branch() == "refs/remotes/origin/main"
+
+    # Test error case
+    mocker.patch(
+        "commitizen.cmd.run",
+        return_value=FakeCommand(
+            err="fatal: ref refs/remotes/origin/HEAD is not a symbolic ref",
+            return_code=1,
+        ),
+    )
+    with pytest.raises(exceptions.GitCommandError):
+        git.get_default_branch()
