Function TestUpstreamError

internal/integrationtest/responses_test.go:666–739

TODO set MaxRetries to speed up this test option.WithMaxRetries(0), in base responses interceptor https://github.com/coder/aibridge/issues/115

(t *testing.T)

// option.WithMaxRetries(0), in base responses interceptor
// https://github.com/coder/aibridge/issues/115
func TestUpstreamError(t *testing.T) {
	t.Parallel()

	responsesError := `{"error":{"message":"Something went wrong","type":"invalid_request_error","param":null,"code":"invalid_request"}}`
	nonResponsesError := `plain text error`

	tests := []struct {
		name        string
		streaming   bool
		statusCode  int
		contentType string
		body        string
	}{
		{
			name:        "blocking_responses_error",
			streaming:   false,
			statusCode:  http.StatusBadRequest,
			contentType: "application/json",
			body:        responsesError,
		},
		{
			name:        "streaming_responses_error",
			streaming:   true,
			statusCode:  http.StatusBadRequest,
			contentType: "application/json",
			body:        responsesError,
		},
		{
			name:        "blocking_non_responses_error",
			streaming:   false,
			statusCode:  http.StatusBadGateway,
			contentType: "text/plain",
			body:        nonResponsesError,
		},
		{
			name:        "streaming_non_responses_error",
			streaming:   true,
			statusCode:  http.StatusBadGateway,
			contentType: "text/plain",
			body:        nonResponsesError,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			ctx, cancel := context.WithTimeout(t.Context(), testutil.WaitLong)
			t.Cleanup(cancel)

			upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.Header().Set("Content-Type", tc.contentType)
				w.WriteHeader(tc.statusCode)
				_, err := w.Write([]byte(tc.body))
				require.NoError(t, err)
			}))
			t.Cleanup(upstream.Close)