Function TestErrorHandling

aibridge/internal/integrationtest/bridge_internal_test.go:1421–1562

(t *testing.T)

}

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

	// Tests that errors which occur *before* a streaming response begins, or in non-streaming requests, are handled as expected.
	t.Run("non-stream error", func(t *testing.T) {
		t.Parallel()

		cases := []struct {
			name              string
			fixture           []byte
			path              string
			responseHandlerFn func(resp *http.Response)
		}{
			{
				name:    config.ProviderAnthropic,
				fixture: fixtures.AntNonStreamError,
				path:    pathAnthropicMessages,
				responseHandlerFn: func(resp *http.Response) {
					require.Equal(t, http.StatusBadRequest, resp.StatusCode)
					body, err := io.ReadAll(resp.Body)
					require.NoError(t, err)
					require.Equal(t, "error", gjson.GetBytes(body, "type").Str)
					require.Equal(t, "invalid_request_error", gjson.GetBytes(body, "error.type").Str)
					require.Contains(t, gjson.GetBytes(body, "error.message").Str, "prompt is too long")
				},
			},
			{
				name:    config.ProviderOpenAI,
				fixture: fixtures.OaiChatNonStreamError,
				path:    pathOpenAIChatCompletions,
				responseHandlerFn: func(resp *http.Response) {
					require.Equal(t, http.StatusBadRequest, resp.StatusCode)
					body, err := io.ReadAll(resp.Body)
					require.NoError(t, err)
					require.Equal(t, "context_length_exceeded", gjson.GetBytes(body, "error.code").Str)
					require.Equal(t, "invalid_request_error", gjson.GetBytes(body, "error.type").Str)
					require.Contains(t, gjson.GetBytes(body, "error.message").Str, "Input tokens exceed the configured limit")
				},
			},
		}

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

				for _, streaming := range []bool{true, false} {
					t.Run(fmt.Sprintf("streaming=%v", streaming), func(t *testing.T) {
						t.Parallel()

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

						// Setup mock server. Error fixtures contain raw HTTP
						// responses that may cause the bridge to retry.
						fix := fixtures.Parse(t, tc.fixture)
						upstream := newMockUpstream(ctx, t, newFixtureResponse(fix))

						bridgeServer := newBridgeTestServer(ctx, t, upstream.URL)