Function TestFetchBranchDiff_Ratelimit

coderd/externalauth/gitprovider/github_test.go:615–647

(t *testing.T)

}

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

	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if strings.Contains(r.URL.Path, "/compare/") {
			// Second request: compare endpoint returns 429.
			w.Header().Set("Retry-After", "60")
			w.WriteHeader(http.StatusTooManyRequests)
			_, _ = w.Write([]byte(`{"message": "rate limit"}`))
			return
		}
		// First request: repo metadata.
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{"default_branch":"main"}`))
	}))
	defer srv.Close()

	gp, err := gitprovider.New("github", srv.URL+"/api/v3", srv.Client())
	require.NoError(t, err)
	require.NotNil(t, gp)

	_, err = gp.FetchBranchDiff(
		context.Background(),
		"test-token",
		gitprovider.BranchRef{Owner: "org", Repo: "repo", Branch: "feat"},
	)
	require.Error(t, err)

	var rlErr *gitprovider.RateLimitError
	require.True(t, errors.As(err, &rlErr), "error should be *RateLimitError, got: %T", err)
	expected := time.Now().Add(60 * time.Second)
	assert.WithinDuration(t, expected.Add(gitprovider.RateLimitPadding), rlErr.RetryAfter, 5*time.Second)
}

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