cache: add tier definitions and EngineTier function
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
parent
baf98ca80e
commit
ff4149ecbd
2 changed files with 89 additions and 0 deletions
33
internal/cache/tiers_test.go
vendored
Normal file
33
internal/cache/tiers_test.go
vendored
Normal file
|
|
@ -0,0 +1,33 @@
|
|||
package cache
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestEngineTier(t *testing.T) {
|
||||
// Test default static tier
|
||||
tier := EngineTier("wikipedia", nil)
|
||||
if tier.Name != "static" || tier.Duration != 24*time.Hour {
|
||||
t.Errorf("wikipedia: expected static/24h, got %s/%v", tier.Name, tier.Duration)
|
||||
}
|
||||
|
||||
// Test default api_general tier
|
||||
tier = EngineTier("braveapi", nil)
|
||||
if tier.Name != "api_general" || tier.Duration != 1*time.Hour {
|
||||
t.Errorf("braveapi: expected api_general/1h, got %s/%v", tier.Name, tier.Duration)
|
||||
}
|
||||
|
||||
// Test override takes precedence — override tier name is just the engine name
|
||||
override := 48 * time.Hour
|
||||
tier = EngineTier("wikipedia", map[string]time.Duration{"wikipedia": override})
|
||||
if tier.Name != "wikipedia" || tier.Duration != 48*time.Hour {
|
||||
t.Errorf("wikipedia override: expected wikipedia/48h, got %s/%v", tier.Name, tier.Duration)
|
||||
}
|
||||
|
||||
// Test unknown engine gets default
|
||||
tier = EngineTier("unknown_engine", nil)
|
||||
if tier.Name != "unknown" || tier.Duration != 1*time.Hour {
|
||||
t.Errorf("unknown engine: expected unknown/1h, got %s/%v", tier.Name, tier.Duration)
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue