| package executor |
|
|
| import ( |
| "testing" |
| "time" |
|
|
| miniredis "github.com/alicebob/miniredis/v2" |
| "github.com/router-for-me/CLIProxyAPI/v6/internal/config" |
| ) |
|
|
| func TestGetOrCreateCodexPromptCacheRedis_StableKey(t *testing.T) { |
| resetCodexPromptCacheRedisForTest() |
| defer resetCodexPromptCacheRedisForTest() |
|
|
| mr, err := miniredis.Run() |
| if err != nil { |
| t.Fatalf("start miniredis: %v", err) |
| } |
| defer mr.Close() |
|
|
| cfg := &config.Config{} |
| cfg.CodexPromptCache.RedisURL = "redis://" + mr.Addr() + "/0" |
| cfg.CodexPromptCache.KeyPrefix = "cliproxy:test:" |
| cfg.CodexPromptCache.TTLSeconds = 3600 |
| cfg.CodexPromptCache.TimeoutMS = 500 |
|
|
| id1, ok1 := getOrCreateCodexPromptCacheRedis(cfg, "gpt-5-user-1", 0) |
| if !ok1 || id1 == "" { |
| t.Fatalf("first redis cache id not created: ok=%v id=%q", ok1, id1) |
| } |
| id2, ok2 := getOrCreateCodexPromptCacheRedis(cfg, "gpt-5-user-1", 0) |
| if !ok2 || id2 == "" { |
| t.Fatalf("second redis cache id not returned: ok=%v id=%q", ok2, id2) |
| } |
| if id1 != id2 { |
| t.Fatalf("redis cache id mismatch: %q vs %q", id1, id2) |
| } |
| } |
|
|
| func TestGetOrCreateCodexPromptCacheRedis_FallbackOnInvalidURL(t *testing.T) { |
| resetCodexPromptCacheRedisForTest() |
| defer resetCodexPromptCacheRedisForTest() |
|
|
| cfg := &config.Config{} |
| cfg.CodexPromptCache.RedisURL = "::://invalid" |
| cfg.CodexPromptCache.TimeoutMS = 100 |
|
|
| id, ok := getOrCreateCodexPromptCacheRedis(cfg, "gpt-5-user-1", 0) |
| if ok { |
| t.Fatalf("expected redis disabled/fallback on invalid URL, got ok=true id=%q", id) |
| } |
| } |
|
|
| func TestGetOrCreateCodexPromptCacheRedis_UsesConfiguredTTL(t *testing.T) { |
| resetCodexPromptCacheRedisForTest() |
| defer resetCodexPromptCacheRedisForTest() |
|
|
| mr, err := miniredis.Run() |
| if err != nil { |
| t.Fatalf("start miniredis: %v", err) |
| } |
| defer mr.Close() |
|
|
| cfg := &config.Config{} |
| cfg.CodexPromptCache.RedisURL = "redis://" + mr.Addr() + "/0" |
| cfg.CodexPromptCache.KeyPrefix = "cliproxy:test:" |
| cfg.CodexPromptCache.TTLSeconds = 7 |
| cfg.CodexPromptCache.TimeoutMS = 500 |
|
|
| _, ok := getOrCreateCodexPromptCacheRedis(cfg, "gpt-5-user-ttl", 0) |
| if !ok { |
| t.Fatalf("expected redis set to succeed") |
| } |
|
|
| ttl := mr.TTL("cliproxy:test:gpt-5-user-ttl") |
| if ttl <= 0 { |
| t.Fatalf("expected positive ttl, got %v", ttl) |
| } |
| if ttl > 8*time.Second || ttl < 6*time.Second { |
| t.Fatalf("expected ttl around 7s, got %v", ttl) |
| } |
| } |
|
|
| func TestGetOrCreateCodexPromptCacheRedis_RequestTTLOverridesConfiguredTTL(t *testing.T) { |
| resetCodexPromptCacheRedisForTest() |
| defer resetCodexPromptCacheRedisForTest() |
|
|
| mr, err := miniredis.Run() |
| if err != nil { |
| t.Fatalf("start miniredis: %v", err) |
| } |
| defer mr.Close() |
|
|
| cfg := &config.Config{} |
| cfg.CodexPromptCache.RedisURL = "redis://" + mr.Addr() + "/0" |
| cfg.CodexPromptCache.KeyPrefix = "cliproxy:test:" |
| cfg.CodexPromptCache.TTLSeconds = 3600 |
| cfg.CodexPromptCache.TimeoutMS = 500 |
|
|
| _, ok := getOrCreateCodexPromptCacheRedis(cfg, "gpt-5-user-ttl-override", 75*time.Second) |
| if !ok { |
| t.Fatalf("expected redis set to succeed") |
| } |
|
|
| ttl := mr.TTL("cliproxy:test:gpt-5-user-ttl-override") |
| if ttl <= 0 { |
| t.Fatalf("expected positive ttl, got %v", ttl) |
| } |
| if ttl > 76*time.Second || ttl < 74*time.Second { |
| t.Fatalf("expected ttl around 75s, got %v", ttl) |
| } |
| } |
|
|
