api_key_auth_cache_codex_manifest_test.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package service
  2. import (
  3. "context"
  4. "encoding/json"
  5. "testing"
  6. "github.com/stretchr/testify/require"
  7. )
  8. // 固定账号 manifest 配置必须在认证快照 JSON 往返后完整保留,
  9. // 否则开关在缓存路径上会静默失效(投影对账由集成测试兜底,此处覆盖快照序列化)。
  10. func TestAPIKeyAuthSnapshotGroupCodexModelsManifestRoundtrip(t *testing.T) {
  11. groupID := int64(60)
  12. apiKey := &APIKey{
  13. ID: 92, UserID: 46, GroupID: &groupID, Key: "sk-codex-manifest-roundtrip", Status: StatusActive,
  14. User: &User{ID: 46, Status: StatusActive},
  15. Group: &Group{
  16. ID: groupID, Name: "codex-manifest-roundtrip", Platform: PlatformOpenAI, Status: StatusActive,
  17. Hydrated: true,
  18. CodexModelsManifestConfig: GroupCodexModelsManifestConfig{
  19. Enabled: true,
  20. AccountIDs: []int64{7, 8},
  21. FallbackToScheduler: true,
  22. },
  23. },
  24. }
  25. svc := &APIKeyService{}
  26. payload, err := json.Marshal(&APIKeyAuthCacheEntry{Snapshot: svc.snapshotFromAPIKey(context.Background(), apiKey)})
  27. require.NoError(t, err)
  28. var cached APIKeyAuthCacheEntry
  29. require.NoError(t, json.Unmarshal(payload, &cached))
  30. materialized, used, err := svc.applyAuthCacheEntry(apiKey.Key, &cached)
  31. require.NoError(t, err)
  32. require.True(t, used)
  33. require.NotNil(t, materialized.Group)
  34. require.True(t, materialized.Group.CodexModelsManifestConfig.Enabled)
  35. require.Equal(t, []int64{7, 8}, materialized.Group.CodexModelsManifestConfig.AccountIDs)
  36. require.True(t, materialized.Group.CodexModelsManifestConfig.FallbackToScheduler)
  37. require.Equal(t, apiKeyAuthSnapshotVersion, cached.Snapshot.Version)
  38. }