Function TestFallthrough

aibridge/internal/integrationtest/bridge_internal_test.go:1086–1163

(t *testing.T)

}

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

	testCases := []struct {
		name                 string
		fixture              []byte
		basePath             string
		requestPath          string
		expectedUpstreamPath string
		expectAuthHeader     string
	}{
		{
			name:                 "ant_empty_base_url_path",
			fixture:              fixtures.AntFallthrough,
			basePath:             "",
			requestPath:          "/anthropic/v1/models",
			expectedUpstreamPath: "/v1/models",
			expectAuthHeader:     "X-Api-Key",
		},
		{
			name:                 "oai_empty_base_url_path",
			fixture:              fixtures.OaiChatFallthrough,
			basePath:             "",
			requestPath:          "/openai/v1/models",
			expectedUpstreamPath: "/models",
			expectAuthHeader:     "Authorization",
		},
		{
			name:                 "ant_some_base_url_path",
			fixture:              fixtures.AntFallthrough,
			basePath:             "/api",
			requestPath:          "/anthropic/v1/models",
			expectedUpstreamPath: "/api/v1/models",
			expectAuthHeader:     "X-Api-Key",
		},
		{
			name:                 "oai_some_base_url_path",
			fixture:              fixtures.OaiChatFallthrough,
			basePath:             "/api",
			requestPath:          "/openai/v1/models",
			expectedUpstreamPath: "/api/models",
			expectAuthHeader:     "Authorization",
		},
	}

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

			fix := fixtures.Parse(t, tc.fixture)
			upstream := newMockUpstream(t.Context(), t, newFixtureResponse(fix))
			bridgeServer := newBridgeTestServer(t.Context(), t, upstream.URL+tc.basePath)

			resp, err := bridgeServer.makeRequest(t, http.MethodGet, tc.requestPath, nil)
			require.NoError(t, err)
			defer resp.Body.Close()

			require.Equal(t, http.StatusOK, resp.StatusCode)