29 lines
974 B
Go
29 lines
974 B
Go
package auth
|
|
|
|
import (
|
|
"testing"
|
|
|
|
internalconfig "github.com/router-for-me/CLIProxyAPI/v7/internal/config"
|
|
cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor"
|
|
)
|
|
|
|
func TestResolvedAPIKeyModelInfoPropagatesIsCompat(t *testing.T) {
|
|
manager := NewManager(nil, nil, nil)
|
|
manager.SetConfig(&internalconfig.Config{ClaudeKey: []internalconfig.ClaudeKey{{
|
|
APIKey: "compat-key",
|
|
Prefix: "tenant",
|
|
Models: []internalconfig.ClaudeModel{{
|
|
Name: "deepseek-upstream",
|
|
Alias: "deepseek-alias",
|
|
IsCompat: true,
|
|
}},
|
|
}}})
|
|
auth := configuredCapabilityTestAuth("compat-auth", "compat-key")
|
|
registerCapabilityTestAuth(t, manager, auth)
|
|
|
|
req := manager.attachResolvedAPIKeyModelInfo(cliproxyexecutor.Request{}, auth, "tenant/deepseek-alias", "deepseek-upstream")
|
|
info, ok := ResolvedAPIKeyModelInfo(req)
|
|
if !ok || info == nil || !info.IsCompat {
|
|
t.Fatalf("ResolvedAPIKeyModelInfo() = (%+v, %t), want IsCompat=true", info, ok)
|
|
}
|
|
}
|