usage_query_cache_test.go 812 B

12345678910111213141516171819202122232425262728
  1. package admin
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/Wei-Shaw/sub2api/internal/pkg/usagestats"
  6. "github.com/stretchr/testify/require"
  7. )
  8. func TestUsageStatsCacheKey_StableAndDistinct(t *testing.T) {
  9. start := time.Date(2026, 5, 29, 0, 0, 0, 0, time.UTC)
  10. end := time.Date(2026, 5, 31, 0, 0, 0, 0, time.UTC)
  11. base := usagestats.UsageLogFilters{StartTime: &start, EndTime: &end, Model: "claude-3"}
  12. k1 := usageStatsCacheKey(base)
  13. k2 := usageStatsCacheKey(base)
  14. require.NotEmpty(t, k1)
  15. require.Equal(t, k1, k2, "same filters must produce same key")
  16. other := base
  17. other.Model = "gpt-4o"
  18. require.NotEqual(t, k1, usageStatsCacheKey(other), "different model must change key")
  19. withUser := base
  20. withUser.UserID = 7
  21. require.NotEqual(t, k1, usageStatsCacheKey(withUser), "different user must change key")
  22. }