Function TestValidateLegacyAIBridgeConfig

cli/server_aibridge_internal_test.go:510–577

(t *testing.T)

}

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

	tests := []struct {
		name        string
		cfg         codersdk.AIBridgeConfig
		errContains string
	}{
		{
			name: "BareAnthropicKey",
			cfg: codersdk.AIBridgeConfig{
				LegacyAnthropic: codersdk.AIBridgeAnthropicConfig{Key: "sk-ant"},
			},
		},
		{
			name: "BareBedrockRegion",
			cfg: codersdk.AIBridgeConfig{
				LegacyBedrock: codersdk.AIBridgeBedrockConfig{Region: "us-east-1"},
			},
		},
		{
			name: "BedrockCredentialsOnly",
			cfg: codersdk.AIBridgeConfig{
				LegacyBedrock: codersdk.AIBridgeBedrockConfig{
					AccessKey:       "AKIA",
					AccessKeySecret: "secret",
				},
			},
		},
		{
			name: "AnthropicKeyAndBedrockConflict",
			cfg: codersdk.AIBridgeConfig{
				LegacyAnthropic: codersdk.AIBridgeAnthropicConfig{Key: "sk-ant"},
				LegacyBedrock: codersdk.AIBridgeBedrockConfig{
					Region:          "us-east-1",
					AccessKey:       "AKIA",
					AccessKeySecret: "secret",
				},
			},
			errContains: "CODER_AIBRIDGE_ANTHROPIC_KEY and CODER_AIBRIDGE_BEDROCK_* are mutually exclusive",
		},
		{
			name: "AnthropicKeyWithBedrockModelDefaultsIsFine",
			cfg: codersdk.AIBridgeConfig{
				LegacyAnthropic: codersdk.AIBridgeAnthropicConfig{Key: "sk-ant"},
				// Model defaults shouldn't trip the conflict; they're
				// always populated in a real deployment.
				LegacyBedrock: codersdk.AIBridgeBedrockConfig{
					Model:          "anthropic.claude-3-5-sonnet",
					SmallFastModel: "anthropic.claude-3-5-haiku",
				},
			},
		},
	}

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