File size: 2,307 Bytes
2986042 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | package app
import (
"net/http"
"testing"
"time"
)
func TestHandleActiveRequests(t *testing.T) {
t.Parallel()
m := newActiveRequestManager()
id := m.Register(time.Now(), "m1", "1.2.3.4", true)
m.Update(id, 10, "ch", "openai", "sk-test", 7, 1.5) //nolint:gosec // 测试用假凭证
m.AddBytes(id, 123)
m.SetClientFirstByteTime(id, 50*time.Millisecond)
s := &Server{activeRequests: m}
c, w := newTestContext(t, newRequest(http.MethodGet, "/admin/active_requests", nil))
s.HandleActiveRequests(c)
if w.Code != http.StatusOK {
t.Fatalf("status=%d, want %d", w.Code, http.StatusOK)
}
var resp struct {
Success bool `json:"success"`
Data []ActiveRequest `json:"data"`
Count int `json:"count"`
}
mustUnmarshalJSON(t, w.Body.Bytes(), &resp)
if !resp.Success || resp.Count != 1 || len(resp.Data) != 1 {
t.Fatalf("unexpected resp: %+v", resp)
}
if resp.Data[0].BytesReceived != 123 {
t.Fatalf("bytes_received=%d, want 123", resp.Data[0].BytesReceived)
}
if resp.Data[0].ClientFirstByteTime <= 0 {
t.Fatalf("client_first_byte_time=%v, want >0", resp.Data[0].ClientFirstByteTime)
}
if resp.Data[0].CostMultiplier != 1.5 {
t.Fatalf("cost_multiplier=%v, want 1.5", resp.Data[0].CostMultiplier)
}
}
func TestHandleActiveRequests_PreservesZeroCostMultiplier(t *testing.T) {
t.Parallel()
m := newActiveRequestManager()
id := m.Register(time.Now(), "m1", "1.2.3.4", true)
m.Update(id, 10, "free-channel", "openai", "sk-test", 7, 0) //nolint:gosec // 测试用假凭证
s := &Server{activeRequests: m}
c, w := newTestContext(t, newRequest(http.MethodGet, "/admin/active_requests", nil))
s.HandleActiveRequests(c)
if w.Code != http.StatusOK {
t.Fatalf("status=%d, want %d", w.Code, http.StatusOK)
}
var resp struct {
Success bool `json:"success"`
Data []map[string]any `json:"data"`
Count int `json:"count"`
}
mustUnmarshalJSON(t, w.Body.Bytes(), &resp)
if !resp.Success || resp.Count != 1 || len(resp.Data) != 1 {
t.Fatalf("unexpected resp: %+v", resp)
}
value, ok := resp.Data[0]["cost_multiplier"]
if !ok {
t.Fatalf("cost_multiplier missing in response: %+v", resp.Data[0])
}
if got, ok := value.(float64); !ok || got != 0 {
t.Fatalf("cost_multiplier=%v, want 0", value)
}
}
|