fix coverage problem

Author: zhaochunqi

cmd/git_test.go

@@ -11,7 +11,7 @@ import (
 
 func Test_getCurrentGitDirectory(t *testing.T) {
 	// Use setupTestRepo for setup
-	_, cleanup := setupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
+	_, cleanup := SetupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
 	defer cleanup()
 
 	tests := []struct {
@@ -152,7 +152,7 @@ func Test_getRemoteURL(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			// If we have a custom test, run it instead of the standard test
 			if tt.customTest != nil {
-				_, cleanup := setupTestRepo(t, tt.remoteURL, "main")
+				_, cleanup := SetupTestRepo(t, tt.remoteURL, "main")
 				defer cleanup()
 
 				repo, err := getCurrentGitDirectory()
@@ -170,7 +170,7 @@ func Test_getRemoteURL(t *testing.T) {
 			}
 
 			// Standard test path
-			_, cleanup := setupTestRepo(t, tt.remoteURL, "main")
+			_, cleanup := SetupTestRepo(t, tt.remoteURL, "main")
 			defer cleanup()
 
 			repo, err := getCurrentGitDirectory()

cmd/root_test.go

@@ -11,10 +11,10 @@ import (
 
 func Test_rootCmd(t *testing.T) {
 	// Setup test repository
-	_, cleanup := setupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
+	_, cleanup := SetupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
 	defer cleanup()
 
-	// Save original openURLInBrowser function
+	// Save original openURLInBrowser function and restore after test
 	original := OpenURLInBrowser
 	defer func() {
 		OpenURLInBrowser = original
@@ -56,7 +56,7 @@ func Test_rootCmd(t *testing.T) {
 				t.Errorf("rootCmd.RunE() error = %v", err)
 				return
 			}
-			
+
 			if got := buf.String(); got != tt.wantOutput {
 				t.Errorf("root command output = %q, want %q", got, tt.wantOutput)
 			}
@@ -118,16 +118,16 @@ func Test_Execute(t *testing.T) {
 			// Setup logic moved directly into the test case
 			switch tt.name {
 			case "normal execution - github main branch":
-				_, cleanup := setupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
+				_, cleanup := SetupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "main")
 				t.Cleanup(cleanup)
 			case "normal execution - github feature branch":
-				_, cleanup := setupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "feature-branch")
+				_, cleanup := SetupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "feature-branch")
 				t.Cleanup(cleanup)
 			case "normal execution - gitlab feature branch":
-				_, cleanup := setupTestRepo(t, "https://gitlab.com/zhaochunqi/git-open.git", "feature-branch")
+				_, cleanup := SetupTestRepo(t, "https://gitlab.com/zhaochunqi/git-open.git", "feature-branch")
 				t.Cleanup(cleanup)
 			case "normal execution - bitbucket feature branch":
-				_, cleanup := setupTestRepo(t, "https://bitbucket.org/zhaochunqi/git-open.git", "feature-branch")
+				_, cleanup := SetupTestRepo(t, "https://bitbucket.org/zhaochunqi/git-open.git", "feature-branch")
 				t.Cleanup(cleanup)
 			case "no git repo":
 				// Create and change to temp dir without git repo

cmd/testing_helper.go

@@ -14,7 +14,7 @@ import (
 
 // setupTestRepo creates a temporary git repository for testing.
 // It returns the temporary directory path and a cleanup function.
-func setupTestRepo(t *testing.T, remoteURL string, branchName string) (string, func()) {
+func SetupTestRepo(t *testing.T, remoteURL string, branchName string) (string, func()) {
 	t.Helper()
 
 	// Create temporary directory

main_test.go

@@ -37,13 +37,17 @@ func Test_main(t *testing.T) {
 		{
 			name:      "default behavior",
 			args:      []string{"git-open"},
-			wantURL:   "https://github.com/zhaochunqi/git-open",
+			wantURL:   "https://github.com/zhaochunqi/git-open/tree/feat/open-branch",
 			wantError: false,
 		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
+			// Setup test repository for this test case
+			_, cleanup := cmd.SetupTestRepo(t, "https://github.com/zhaochunqi/git-open.git", "feat/open-branch")
+			t.Cleanup(cleanup)
+
 			os.Args = tt.args
 			openedURL = ""
 
@@ -73,4 +77,4 @@ func Test_main(t *testing.T) {
 			}
 		})
 	}
-}
+}