4957f8bc22
新增运行策略 agentMaxSessions（默认 30，范围 1-64），保存后同时下发到内嵌进程闸门与独立 yingce-agent 的 /v1/limit，降低上限不会中断已开始的对话。系统性能页展示 Agent 服务健康、进行中/排队数量，并可直接修改上限。
103 lines
3.4 KiB
Go
103 lines
3.4 KiB
Go
package runtime
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// fakeRuntime 用一个假的 agent-runtime.mjs 替换真实运行时，只验证 Go 侧的进程与输出处理。
|
||
func fakeRuntime(t *testing.T, script string) {
|
||
t.Helper()
|
||
if _, err := exec.LookPath("node"); err != nil {
|
||
t.Skip("node is not installed")
|
||
}
|
||
dir := t.TempDir()
|
||
if err := os.WriteFile(filepath.Join(dir, "agent-runtime.mjs"), []byte(script), 0o600); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Setenv("CANVAS_PI_RUNTIME_DIR", dir)
|
||
}
|
||
|
||
func TestSetProcessLimitDoesNotStopActiveWork(t *testing.T) {
|
||
t.Cleanup(func() { SetProcessLimit(defaultAgentProcessLimit) })
|
||
release, err := acquireAgentProcess(context.Background())
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
active, limit := SetProcessLimit(1)
|
||
if active != 1 || limit != 1 {
|
||
t.Fatalf("active=%d limit=%d, want active 1 and limit 1", active, limit)
|
||
}
|
||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Millisecond)
|
||
defer cancel()
|
||
if _, err := acquireAgentProcess(ctx); err == nil {
|
||
t.Fatal("lowered limit still admitted another process")
|
||
}
|
||
release()
|
||
SetProcessLimit(defaultAgentProcessLimit)
|
||
}
|
||
|
||
func noopBridge() Bridge {
|
||
handler := func(context.Context, map[string]json.RawMessage) (any, error) { return map[string]any{"ok": true}, nil }
|
||
return Bridge{Model: handler, Tool: handler, Event: handler}
|
||
}
|
||
|
||
// 回归：Wait 早于读完 stdout 时，大量输出后正常退出会被判成
|
||
// "read |0: file already closed"。
|
||
func TestRunDrainsStdoutBeforeWait(t *testing.T) {
|
||
fakeRuntime(t, `
|
||
const line = JSON.stringify({ event: "progress", pad: "x".repeat(2048) }) + "\n";
|
||
for (let i = 0; i < 2000; i++) process.stdout.write(line);
|
||
process.stdout.write(JSON.stringify({ event: "settled" }) + "\n");
|
||
`)
|
||
for i := 0; i < 5; i++ {
|
||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||
err := Run(ctx, ProcessRequest{Model: map[string]any{"id": "m"}}, noopBridge())
|
||
cancel()
|
||
if err != nil {
|
||
t.Fatalf("attempt %d: normal exit reported as failure: %v", i, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// runtime_error 之后仍要读到 EOF，并把该错误作为运行结果返回。
|
||
func TestRunReportsRuntimeErrorAfterDrainingOutput(t *testing.T) {
|
||
fakeRuntime(t, `
|
||
process.stdout.write(JSON.stringify({ event: "runtime_error", message: "boom" }) + "\n");
|
||
const line = JSON.stringify({ event: "progress", pad: "x".repeat(2048) }) + "\n";
|
||
for (let i = 0; i < 500; i++) process.stdout.write(line);
|
||
process.exitCode = 1;
|
||
`)
|
||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||
defer cancel()
|
||
err := Run(ctx, ProcessRequest{Model: map[string]any{"id": "m"}}, noopBridge())
|
||
if err == nil || err.Error() != "boom" {
|
||
t.Fatalf("want runtime_error message, got %v", err)
|
||
}
|
||
}
|
||
|
||
// 子进程只能拿到白名单环境变量，拿不到数据库连接等密钥。
|
||
func TestRunDoesNotLeakBackendEnvironment(t *testing.T) {
|
||
t.Setenv("DATABASE_URL", "postgres://secret")
|
||
fakeRuntime(t, `
|
||
if (process.env.DATABASE_URL) {
|
||
process.stdout.write(JSON.stringify({ event: "runtime_error", message: "leaked" }) + "\n");
|
||
process.exitCode = 1;
|
||
}
|
||
`)
|
||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||
defer cancel()
|
||
if err := Run(ctx, ProcessRequest{Model: map[string]any{"id": "m"}}, noopBridge()); err != nil {
|
||
if strings.Contains(err.Error(), "leaked") {
|
||
t.Fatal("backend environment leaked into the Agent runtime")
|
||
}
|
||
t.Fatal(err)
|
||
}
|
||
}
|