airkjw/CLIProxyAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(auth): add global model registry usage to conductor override tests

Browse Source
This commit is contained in:
hkfires
2026-03-03 09:18:56 +08:00
parent c44793789b
commit 532107b4fa
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sdk/cliproxy/auth/conductor_overrides_test.go
+15 -2
View File
@@ -7,6 +7,8 @@ import (
"testing" "testing"
"time" "time"
"github.com/google/uuid"
"github.com/router-for-me/CLIProxyAPI/v6/internal/registry"
cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/executor" cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/executor"
) )
@@ -115,8 +117,19 @@ func newCredentialRetryLimitTestManager(t *testing.T, maxRetryCredentials int) (
executor := &credentialRetryLimitExecutor{id: "claude"} executor := &credentialRetryLimitExecutor{id: "claude"}
m.RegisterExecutor(executor) m.RegisterExecutor(executor)
auth1 := &Auth{ID: "auth-1", Provider: "claude"} baseID := uuid.NewString()
auth2 := &Auth{ID: "auth-2", Provider: "claude"} auth1 := &Auth{ID: baseID + "-auth-1", Provider: "claude"}
auth2 := &Auth{ID: baseID + "-auth-2", Provider: "claude"}
// Auth selection requires that the global model registry knows each credential supports the model.
reg := registry.GetGlobalRegistry()
reg.RegisterClient(auth1.ID, "claude", []*registry.ModelInfo{{ID: "test-model"}})
reg.RegisterClient(auth2.ID, "claude", []*registry.ModelInfo{{ID: "test-model"}})
t.Cleanup(func() {
reg.UnregisterClient(auth1.ID)
reg.UnregisterClient(auth2.ID)
})
if _, errRegister := m.Register(context.Background(), auth1); errRegister != nil { if _, errRegister := m.Register(context.Background(), auth1); errRegister != nil {
t.Fatalf("register auth1: %v", errRegister) t.Fatalf("register auth1: %v", errRegister)
} }