86 lines
4.2 KiB
Go
86 lines
4.2 KiB
Go
package agent
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestCoopBaseURL(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
provider string
|
||
engine string
|
||
baseURL string
|
||
want string
|
||
}{
|
||
// ---- pi / claude（用 @anthropic-ai/sdk，自带 /v1/messages，不能补 /v1）----
|
||
// Anthropic：剥离 /v1，避免双重 /v1/v1/messages
|
||
{"pi/anthropic bare", "anthropic", "pi", "https://api.deepseek.com/anthropic", "https://api.deepseek.com/anthropic"},
|
||
{"pi/anthropic trailing slash", "anthropic", "pi", "https://api.deepseek.com/anthropic/", "https://api.deepseek.com/anthropic"},
|
||
{"pi/anthropic with /v1", "anthropic", "pi", "https://api.deepseek.com/anthropic/v1", "https://api.deepseek.com/anthropic"},
|
||
{"pi/anthropic with /v1/messages", "anthropic", "pi", "https://api.deepseek.com/anthropic/v1/messages", "https://api.deepseek.com/anthropic"},
|
||
{"pi/anthropic with /messages", "anthropic", "pi", "https://api.deepseek.com/anthropic/messages", "https://api.deepseek.com/anthropic"},
|
||
{"pi/anthropic anthropic.com", "anthropic", "pi", "https://api.anthropic.com", "https://api.anthropic.com"},
|
||
// claude 行为与 pi 一致
|
||
{"claude/anthropic bare", "anthropic", "claude", "https://api.deepseek.com/anthropic", "https://api.deepseek.com/anthropic"},
|
||
{"claude/anthropic with /v1", "anthropic", "claude", "https://api.deepseek.com/anthropic/v1", "https://api.deepseek.com/anthropic"},
|
||
|
||
// ---- pigo（anthropicCompatDriver 只追加 /messages，需补 /v1）----
|
||
{"pigo/anthropic bare", "anthropic", "pigo", "https://api.deepseek.com/anthropic", "https://api.deepseek.com/anthropic/v1"},
|
||
{"pigo/anthropic trailing slash", "anthropic", "pigo", "https://api.deepseek.com/anthropic/", "https://api.deepseek.com/anthropic/v1"},
|
||
{"pigo/anthropic with /v1", "anthropic", "pigo", "https://api.deepseek.com/anthropic/v1", "https://api.deepseek.com/anthropic/v1"},
|
||
{"pigo/anthropic with /v1/messages", "anthropic", "pigo", "https://api.deepseek.com/anthropic/v1/messages", "https://api.deepseek.com/anthropic/v1"},
|
||
{"pigo/anthropic with /messages", "anthropic", "pigo", "https://api.deepseek.com/anthropic/messages", "https://api.deepseek.com/anthropic/v1"},
|
||
{"pigo/anthropic anthropic.com", "anthropic", "pigo", "https://api.anthropic.com", "https://api.anthropic.com/v1"},
|
||
|
||
// ---- OpenAI 协议（三种 engine 行为一致：保留 /v1，剥离 /chat/completions）----
|
||
{"pi/openai full url", "openai", "pi", "https://api.siliconflow.cn/v1/chat/completions", "https://api.siliconflow.cn/v1"},
|
||
{"pi/openai with /v1", "openai", "pi", "https://api.siliconflow.cn/v1", "https://api.siliconflow.cn/v1"},
|
||
{"pi/openai bare", "openai", "pi", "https://api.siliconflow.cn", "https://api.siliconflow.cn"},
|
||
{"pigo/openai full url", "openai", "pigo", "https://api.siliconflow.cn/v1/chat/completions", "https://api.siliconflow.cn/v1"},
|
||
{"pigo/openai bare", "openai", "pigo", "https://api.siliconflow.cn", "https://api.siliconflow.cn"},
|
||
}
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got := coopBaseURL(tt.provider, tt.baseURL, tt.engine)
|
||
if got != tt.want {
|
||
t.Errorf("coopBaseURL(%q, %q, %q) = %q, want %q", tt.provider, tt.baseURL, tt.engine, got, tt.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestRunCoopErrorPaths(t *testing.T) {
|
||
apiCfg := &APIConfigStore{}
|
||
apiCfg.config = APIConfig{APIKey: "test-key", BaseURL: "https://api.deepseek.com/anthropic", Model: "deepseek-chat"}
|
||
|
||
toolset := &Toolset{
|
||
Workspace: ".",
|
||
Timeout: 30 * time.Second,
|
||
MaxOutput: 4000,
|
||
apiCfg: apiCfg,
|
||
dockerSocket: defaultDockerSocket,
|
||
coop: NewCoopManager(),
|
||
sessionRoot: t.TempDir(),
|
||
}
|
||
|
||
// 1. 缺少 task
|
||
r := toolset.runCoop(context.Background(), "test-session", map[string]any{})
|
||
if r.Success || r.Output == "" {
|
||
t.Fatalf("empty task should fail, got %#v", r)
|
||
}
|
||
|
||
// 2. 正常参数：应进入 Docker 检查阶段（本机无 Docker 时返回连接/镜像错误而非 panic）
|
||
r = toolset.runCoop(context.Background(), "test-session", map[string]any{
|
||
"task": "测试任务",
|
||
"round_max": 2,
|
||
"timeout": 30,
|
||
})
|
||
t.Logf("runCoop output: %s", r.Output)
|
||
if r.Success {
|
||
// 若真的跑成功了（有 Docker 且镜像存在），也无妨
|
||
t.Logf("coop unexpectedly succeeded")
|
||
}
|
||
}
|