diff --git a/internal/server/piagent_e2e_test.go b/internal/server/piagent_e2e_test.go index 685c8403..ca0d5c5f 100644 --- a/internal/server/piagent_e2e_test.go +++ b/internal/server/piagent_e2e_test.go @@ -25,20 +25,30 @@ import ( // // 运行方式(需 pi-agent runner 在线,建议 1h 超时): // -// go test ./internal/server/ -run TestPiAgent_FullLifecycle -timeout 1h -v +// go test ./internal/server/ -run TestPiAgent_FullLifecycle -timeout 2h -v func TestPiAgent_FullLifecycle_MultiRole_RealWorld_E2E(t *testing.T) { if testing.Short() { t.Skip("skipping slow real-world E2E test in short mode") } // 1. 初始化配置并强制指定 pi-agent 后端与真实 Runner + runnerURL := os.Getenv("PI_AGENT_RUNNER_URL") + if runnerURL == "" { + runnerURL = "http://127.0.0.1:18865" + } + runnerToken := os.Getenv("PI_AGENT_RUNNER_AUTH_TOKEN") + if runnerToken == "" { + runnerToken = os.Getenv("RUNNER_AUTH_TOKEN") + } cfg := &config.Config{ - Env: "dev", - AgentBackend: "pi-agent", - PiAgentRunnerURL: "http://127.0.0.1:18865", - PiAgentModel: "MiniMax-M3", - PiAgentProvider: "minimax", - DataDir: t.TempDir(), - AllowInMemory: true, + Env: "dev", + AgentBackend: "pi-agent", + AgentTimeout: 60 * time.Minute, + PiAgentRunnerURL: runnerURL, + PiAgentRunnerAuthToken: runnerToken, + PiAgentModel: "MiniMax-M3", + PiAgentProvider: "minimax", + DataDir: t.TempDir(), + AllowInMemory: true, } svc := bizdev.MustNewService(t, cfg) @@ -46,8 +56,8 @@ func TestPiAgent_FullLifecycle_MultiRole_RealWorld_E2E(t *testing.T) { ts := httptest.NewServer(srv.Routes()) defer ts.Close() - // 单次 pi-agent 远程调用可能超过 10min;HTTP client 留 50min 余量,整体由 go test -timeout 1h 兜底。 - client := &http.Client{Timeout: 50 * time.Minute} + // 单次 pi-agent 远程调用可能超过 30min;HTTP client 与 AgentTimeout 对齐,整体由 go test -timeout 2h 兜底。 + client := &http.Client{Timeout: 65 * time.Minute} ctx := context.Background() stepStart := time.Now() logStep := func(label string) {