Function TestTryDetermineDefaultPRHead

pkg/cmd/pr/shared/find_refs_resolution_test.go:255–488

(t *testing.T)

}

func TestTryDetermineDefaultPRHead(t *testing.T) {
	t.Parallel()

	baseRepo := ghrepo.New("owner", "repo")
	baseRemote := ghContext.Remote{
		Remote: &git.Remote{
			Name: "upstream",
		},
		Repo: baseRepo,
	}

	forkRepo := ghrepo.New("otherowner", "repo-fork")
	forkRemote := ghContext.Remote{
		Remote: &git.Remote{
			Name: "origin",
		},
		Repo: forkRepo,
	}
	forkRepoURL, err := url.Parse("https://github.com/otherowner/repo-fork.git")
	require.NoError(t, err)

	t.Run("when the push revision is set, use that", func(t *testing.T) {
		t.Parallel()

		repoResolvedFromPushRevisionClient := stubGitConfigClient{
			pushRevisionFn: stubPushRevision(git.RemoteTrackingRef{
				Remote: "origin",
				Branch: "remote-feature-branch",
			}, nil),
		}

		defaultPRHead, err := TryDetermineDefaultPRHead(
			repoResolvedFromPushRevisionClient,
			stubRemoteToRepoResolver(ghContext.Remotes{&baseRemote, &forkRemote}, nil),
			"feature-branch",
		)
		require.NoError(t, err)

		require.True(t, ghrepo.IsSame(defaultPRHead.Repo.Unwrap(), forkRepo), "expected repos to be the same")
		require.Equal(t, "remote-feature-branch", defaultPRHead.BranchName)
	})

	t.Run("when the branch config push remote is set to a name, use that", func(t *testing.T) {
		t.Parallel()

		repoResolvedFromPushRemoteClient := stubGitConfigClient{
			pushRevisionFn: stubPushRevision(git.RemoteTrackingRef{}, errors.New("no push revision")),
			readBranchConfigFn: stubBranchConfig(git.BranchConfig{
				PushRemoteName: "origin",
			}, nil),
			pushDefaultFn: stubPushDefault(git.PushDefaultCurrent, nil),
		}

		defaultPRHead, err := TryDetermineDefaultPRHead(
			repoResolvedFromPushRemoteClient,
			stubRemoteToRepoResolver(ghContext.Remotes{&baseRemote, &forkRemote}, nil),
			"feature-branch",
		)