mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-06 16:15:50 +08:00
4b5f1eab25
- Introduce `WebSocketResponseObserver` capability and bump plugin ABI schema version to 4. - Forward upstream WebSocket response frames from Codex and xAI executors to configured observers. - Wire `WebSocketResponseObserver` across API handlers and plugin host dispatchers. Closes: #5248
74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
package helps
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
cliproxyauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
|
|
cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor"
|
|
sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator"
|
|
)
|
|
|
|
func TestEmitWebSocketResponseEvent(t *testing.T) {
|
|
var received []cliproxyexecutor.WebSocketResponseEvent
|
|
opts := cliproxyexecutor.Options{
|
|
SourceFormat: sdktranslator.FormatOpenAI,
|
|
Metadata: map[string]any{
|
|
"request_id": "test-req-1",
|
|
"trace_id": "test-trace-1",
|
|
},
|
|
WebSocketResponseObserver: func(_ context.Context, ev cliproxyexecutor.WebSocketResponseEvent) {
|
|
received = append(received, ev)
|
|
},
|
|
}
|
|
|
|
auth := &cliproxyauth.Auth{
|
|
ID: "auth-123",
|
|
Label: "primary-auth",
|
|
Provider: "codex",
|
|
Metadata: map[string]any{
|
|
"email": "alice@example.com",
|
|
},
|
|
Attributes: map[string]string{
|
|
"auth_kind": "oauth",
|
|
},
|
|
}
|
|
|
|
payload := []byte(`{"type":"codex.rate_limits","rate_limits":{"primary":{"used_percent":42}}}`)
|
|
EmitWebSocketResponseEvent(context.Background(), opts, auth, "codex", "gpt-5.3-codex", payload)
|
|
|
|
if len(received) != 1 {
|
|
t.Fatalf("received %d events, want 1", len(received))
|
|
}
|
|
ev := received[0]
|
|
if ev.RequestID != "test-req-1" {
|
|
t.Fatalf("RequestID = %q, want test-req-1", ev.RequestID)
|
|
}
|
|
if ev.TraceID != "test-trace-1" {
|
|
t.Fatalf("TraceID = %q, want test-trace-1", ev.TraceID)
|
|
}
|
|
if ev.Provider != "codex" {
|
|
t.Fatalf("Provider = %q, want codex", ev.Provider)
|
|
}
|
|
if ev.AuthID != "auth-123" || ev.AuthLabel != "primary-auth" {
|
|
t.Fatalf("Auth = (%q, %q), want (auth-123, primary-auth)", ev.AuthID, ev.AuthLabel)
|
|
}
|
|
if ev.AuthType != "oauth" {
|
|
t.Fatalf("AuthType = %q, want oauth", ev.AuthType)
|
|
}
|
|
if ev.EventType != "codex.rate_limits" {
|
|
t.Fatalf("EventType = %q, want codex.rate_limits", ev.EventType)
|
|
}
|
|
if !bytes.Equal(ev.Payload, payload) {
|
|
t.Fatalf("Payload = %s, want %s", ev.Payload, payload)
|
|
}
|
|
}
|
|
|
|
func TestEmitWebSocketResponseEventNilObserverNoOp(t *testing.T) {
|
|
opts := cliproxyexecutor.Options{}
|
|
payload := []byte(`{"type":"codex.rate_limits"}`)
|
|
// Should not panic
|
|
EmitWebSocketResponseEvent(context.Background(), opts, nil, "codex", "gpt-5.3-codex", payload)
|
|
}
|