Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
2.3 kB · 73 lines
Go
at main
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374package 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)}