Function TestAIBridgeAllowBYOK

enterprise/coderd/aibridge_test.go:2516–2609

(t *testing.T)

}

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

	cases := []struct {
		name           string
		allowBYOK      bool
		reqHeaders     map[string]string
		expectedStatus int
	}{
		{
			name:      "byok_enabled/centralized_request",
			allowBYOK: true,
			reqHeaders: map[string]string{
				"Authorization": "Bearer coder-token",
			},
			expectedStatus: http.StatusOK,
		},
		{
			name:      "byok_enabled/byok_request",
			allowBYOK: true,
			reqHeaders: map[string]string{
				agplaibridge.HeaderCoderToken: "coder-token",
				"Authorization":               "Bearer user-llm-key",
			},
			expectedStatus: http.StatusOK,
		},
		{
			name:      "byok_disabled/centralized_request",
			allowBYOK: false,
			reqHeaders: map[string]string{
				"Authorization": "Bearer coder-token",
			},
			expectedStatus: http.StatusOK,
		},
		{
			name:      "byok_disabled/byok_request",
			allowBYOK: false,
			reqHeaders: map[string]string{
				agplaibridge.HeaderCoderToken: "coder-token",
				"Authorization":               "Bearer user-llm-key",
			},
			expectedStatus: http.StatusForbidden,
		},
	}

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

			dv := coderdtest.DeploymentValues(t)
			dv.AI.BridgeConfig.Enabled = serpent.Bool(true)
			dv.AI.BridgeConfig.AllowBYOK = serpent.Bool(tc.allowBYOK)

			client, closer, api, _ := coderdenttest.NewWithAPI(t, &coderdenttest.Options{
				Options: &coderdtest.Options{
					DeploymentValues: dv,
				},
				LicenseOptions: &coderdenttest.LicenseOptions{