Luis Pater
6187919000
- Implemented `RecentRequestsSnapshot` in `Auth` to capture bucketed recent request data. - Added new fields and methods to `Auth` for tracking request success and failure counts over time. - Updated `/v0/management/auth-files` response to include recent request data for each auth record. - Introduced unit tests to validate request tracking and snapshot generation logic.
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestManagerMarkResultRecordsRecentRequests(t *testing.T) {
|
|
mgr := NewManager(nil, nil, nil)
|
|
auth := &Auth{
|
|
ID: "auth-1",
|
|
Provider: "antigravity",
|
|
Attributes: map[string]string{
|
|
"runtime_only": "true",
|
|
},
|
|
Metadata: map[string]any{
|
|
"type": "antigravity",
|
|
},
|
|
}
|
|
|
|
if _, err := mgr.Register(WithSkipPersist(context.Background()), auth); err != nil {
|
|
t.Fatalf("Register returned error: %v", err)
|
|
}
|
|
|
|
mgr.MarkResult(context.Background(), Result{AuthID: "auth-1", Provider: "antigravity", Model: "gpt-5", Success: true})
|
|
mgr.MarkResult(context.Background(), Result{AuthID: "auth-1", Provider: "antigravity", Model: "gpt-5", Success: false})
|
|
|
|
gotAuth, ok := mgr.GetByID("auth-1")
|
|
if !ok || gotAuth == nil {
|
|
t.Fatalf("GetByID returned ok=%v auth=%v", ok, gotAuth)
|
|
}
|
|
|
|
snapshot := gotAuth.RecentRequestsSnapshot(time.Now())
|
|
var successTotal int64
|
|
var failedTotal int64
|
|
for _, bucket := range snapshot {
|
|
successTotal += bucket.Success
|
|
failedTotal += bucket.Failed
|
|
}
|
|
if successTotal != 1 || failedTotal != 1 {
|
|
t.Fatalf("totals = success=%d failed=%d, want 1/1", successTotal, failedTotal)
|
|
}
|
|
}
|