41 lines
2.3 KiB
Go
41 lines
2.3 KiB
Go
package model
|
|
|
|
import (
|
|
"slices"
|
|
"sync"
|
|
"testing"
|
|
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func TestGORMModelsMatchMigrationColumns(t *testing.T) {
|
|
tests := []struct {
|
|
model any
|
|
want []string
|
|
}{
|
|
{User{}, []string{"id", "username", "email", "password_hash", "display_name", "status", "created_at", "updated_at"}},
|
|
{Provider{}, []string{"id", "slug", "name", "base_url", "auth_type", "api_key_enc", "enabled", "created_at", "updated_at"}},
|
|
{ProviderModel{}, []string{"id", "provider_id", "name", "model_id", "api_type", "kind", "extra_body", "timeout_ms", "weight", "enabled", "created_at", "updated_at"}},
|
|
{PromptTemplate{}, []string{"id", "template_key", "kind", "api_type", "capability", "name", "version", "template_text", "default_role_rule", "enabled", "created_at", "updated_at"}},
|
|
{Generation{}, []string{"id", "user_id", "provider_model_id", "route_pool_id", "route_pool_version", "prompt_template_id", "route_snapshot", "role_rule", "provider_attempt_count", "kind", "status", "idempotency_key", "user_prompt", "rendered_prompt", "attempts", "attempt_count", "error_code", "error_message", "lease_owner", "lease_token", "lease_until", "available_at", "started_at", "completed_at", "created_at", "updated_at"}},
|
|
{GenerationInput{}, []string{"id", "generation_id", "position", "role", "note", "original_name", "mime_type", "storage_key", "size_bytes", "width", "height", "created_at"}},
|
|
{GenerationOutput{}, []string{"id", "generation_id", "kind", "text_content", "storage_key", "thumbnail_storage_key", "mime_type", "size_bytes", "width", "height", "created_at"}},
|
|
{APIKey{}, []string{"id", "user_id", "name", "public_id", "key_prefix", "secret_hash", "expires_at", "last_used_at", "revoked_at", "created_at", "updated_at"}},
|
|
{APIAuditEvent{}, []string{"id", "user_id", "api_key_id", "generation_id", "action", "result", "request_id", "status_code", "error_code", "summary", "created_at"}},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
parsed, err := schema.Parse(test.model, &sync.Map{}, schema.NamingStrategy{})
|
|
if err != nil {
|
|
t.Fatalf("schema.Parse(%T) error = %v", test.model, err)
|
|
}
|
|
got := append([]string(nil), parsed.DBNames...)
|
|
slices.Sort(got)
|
|
want := append([]string(nil), test.want...)
|
|
slices.Sort(want)
|
|
if !slices.Equal(got, want) {
|
|
t.Errorf("%T columns = %v, want %v", test.model, got, want)
|
|
}
|
|
}
|
|
}
|