mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
Danny Kopping
a1330e3a8c
Adds `ai` to sqlc's `gen.go.initialisms` in `coderd/database/sqlc.yaml` so the generated DB code follows Go's initialism convention. Adds the matching `ai` -> `AI` case to the dbgen PascalCase helper (`scripts/dbgen/main.go`) so the corresponding `dbmem` / mock identifiers stay in sync. `make gen` regenerates the rest; hand-written call sites that consume DB-generated identifiers (`enterprise/audit/table.go`, `coderd/database/modelmethods.go`, `enterprise/coderd/aigatewaykeys.go`, `coderd/database/dbauthz/*`, etc.) are updated to match. Scope is deliberately limited to the database layer: - `coderd/rbac/*` (resource and scope generators) is untouched — `ResourceAi*` / `ScopeAi*` constants stay on main's casing. - `codersdk/*` (Go SDK) is untouched — `codersdk.ResourceAi*` / `codersdk.APIKeyScopeAi*` constants stay on main's casing, so external Go SDK consumers see no source-level break. - `Aibridge*` identifiers (one SQL token `aibridge`, not `ai_bridge`) are out of scope. On-the-wire values are unchanged: enum strings, RBAC resource type strings, API key scope strings, and JSON tags all stay the same. The HTTP/JSON surface is unaffected. Refs: [AIGOV-369](https://linear.app/codercom/issue/AIGOV-369/change-ai-references-in-coderddatabasemodelsgo-to-ai) 🤖 Generated with [Coder Agents](https://coder.com)
106 lines
3.3 KiB
Go
106 lines
3.3 KiB
Go
//go:build !slim
|
|
|
|
package cli
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/coder/coder/v2/coderd/aibridged"
|
|
"github.com/coder/coder/v2/coderd/database"
|
|
)
|
|
|
|
// TestClassifyProviderRow covers every branch of the classifier so the
|
|
// disabled, error, and enabled paths are exercised through the
|
|
// production code instead of relying on classifyRaw, the test mirror in
|
|
// reload_test.go.
|
|
func TestClassifyProviderRow(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
enabledRow := func(name, baseURL string) database.AIProvider {
|
|
return database.AIProvider{
|
|
Name: name,
|
|
Type: database.AIProviderTypeOpenai,
|
|
Enabled: true,
|
|
BaseUrl: baseURL,
|
|
}
|
|
}
|
|
|
|
t.Run("Enabled", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
got := classifyProviderRow(enabledRow("openai", "https://api.openai.com/v1"), seen)
|
|
assert.Equal(t, "openai", got.Name)
|
|
assert.Equal(t, string(database.AIProviderTypeOpenai), got.Type)
|
|
assert.Equal(t, aibridged.ProviderStatusEnabled, got.Status)
|
|
assert.Equal(t, "api.openai.com", got.Host)
|
|
assert.NoError(t, got.Err)
|
|
assert.Equal(t, "openai", seen["api.openai.com"])
|
|
})
|
|
|
|
t.Run("DisabledRow", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
row := enabledRow("off", "https://api.off.example.com/v1")
|
|
row.Enabled = false
|
|
got := classifyProviderRow(row, seen)
|
|
assert.Equal(t, aibridged.ProviderStatusDisabled, got.Status)
|
|
assert.Empty(t, got.Host, "disabled provider must not claim a host")
|
|
assert.NoError(t, got.Err)
|
|
assert.Empty(t, seen, "disabled provider must not occupy a host slot")
|
|
})
|
|
|
|
t.Run("EmptyBaseURL", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
got := classifyProviderRow(enabledRow("no-url", " "), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusError, got.Status)
|
|
assert.Empty(t, got.Host)
|
|
assert.ErrorContains(t, got.Err, "base url is empty")
|
|
})
|
|
|
|
t.Run("MalformedBaseURL", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
got := classifyProviderRow(enabledRow("bad", "://not-a-url"), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusError, got.Status)
|
|
assert.ErrorContains(t, got.Err, "invalid base url")
|
|
})
|
|
|
|
t.Run("BaseURLWithoutHostname", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
got := classifyProviderRow(enabledRow("no-host", "https://"), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusError, got.Status)
|
|
assert.ErrorContains(t, got.Err, "no hostname")
|
|
})
|
|
|
|
t.Run("DuplicateHostnameFirstWins", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
first := classifyProviderRow(enabledRow("first", "https://shared.example.com/v1"), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusEnabled, first.Status)
|
|
|
|
second := classifyProviderRow(enabledRow("second", "https://shared.example.com/v2"), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusError, second.Status)
|
|
assert.ErrorContains(t, second.Err, "already claimed by provider \"first\"")
|
|
assert.Equal(t, "first", seen["shared.example.com"], "first wins must not be overwritten")
|
|
})
|
|
|
|
t.Run("HostnameLowercased", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seen := map[string]string{}
|
|
got := classifyProviderRow(enabledRow("mixed", "https://API.Example.COM/v1"), seen)
|
|
assert.Equal(t, aibridged.ProviderStatusEnabled, got.Status)
|
|
assert.Equal(t, "api.example.com", got.Host)
|
|
})
|
|
}
|