zhaozewu
6c93f01c97
Remove unsupported none handling, align the auth cache version with the current baseline, consolidate the unshipped migrations, and update the public API contract.
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package service
|
|
|
|
import "testing"
|
|
|
|
func TestAPIKeyService_RejectsV10AuthSnapshotWithoutModelsListConfig(t *testing.T) {
|
|
groupID := int64(9)
|
|
svc := &APIKeyService{}
|
|
|
|
apiKey, ok, err := svc.applyAuthCacheEntry("k-legacy-models-list", &APIKeyAuthCacheEntry{
|
|
Snapshot: &APIKeyAuthSnapshot{
|
|
Version: 10,
|
|
APIKeyID: 1,
|
|
UserID: 2,
|
|
GroupID: &groupID,
|
|
Status: StatusActive,
|
|
User: APIKeyAuthUserSnapshot{
|
|
ID: 2,
|
|
Status: StatusActive,
|
|
Role: RoleUser,
|
|
Balance: 10,
|
|
Concurrency: 3,
|
|
},
|
|
Group: &APIKeyAuthGroupSnapshot{
|
|
ID: groupID,
|
|
Name: "openai",
|
|
Platform: PlatformOpenAI,
|
|
Status: StatusActive,
|
|
SubscriptionType: SubscriptionTypeStandard,
|
|
RateMultiplier: 1,
|
|
},
|
|
},
|
|
})
|
|
|
|
if err != nil {
|
|
t.Fatalf("expected stale snapshot to be ignored without error, got %v", err)
|
|
}
|
|
if ok {
|
|
t.Fatalf("expected v10 auth snapshot to be rejected after models_list_config was added")
|
|
}
|
|
if apiKey != nil {
|
|
t.Fatalf("expected no API key from stale snapshot, got %#v", apiKey)
|
|
}
|
|
}
|
|
|
|
func TestAPIKeyService_RejectsV15AuthSnapshotWithoutReasoningEffortPolicy(t *testing.T) {
|
|
svc := &APIKeyService{}
|
|
|
|
apiKey, ok, err := svc.applyAuthCacheEntry("k-legacy-reasoning-mappings", &APIKeyAuthCacheEntry{
|
|
Snapshot: &APIKeyAuthSnapshot{Version: 15},
|
|
})
|
|
|
|
if err != nil {
|
|
t.Fatalf("expected stale snapshot to be ignored without error, got %v", err)
|
|
}
|
|
if ok {
|
|
t.Fatal("expected v15 auth snapshot to be rejected after reasoning effort policy was added")
|
|
}
|
|
if apiKey != nil {
|
|
t.Fatalf("expected no API key from stale snapshot, got %#v", apiKey)
|
|
}
|
|
}
|