Function TestWorktreeCheckoutCommands

pkg/cmd/pr/shared/worktree_test.go:174–222

(t *testing.T)

}

func TestWorktreeCheckoutCommands(t *testing.T) {
	tests := []struct {
		name         string
		target       WorktreeTarget
		branchExists bool
		wantCommands [][]string
	}{
		{
			name:         "fresh target with existing branch",
			target:       WorktreeTarget{Path: "/path/to/wt"},
			branchExists: true,
			wantCommands: [][]string{{"worktree", "add", "--", "/path/to/wt", "feature"}},
		},
		{
			name:         "fresh target with new branch",
			target:       WorktreeTarget{Path: "/path/to/wt"},
			wantCommands: [][]string{{"worktree", "add", "--track", "-b", "feature", "--", "/path/to/wt", "origin/feature"}},
		},
		{
			name:         "reused target with existing branch",
			target:       WorktreeTarget{Path: "/path/to/wt", Reuse: true},
			branchExists: true,
			wantCommands: [][]string{{"-C", "/path/to/wt", "checkout", "feature"}},
		},
		{
			name:         "reused target with new branch",
			target:       WorktreeTarget{Path: "/path/to/wt", Reuse: true},
			wantCommands: [][]string{{"-C", "/path/to/wt", "checkout", "-b", "feature", "--track", "origin/feature"}},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			cs, teardown := run.Stub()
			defer teardown(t)
			exitCode := 1
			if tt.branchExists {
				exitCode = 0
			}
			cs.Register(`git rev-parse --verify refs/heads/feature`, exitCode, "")

			client := &git.Client{GitPath: "/some/path/git"}
			commands, branchExists := WorktreeCheckoutCommands(client, tt.target, "feature", "origin/feature")

			assert.Equal(t, tt.branchExists, branchExists)
			assert.Equal(t, tt.wantCommands, commands)
		})
	}
}