lywsvip/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-09-03 06:35:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): improve test reliability by avoiding wall-clock time in TTL and cache-eviction tests

Browse Source
This commit is contained in:
hkfires
2026-09-01 09:24:05 +08:00
parent 8bcae8cf20
commit 22a57401a1
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/api/server_test.go
View File

@@ -151,6 +151,7 @@ func (e *codexSearchCaptureExecutor) HttpRequest(_ context.Context, selected *au
}
type codexSearchHomeDispatcher struct {
authID string
calls atomic.Int32
policy atomic.Value
}
@@ -199,18 +200,22 @@ func (*codexSearchHomeDispatcher) HeartbeatOK() bool { return true }
func (d *codexSearchHomeDispatcher) RPopAuth(_ context.Context, model string, _ string, _ http.Header, _ int) ([]byte, error) {
d.calls.Add(1)
authID := d.authID
if authID == "" {
authID = "home-codex-search"
}
return json.Marshal(map[string]any{
"model": model,
"auth_index": "home-codex-search",
"auth_index": authID,
"auth": map[string]any{
"id": "home-codex-search",
"id": authID,
"provider": "codex",
"status": "active",
"metadata": map[string]any{"access_token": "home-search-token"},
},
"concurrency": map[string]any{
"accounted": true,
"credential_id": "home-codex-search",
"credential_id": authID,
"model": model,
},
})
@@ -467,7 +472,8 @@ func TestHomeCodexAlphaSearchReportsUnauthorizedBeforeEarlyReturn(t *testing.T)
server := newTestServer(t)
registry := executionregistry.New()
server.handlers.AuthManager.SetConfig(&proxyconfig.Config{Home: proxyconfig.HomeConfig{Enabled: true}})
server.handlers.AuthManager.PublishHomeDispatch(&codexSearchHomeDispatcher{}, registry, 1)
testAuthID := "home-codex-search-" + strings.ReplaceAll(test.name, " ", "-")
server.handlers.AuthManager.PublishHomeDispatch(&codexSearchHomeDispatcher{authID: testAuthID}, registry, 1)
executor := &codexSearchCaptureExecutor{
statuses: []int{http.StatusUnauthorized},
responseBody: test.responseBody(),
@@ -476,7 +482,7 @@ func TestHomeCodexAlphaSearchReportsUnauthorizedBeforeEarlyReturn(t *testing.T)
executor.beforeReturn = func() { test.beforeReturn(registry) }
}
server.handlers.AuthManager.RegisterExecutor(executor)
usageCapture := registerHomeUnauthorizedUsageCapture(t, t.Name(), "home-codex-search")
usageCapture := registerHomeUnauthorizedUsageCapture(t, t.Name(), testAuthID)
recorder := httptest.NewRecorder()
request := httptest.NewRequest(http.MethodPost, "/v1/alpha/search", strings.NewReader(`{"model":"gpt-5-codex","query":"test"}`))

6
internal/cache/antigravity_reasoning_replay_cache_test.go vendored
View File

@@ -259,6 +259,12 @@ func TestAntigravityReasoningReplayUnrelatedEvictionDoesNotBlockAbsentSnapshot(t
if !CacheAntigravityReasoningReplayItems(model, "older-live-entry", [][]byte{liveItem}) {
t.Fatal("live entry write failed")
}
antigravityReasoningReplayMu.Lock()
if entry, ok := antigravityReasoningReplayEntries[antigravityReasoningReplayCacheKey(model, "older-live-entry")]; ok {
entry.Timestamp = time.Now().Add(-time.Minute)
antigravityReasoningReplayEntries[antigravityReasoningReplayCacheKey(model, "older-live-entry")] = entry
}
antigravityReasoningReplayMu.Unlock()
_, snapshot, found, errGet := GetAntigravityReasoningReplayItemsWithSnapshotRequired(context.Background(), model, "untouched-absent-session")
if errGet != nil || found {
t.Fatalf("initial absent snapshot = found %v, err %v", found, errGet)