Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
6.0 kB · 192 lines
Go
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193package cliproxy
import ( "context" "net/http" "testing"
"github.com/router-for-me/CLIProxyAPI/v7/internal/pluginhost" runtimeexecutor "github.com/router-for-me/CLIProxyAPI/v7/internal/runtime/executor" coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth" cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor" "github.com/router-for-me/CLIProxyAPI/v7/sdk/config")
type serviceTestPluginExecutor struct{}type serviceTestSDKExecutor struct{ serviceTestPluginExecutor }
func (serviceTestSDKExecutor) Identifier() string { return "sdk-provider" }
func (serviceTestPluginExecutor) Identifier() string { return "plugin-provider"}
func (serviceTestPluginExecutor) Execute(context.Context, *coreauth.Auth, cliproxyexecutor.Request, cliproxyexecutor.Options) (cliproxyexecutor.Response, error) { return cliproxyexecutor.Response{}, nil}
func (serviceTestPluginExecutor) ExecuteStream(context.Context, *coreauth.Auth, cliproxyexecutor.Request, cliproxyexecutor.Options) (*cliproxyexecutor.StreamResult, error) { return nil, nil}
func (serviceTestPluginExecutor) Refresh(_ context.Context, auth *coreauth.Auth) (*coreauth.Auth, error) { return auth, nil}
func (serviceTestPluginExecutor) CountTokens(context.Context, *coreauth.Auth, cliproxyexecutor.Request, cliproxyexecutor.Options) (cliproxyexecutor.Response, error) { return cliproxyexecutor.Response{}, nil}
func (serviceTestPluginExecutor) HttpRequest(context.Context, *coreauth.Auth, *http.Request) (*http.Response, error) { return nil, nil}
func TestRegisterAvailableExecutors(t *testing.T) { oldRegisterPluginExecutors := registerPluginExecutors pluginRegisterCalls := 0 var expectedPluginHost *pluginhost.Host var expectedManager *coreauth.Manager registerPluginExecutors = func(host *pluginhost.Host, manager *coreauth.Manager) { pluginRegisterCalls++ if host != expectedPluginHost { t.Fatalf("plugin executor registration host = %p, want %p", host, expectedPluginHost) } if manager != expectedManager { t.Fatalf("plugin executor registration manager = %p, want %p", manager, expectedManager) } manager.RegisterExecutor(serviceTestPluginExecutor{}) } t.Cleanup(func() { registerPluginExecutors = oldRegisterPluginExecutors })
service := &Service{ cfg: &config.Config{}, coreManager: coreauth.NewManager(nil, nil, nil), pluginHost: pluginhost.New(), } expectedPluginHost = service.pluginHost expectedManager = service.coreManager service.ensureWebsocketGateway()
service.registerAvailableExecutors(nil, executorRegistrationOptions{ includeBaseline: true, includePlugins: true, })
if pluginRegisterCalls != 1 { t.Fatalf("plugin executor registration calls = %d, want 1", pluginRegisterCalls) }
providers := []string{ "codex", "claude", "gemini", "gemini-interactions", "vertex", "aistudio", "antigravity", "kimi", "xai", "openai-compatibility", "plugin-provider", } for _, provider := range providers { resolved, ok := service.coreManager.Executor(provider) if !ok || resolved == nil { t.Fatalf("expected executor for provider %s after registration", provider) } }
resolved, _ := service.coreManager.Executor("plugin-provider") if _, isPlugin := resolved.(serviceTestPluginExecutor); !isPlugin { t.Fatalf("executor type = %T, want serviceTestPluginExecutor", resolved) }}
func TestSyncPluginModelRuntimePreservesSDKExecutorUnlessForced(t *testing.T) { manager := coreauth.NewManager(nil, nil, nil) custom := serviceTestSDKExecutor{} manager.RegisterExecutor(custom) auth := &coreauth.Auth{ID: "private-auth", Provider: custom.Identifier()} if _, err := manager.Register(context.Background(), auth); err != nil { t.Fatal(err) } service := &Service{cfg: &config.Config{}, coreManager: manager, pluginHost: pluginhost.New()}
service.syncPluginModelRuntime(context.Background()) got, ok := manager.Executor(custom.Identifier()) if !ok || got != custom { t.Fatalf("plugin model sync replaced SDK executor with %T", got) }
service.registerExecutorForAuth(auth, true) got, ok = manager.Executor(custom.Identifier()) if !ok { t.Fatal("forced registration removed executor") } if _, replaced := got.(*runtimeexecutor.OpenAICompatExecutor); !replaced { t.Fatalf("forced registration kept %T, want *executor.OpenAICompatExecutor", got) }}
func TestRegisterExecutorForAuth_OpenAICompatUsesNamespacedProviderKey(t *testing.T) { testCases := []struct { name string auths []*coreauth.Auth }{ { name: "native first", auths: []*coreauth.Auth{ {ID: "native-kimi", Provider: "kimi"}, openAICompatKimiAuth(), }, }, { name: "compat first", auths: []*coreauth.Auth{ openAICompatKimiAuth(), {ID: "native-kimi", Provider: "kimi"}, }, }, }
for _, tt := range testCases { t.Run(tt.name, func(t *testing.T) { service := &Service{ cfg: &config.Config{}, coreManager: coreauth.NewManager(nil, nil, nil), }
service.registerExecutorsForAuths(tt.auths, true)
nativeExecutor, okNative := service.coreManager.Executor("kimi") if !okNative { t.Fatal("expected native kimi executor") } if _, okKimi := nativeExecutor.(*runtimeexecutor.KimiExecutor); !okKimi { t.Fatalf("native executor type = %T, want *executor.KimiExecutor", nativeExecutor) }
compatExecutor, okCompat := service.coreManager.Executor("openai-compatible-kimi") if !okCompat { t.Fatal("expected namespaced OpenAI-compatible executor") } if _, okOpenAICompat := compatExecutor.(*runtimeexecutor.OpenAICompatExecutor); !okOpenAICompat { t.Fatalf("compat executor type = %T, want *executor.OpenAICompatExecutor", compatExecutor) } }) }}
func openAICompatKimiAuth() *coreauth.Auth { return &coreauth.Auth{ ID: "compat-kimi", Provider: "openai-compatibility", Label: "kimi", Attributes: map[string]string{ "compat_name": "kimi", "provider_key": "kimi", }, }}