Function TestDo_AcceptedError_LargeBodyTruncated

github/github_test.go:2281–2305

TestDo_AcceptedError_LargeBodyTruncated verifies that when the API returns a 202 Accepted with a body larger than maxErrorBodySize, the client reads at most maxErrorBodySize bytes into AcceptedError.Raw and does not allocate unbounded memory.

(t *testing.T)

// most maxErrorBodySize bytes into AcceptedError.Raw and does not allocate
// unbounded memory.
func TestDo_AcceptedError_LargeBodyTruncated(t *testing.T) {
	t.Parallel()
	client, mux, _ := setup(t)

	// Serve a 202 response whose body exceeds the cap by one byte.
	mux.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) {
		w.WriteHeader(http.StatusAccepted)
		fmt.Fprint(w, strings.Repeat("x", maxErrorBodySize+1))
	})

	req, _ := client.NewRequest(t.Context(), "GET", ".", nil)
	_, err := client.Do(req, nil)
	if err == nil {
		t.Fatal("Expected AcceptedError, got nil")
	}

	var aerr *AcceptedError
	if !errors.As(err, &aerr) {
		t.Fatalf("Expected *AcceptedError, got %T: %v", err, err)
	}

	if got, want := len(aerr.Raw), maxErrorBodySize; got != want {
		t.Errorf("AcceptedError.Raw length = %v, want %v (maxErrorBodySize)", got, want)
	}
}

// Test that an error caused by the internal http client's Do() function
// does not leak the client secret.