api_key_auth_cache_openai_fast_test.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  1. package service
  2. import (
  3. "context"
  4. "encoding/json"
  5. "testing"
  6. "github.com/stretchr/testify/require"
  7. )
  8. func TestAPIKeyAuthSnapshotGroupForceOpenAIFastRoundtrip(t *testing.T) {
  9. groupID := int64(50)
  10. apiKey := &APIKey{
  11. ID: 82, UserID: 40, GroupID: &groupID, Key: "sk-fast-roundtrip", Status: StatusActive,
  12. User: &User{ID: 40, Status: StatusActive},
  13. Group: &Group{
  14. ID: groupID, Name: "fast-roundtrip", Platform: PlatformOpenAI, Status: StatusActive,
  15. Hydrated: true, ForceOpenAIFast: true, FreeOpenAIFast: true,
  16. },
  17. }
  18. svc := &APIKeyService{}
  19. payload, err := json.Marshal(&APIKeyAuthCacheEntry{Snapshot: svc.snapshotFromAPIKey(context.Background(), apiKey)})
  20. require.NoError(t, err)
  21. var cached APIKeyAuthCacheEntry
  22. require.NoError(t, json.Unmarshal(payload, &cached))
  23. materialized, used, err := svc.applyAuthCacheEntry(apiKey.Key, &cached)
  24. require.NoError(t, err)
  25. require.True(t, used)
  26. require.NotNil(t, materialized.Group)
  27. require.True(t, materialized.Group.Hydrated)
  28. require.True(t, materialized.Group.ForceOpenAIFast)
  29. require.True(t, materialized.Group.FreeOpenAIFast)
  30. require.Equal(t, apiKeyAuthSnapshotVersion, cached.Snapshot.Version)
  31. }