Function TestUpdateChatProvider

coderd/exp_chats_test.go:2632–2923

(t *testing.T)

}

func TestUpdateChatProvider(t *testing.T) {
	t.Parallel()
	t.Skip("legacy chat provider API removed in favor of AI provider API")

	t.Run("Success", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitLong)
		client := newChatClient(t)
		_ = coderdtest.CreateFirstUser(t, client.Client)

		provider, err := client.CreateChatProvider(ctx, codersdk.CreateChatProviderConfigRequest{
			Provider: "openai",
			APIKey:   "test-api-key",
		})
		require.NoError(t, err)

		enabled := false
		baseURL := "https://example.com/v1"
		updated, err := client.UpdateChatProvider(ctx, provider.ID, codersdk.UpdateChatProviderConfigRequest{
			DisplayName: "OpenAI Updated",
			Enabled:     &enabled,
			BaseURL:     &baseURL,
		})
		require.NoError(t, err)
		require.Equal(t, provider.ID, updated.ID)
		require.Equal(t, "OpenAI Updated", updated.DisplayName)
		require.False(t, updated.Enabled)
		require.Equal(t, baseURL, updated.BaseURL)
	})

	t.Run("AllowsClearingBedrockAPIKeyWithCentralAPIKeyEnabled", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitLong)
		client := newChatClient(t)
		_ = coderdtest.CreateFirstUser(t, client.Client)

		provider, err := client.CreateChatProvider(ctx, codersdk.CreateChatProviderConfigRequest{
			Provider:             "bedrock",
			DisplayName:          "AWS Bedrock",
			APIKey:               "bedrock-bearer-token",
			CentralAPIKeyEnabled: ptr.Ref(true),
		})
		require.NoError(t, err)
		require.True(t, provider.HasAPIKey)
		require.True(t, provider.CentralAPIKeyEnabled)

		updated, err := client.UpdateChatProvider(ctx, provider.ID, codersdk.UpdateChatProviderConfigRequest{
			APIKey:               ptr.Ref(""),
			CentralAPIKeyEnabled: ptr.Ref(true),
		})
		require.NoError(t, err)
		require.Equal(t, provider.ID, updated.ID)
		require.Equal(t, "bedrock", updated.Provider)
		require.False(t, updated.HasAPIKey)
		require.True(t, updated.CentralAPIKeyEnabled)
	})