Function TestRefresher_CorrectTokenPerOrigin

coderd/x/gitsync/gitsync_test.go:650–757

(t *testing.T)

}

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

	var tokenCalls atomic.Int32
	tokens := func(_ context.Context, _ uuid.UUID, origin string) (*string, error) {
		tokenCalls.Add(1)
		switch {
		case strings.Contains(origin, "github.com"):
			return ptr.Ref("gh-public-token"), nil
		case strings.Contains(origin, "ghes.corp.com"):
			return ptr.Ref("ghe-private-token"), nil
		default:
			return nil, fmt.Errorf("unexpected origin: %s", origin)
		}
	}

	// Track which token each FetchPullRequestStatus call received,
	// keyed by chat ID. We pass the chat ID through the PRRef.Number
	// field (unique per request) so FetchPullRequestStatus can
	// identify which row it's processing.
	var mu sync.Mutex
	tokensByPR := make(map[int]string)

	mp := &mockProvider{
		parsePullRequestURL: func(raw string) (gitprovider.PRRef, bool) {
			// Extract a unique PR number from the URL to identify
			// each row inside FetchPullRequestStatus.
			var num int
			switch {
			case strings.HasSuffix(raw, "/pull/1"):
				num = 1
			case strings.HasSuffix(raw, "/pull/2"):
				num = 2
			case strings.HasSuffix(raw, "/pull/10"):
				num = 10
			default:
				return gitprovider.PRRef{}, false
			}
			return gitprovider.PRRef{Owner: "org", Repo: "repo", Number: num}, true
		},
		fetchPullRequestStatus: func(_ context.Context, token string, ref gitprovider.PRRef) (*gitprovider.PRStatus, error) {
			mu.Lock()
			tokensByPR[ref.Number] = token
			mu.Unlock()
			return &gitprovider.PRStatus{State: gitprovider.PRStateOpen}, nil
		},
	}

	providers := func(_ context.Context, _ string) gitprovider.Provider { return mp }

	r := gitsync.NewRefresher(providers, tokens, slogtest.Make(t, nil), quartz.NewReal())

	ownerID := uuid.New()

	requests := []gitsync.RefreshRequest{
		{
			Row: database.ChatDiffStatus{
				ChatID:          uuid.New(),