mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 06:47:29 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
667 lines
19 KiB
Go
667 lines
19 KiB
Go
package plugins
|
||
|
||
import (
|
||
"encoding/json"
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
"time"
|
||
|
||
agentAPI "github.com/JianFeeeee/HologramAgent/internal/agent/api"
|
||
agentCore "github.com/JianFeeeee/HologramAgent/internal/agent/core"
|
||
agentIO "github.com/JianFeeeee/HologramAgent/internal/agent/io"
|
||
internalConfig "github.com/JianFeeeee/HologramAgent/internal/config"
|
||
"github.com/JianFeeeee/HologramAgent/internal/knowledge"
|
||
luaVM "github.com/JianFeeeee/HologramAgent/internal/lua"
|
||
"github.com/JianFeeeee/HologramAgent/internal/memory"
|
||
doc "github.com/JianFeeeee/HologramAgent/internal/memory/document"
|
||
"github.com/JianFeeeee/HologramAgent/internal/plugin"
|
||
cli "github.com/JianFeeeee/HologramAgent/internal/plugins/cli"
|
||
"github.com/JianFeeeee/HologramAgent/internal/plugins/webui"
|
||
sdk "github.com/JianFeeeee/HologramAgent/internal/sdk"
|
||
)
|
||
|
||
type testPluginEnv struct {
|
||
tmpDir string
|
||
stageHost *agentCore.StageHost
|
||
iom *agentIO.IOManager
|
||
pluginReg *plugin.Registry
|
||
memDB *memory.GraphDB
|
||
ks *knowledge.Store
|
||
docStore *doc.Store
|
||
cfgReg *internalConfig.ConfigRegistry
|
||
}
|
||
|
||
func setupIntegration(t *testing.T) *testPluginEnv {
|
||
return setupIntegrationWithProvider(t, nil)
|
||
}
|
||
|
||
// requireExternalTool 断言某个工具已在本次装配中注册，未注册则跳过用例。
|
||
//
|
||
// 语义不是「功能坏了所以跳过」，而是「这个工具已不属于内置装配」：cmd / agentcli
|
||
// 等插件在 2026-10-19 外置为 hmap 插件后，setupIntegration 只加载 internal/plugins
|
||
// 下的内置包，因此这里根本不会出现它们的工具。若不加这层保护，用例会拿到
|
||
// 「工具不存在或未注册」这种与真实插件行为无关的假失败，长期看会训练大家
|
||
// 忽略红灯。
|
||
//
|
||
// 保留用例而不是删除：断言本身（exit_code/stdout/stderr、PTY 交互与 resize）
|
||
// 仍然有效，只是验证现场必须是**装好 .hmap 的实例**。真正的端到端覆盖放在
|
||
// 那层跑；这里在纯内置装配下只能 Skip，不能假装通过。
|
||
func requireExternalTool(t *testing.T, env *testPluginEnv, tool string) {
|
||
t.Helper()
|
||
for _, def := range env.stageHost.GetToolDefs() {
|
||
if def.Name == tool {
|
||
return
|
||
}
|
||
}
|
||
t.Skipf("%s 由已外置插件提供（不在本次内置装配中），需在装上 .hmap 的实例上验证", tool)
|
||
}
|
||
|
||
func setupIntegrationWithProvider(t *testing.T, pm *agentAPI.ProviderManager) *testPluginEnv {
|
||
t.Helper()
|
||
|
||
// 测试绝不能绑生产端口。这里加载的是**全部内置插件**（含 webui），
|
||
// 而 webui 默认监听 :8080 —— 那正是生产实例的端口。
|
||
// 不覆盖的后果不是测试失败，而是**测试静默地跟生产抢端口**：
|
||
// 先到者胜，另一个 bind 失败。webui 现在会在 bind 失败时明确报错，
|
||
// 于是表现为「测试随机失败」+「生产 WebUI 随机死掉」，且两边看起来
|
||
// 互不相干，极难定位（实际踩到过）。
|
||
//
|
||
// 用 SetListenOverride 指到 127.0.0.1:0（内核分配空闲端口）：
|
||
// 测试拿到真实可用的 HTTP 服务，且与任何固定端口实例互不干扰。
|
||
if prev := webui.ListenOverride(); prev == "" {
|
||
webui.SetListenOverride("127.0.0.1:0")
|
||
t.Cleanup(func() { webui.SetListenOverride(prev) })
|
||
}
|
||
|
||
tmpDir, err := os.MkdirTemp("", "hc_integration_*")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
stageHost := agentCore.NewStageHost()
|
||
iom := agentIO.NewIOManager()
|
||
pluginReg := plugin.NewRegistry()
|
||
|
||
pluginReg.SetIOManager(iom)
|
||
pluginReg.SetMemory(nil)
|
||
|
||
pluginReg.SetToolRegistrar(func(name string, def sdk.ToolDef, handler sdk.ToolHandler) error {
|
||
return stageHost.RegisterTool(name, def, handler)
|
||
})
|
||
pluginReg.SetStageRegistrar(func(stage sdk.Stage, handler sdk.StageHandler) {
|
||
stageHost.RegisterStage(stage, handler)
|
||
})
|
||
pluginReg.SetAPIRegistrar(func(name string) error {
|
||
return nil
|
||
})
|
||
|
||
memDB, err := memory.NewGraphDB(filepath.Join(tmpDir, "test.db"))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
pluginReg.SetMemory(memDB)
|
||
|
||
ks := knowledge.NewStore(filepath.Join(tmpDir, "knowledge"))
|
||
if err := ks.Start(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
docStore := doc.NewStore(filepath.Join(tmpDir, "documents"), memory.TokenizeWords)
|
||
if err := docStore.Start(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
pluginReg.SetStageHost(stageHost)
|
||
pluginReg.SetProviderManager(pm)
|
||
pluginReg.SetKnowledge(ks)
|
||
pluginReg.SetDocStore(docStore)
|
||
|
||
cli.DefaultSocket = filepath.Join(tmpDir, "cli.sock")
|
||
|
||
// 经 ConfigRegistry 装配内核路径配置（clawhubadapter/pluginmgr 等经 SDK settings 读取）
|
||
cfgReg := internalConfig.NewConfigRegistry("")
|
||
cfgReg.SeedDefaults(tmpDir)
|
||
|
||
// ★ 预置临时端口，避免测试之间（以及本机生产实例）抢固定默认端口。
|
||
//
|
||
// 为什么必须在 Load 之前预置：插件表由 RegisterDef 在插件 Start 时创建，
|
||
// 此时才能 Set；而端口冲突发生在 Start 内部（net.Listen 失败即 HTTP 服务
|
||
// 静默不启动，或测试二进制被信号打断）。SetPluginConfig 会先建表再写，
|
||
// 正好填补这个时序缺口。
|
||
//
|
||
// 用 :0 让 OS 分配空闲端口——固定端口在「并行跑测试」或「本机有 homed
|
||
// 常驻」时必然周期性失败（实测：干净树连跑 20 轮也复现 2 次）。
|
||
for _, kv := range []struct{ plugin, key string }{
|
||
{"pluginmgr", "http_addr"},
|
||
{"remotedevice", "listen_addr"},
|
||
} {
|
||
if err := cfgReg.SetPluginConfig(kv.plugin, kv.key, "127.0.0.1:0"); err != nil {
|
||
t.Fatalf("预置 %s.%s 临时端口: %v", kv.plugin, kv.key, err)
|
||
}
|
||
}
|
||
// webui 的监听地址走独立旁路（SetListenOverride 优先级高于 settings，
|
||
// 因为历史上内核在插件表建立前写 settings 会失败）。
|
||
webui.SetListenOverride("127.0.0.1:0")
|
||
t.Cleanup(func() { webui.SetListenOverride("") })
|
||
|
||
pluginReg.SetConfigRegistry(cfgReg)
|
||
|
||
plgDir := filepath.Join(tmpDir, "plugins")
|
||
os.MkdirAll(plgDir, 0755)
|
||
if err := pluginReg.Load(plgDir); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
return &testPluginEnv{
|
||
tmpDir: tmpDir,
|
||
stageHost: stageHost,
|
||
iom: iom,
|
||
pluginReg: pluginReg,
|
||
memDB: memDB,
|
||
ks: ks,
|
||
docStore: docStore,
|
||
cfgReg: cfgReg,
|
||
}
|
||
}
|
||
|
||
func (e *testPluginEnv) cleanup() {
|
||
e.pluginReg.StopAll()
|
||
e.memDB.Close()
|
||
e.ks.Stop()
|
||
e.docStore.Stop()
|
||
os.RemoveAll(e.tmpDir)
|
||
}
|
||
|
||
// ---------------------------------------------------------------------------
|
||
// Registration
|
||
// ---------------------------------------------------------------------------
|
||
|
||
func TestIntegrationAllPluginsRegister(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
expectedTools := []string{
|
||
"healthcheck", "healthcheck_plugins", "healthcheck_tools", "healthcheck_memory", "healthcheck_perf",
|
||
// ⚠️ 以下插件的工具**故意不在**这里（它们已外置为 hmap 插件）：
|
||
//
|
||
// timer → plugins/timer
|
||
// ai_image / files → 2026-10-19 外置
|
||
// agentcli / cmd / kbtree / localuse / mcp /
|
||
// multimodal → 2026-10-19 外置
|
||
//
|
||
// 本用例的语义是「internal/plugins 下**仍内置**的包是否都注册了工具」，
|
||
// setupIntegration 只加载内置包 ⇒ 已外置的插件不参与本进程装配，
|
||
// 写进来只会造出「永远绿不了」的假失败。
|
||
//
|
||
// 它们的工具是否真的注册成功，由另一层负责：装完 .hmap 后实调一次。
|
||
// 别为了「让它绿」把插件加回 all.go —— 内置与外部两份同时存在会
|
||
// 导致工具重名，而胜者取决于加载顺序（最难排查的一类故障）。
|
||
}
|
||
|
||
defs := env.stageHost.GetToolDefs()
|
||
registered := make(map[string]bool)
|
||
for _, def := range defs {
|
||
registered[def.Name] = true
|
||
}
|
||
|
||
var missing []string
|
||
for _, tool := range expectedTools {
|
||
if !registered[tool] {
|
||
missing = append(missing, tool)
|
||
}
|
||
}
|
||
if len(missing) > 0 {
|
||
all := make([]string, 0, len(defs))
|
||
for _, d := range defs {
|
||
all = append(all, d.Name)
|
||
}
|
||
t.Fatalf("missing tools: %v\nall registered: %v", missing, all)
|
||
}
|
||
t.Logf("all %d expected tools registered (total: %d)", len(expectedTools), len(defs))
|
||
}
|
||
|
||
// ---------------------------------------------------------------------------
|
||
// cmd_run
|
||
// ---------------------------------------------------------------------------
|
||
|
||
func TestIntegrationCmdRun(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "cmd_run")
|
||
|
||
result, err := env.stageHost.ExecuteTool("cmd_run", map[string]interface{}{
|
||
"command": "echo hello_world",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["exit_code"] != 0.0 {
|
||
t.Fatalf("expected exit_code 0, got %v", resp["exit_code"])
|
||
}
|
||
stdout, ok := resp["stdout"].(string)
|
||
if !ok || stdout != "hello_world" {
|
||
t.Fatalf("expected stdout 'hello_world', got %q", stdout)
|
||
}
|
||
t.Logf("cmd_run OK: exit_code=0 stdout=%q", stdout)
|
||
}
|
||
|
||
func TestIntegrationCmdRunWithWorkdir(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "cmd_run")
|
||
|
||
result, err := env.stageHost.ExecuteTool("cmd_run", map[string]interface{}{
|
||
"command": "pwd",
|
||
"workdir": "/tmp",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["exit_code"] != 0.0 {
|
||
t.Fatalf("expected exit_code 0, got %v", resp["exit_code"])
|
||
}
|
||
stdout := resp["stdout"].(string)
|
||
if stdout != "/tmp" {
|
||
t.Fatalf("expected stdout '/tmp', got %q", stdout)
|
||
}
|
||
t.Logf("cmd_run workdir OK: stdout=%q", stdout)
|
||
}
|
||
|
||
func TestIntegrationCmdRunInvalidTimeout(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "cmd_run")
|
||
|
||
result, err := env.stageHost.ExecuteTool("cmd_run", map[string]interface{}{
|
||
"command": "echo ok",
|
||
"timeout": "not-a-duration",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
errMsg, ok := resp["error"].(string)
|
||
if !ok || errMsg == "" {
|
||
t.Fatalf("expected error for invalid timeout, got %v", resp)
|
||
}
|
||
t.Logf("cmd_run invalid_timeout OK: error=%q", errMsg)
|
||
}
|
||
|
||
func TestIntegrationCmdRunStderr(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "cmd_run")
|
||
|
||
result, err := env.stageHost.ExecuteTool("cmd_run", map[string]interface{}{
|
||
"command": "sh -c \"echo stderr_test >&2\"",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["exit_code"] != 0.0 {
|
||
t.Fatalf("expected exit_code 0, got %v", resp["exit_code"])
|
||
}
|
||
stderr := resp["stderr"].(string)
|
||
if stderr != "stderr_test" {
|
||
t.Fatalf("expected stderr 'stderr_test', got %q", stderr)
|
||
}
|
||
t.Logf("cmd_run stderr OK: stderr=%q", stderr)
|
||
}
|
||
|
||
// ---------------------------------------------------------------------------
|
||
// Terminal (PTY)
|
||
// ---------------------------------------------------------------------------
|
||
|
||
func TestIntegrationPtyCreateListClose(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "terminal_create")
|
||
|
||
// Create
|
||
result, err := env.stageHost.ExecuteTool("terminal_create", map[string]interface{}{
|
||
"shell": "/bin/sh",
|
||
"name": "hci_test_shell",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
id, ok := resp["id"].(string)
|
||
if !ok || id == "" {
|
||
status := resp["status"]
|
||
detail := resp["detail"]
|
||
if status == "error" {
|
||
t.Skipf("PTY not available: %v", detail)
|
||
}
|
||
t.Fatalf("expected non-empty terminal id, got id=%q status=%v detail=%v", id, status, detail)
|
||
}
|
||
t.Logf("terminal_create OK: id=%s", id)
|
||
|
||
// List
|
||
result2, err := env.stageHost.ExecuteTool("terminal_list", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
data2, _ := json.Marshal(result2)
|
||
var resp2 map[string]interface{}
|
||
json.Unmarshal(data2, &resp2)
|
||
|
||
terminals, ok := resp2["terminals"].([]interface{})
|
||
if !ok {
|
||
t.Fatalf("expected terminals array, got %T", resp2["terminals"])
|
||
}
|
||
if len(terminals) < 1 {
|
||
t.Fatal("expected at least 1 terminal")
|
||
}
|
||
t.Logf("terminal_list OK: %d terminals", len(terminals))
|
||
|
||
// Close
|
||
_, err = env.stageHost.ExecuteTool("terminal_close", map[string]interface{}{
|
||
"id": id,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Logf("terminal_close OK: id=%s", id)
|
||
}
|
||
|
||
func TestIntegrationPtyInteractive(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "terminal_create")
|
||
|
||
createResult, err := env.stageHost.ExecuteTool("terminal_create", map[string]interface{}{
|
||
"shell": "/bin/sh",
|
||
"name": "hci_interactive",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
cdata, _ := json.Marshal(createResult)
|
||
var cresp map[string]interface{}
|
||
json.Unmarshal(cdata, &cresp)
|
||
|
||
id, ok := cresp["id"].(string)
|
||
if !ok || id == "" {
|
||
status := cresp["status"]
|
||
if status == "error" {
|
||
t.Skipf("PTY not available: %v", cresp["detail"])
|
||
}
|
||
t.Fatalf("expected terminal id, got %v", cresp)
|
||
}
|
||
|
||
// Write a command
|
||
_, err = env.stageHost.ExecuteTool("terminal_write", map[string]interface{}{
|
||
"id": id,
|
||
"input": "echo pty_works\n",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
time.Sleep(500 * time.Millisecond)
|
||
|
||
// Read output
|
||
readResult, err := env.stageHost.ExecuteTool("terminal_read", map[string]interface{}{
|
||
"id": id,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
rdata, _ := json.Marshal(readResult)
|
||
var rresp map[string]interface{}
|
||
json.Unmarshal(rdata, &rresp)
|
||
|
||
output, ok := rresp["output"].(string)
|
||
if !ok || output == "" {
|
||
t.Fatalf("expected output, got output=%q response=%v", output, rresp)
|
||
}
|
||
t.Logf("terminal_write+read OK: output=%q", output)
|
||
|
||
// Close
|
||
env.stageHost.ExecuteTool("terminal_close", map[string]interface{}{
|
||
"id": id,
|
||
})
|
||
}
|
||
|
||
func TestIntegrationPtyResize(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
requireExternalTool(t, env, "terminal_create")
|
||
|
||
createResult, err := env.stageHost.ExecuteTool("terminal_create", map[string]interface{}{
|
||
"shell": "/bin/sh",
|
||
"name": "hci_resize_test",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
cdata, _ := json.Marshal(createResult)
|
||
var cresp map[string]interface{}
|
||
json.Unmarshal(cdata, &cresp)
|
||
|
||
id, ok := cresp["id"].(string)
|
||
if !ok || id == "" {
|
||
status := cresp["status"]
|
||
if status == "error" {
|
||
t.Skipf("PTY not available: %v", cresp["detail"])
|
||
}
|
||
t.Fatalf("expected terminal id, got %v", cresp)
|
||
}
|
||
|
||
_, err = env.stageHost.ExecuteTool("terminal_resize", map[string]interface{}{
|
||
"id": id,
|
||
"rows": 80.0,
|
||
"cols": 200.0,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Logf("terminal_resize OK: id=%s rows=80 cols=200", id)
|
||
|
||
env.stageHost.ExecuteTool("terminal_close", map[string]interface{}{
|
||
"id": id,
|
||
})
|
||
}
|
||
|
||
// ---------------------------------------------------------------------------
|
||
// Healthcheck
|
||
// ---------------------------------------------------------------------------
|
||
|
||
func TestIntegrationHealthcheck(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
result, err := env.stageHost.ExecuteTool("healthcheck", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["status"] != "ok" {
|
||
t.Fatalf("expected status=ok, got %v", resp["status"])
|
||
}
|
||
checks := resp["checks"].([]interface{})
|
||
if len(checks) == 0 {
|
||
t.Fatal("expected non-empty checks array")
|
||
}
|
||
t.Logf("healthcheck OK: %d checks, status=%v", len(checks), resp["status"])
|
||
}
|
||
|
||
func TestIntegrationHealthcheckPlugins(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
result, err := env.stageHost.ExecuteTool("healthcheck_plugins", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["status"] != "ok" {
|
||
t.Fatalf("expected status=ok, got %v", resp["status"])
|
||
}
|
||
t.Logf("healthcheck_plugins OK: status=%v", resp["status"])
|
||
}
|
||
|
||
func TestIntegrationHealthcheckToolsList(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
result, err := env.stageHost.ExecuteTool("healthcheck_tools", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["status"] != "ok" {
|
||
t.Fatalf("expected status=ok, got %v", resp["status"])
|
||
}
|
||
t.Logf("healthcheck_tools OK: status=%v", resp["status"])
|
||
}
|
||
|
||
func TestIntegrationHealthcheckMemory(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
result, err := env.stageHost.ExecuteTool("healthcheck_memory", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
if resp["status"] != "ok" {
|
||
t.Fatalf("expected status=ok, got %v", resp["status"])
|
||
}
|
||
t.Logf("healthcheck_memory OK: status=%v", resp["status"])
|
||
}
|
||
|
||
func TestIntegrationToolNotFound(t *testing.T) {
|
||
env := setupIntegration(t)
|
||
defer env.cleanup()
|
||
|
||
_, err := env.stageHost.ExecuteTool("nonexistent_tool_xyz", nil)
|
||
if err == nil {
|
||
t.Fatal("expected error for nonexistent tool")
|
||
}
|
||
t.Logf("tool_not_found OK: err=%v", err)
|
||
}
|
||
|
||
func TestIntegrationLLMDrivenDiscoveryWithRealKey(t *testing.T) {
|
||
apiKey := os.Getenv("DEEPSEEK_API_KEY")
|
||
if apiKey == "" {
|
||
t.Skip("DEEPSEEK_API_KEY not set")
|
||
}
|
||
|
||
// Lua VM + DeepSeek Provider
|
||
tmpVM := t.TempDir()
|
||
vm := luaVM.NewVM(tmpVM + "/adapters")
|
||
if err := vm.Start(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer vm.Stop()
|
||
|
||
pm := agentAPI.NewProviderManager()
|
||
pm.Register("deepseek", agentAPI.NewLuaAdaptedProvider(agentAPI.BaseConfig{
|
||
Model: "deepseek-v4-flash",
|
||
BaseURL: "https://api.deepseek.com",
|
||
APIKey: apiKey,
|
||
}, vm, "deepseek", "deepseek"))
|
||
|
||
// Setup — 加载所有真实内置插件
|
||
env := setupIntegrationWithProvider(t, pm)
|
||
defer env.cleanup()
|
||
|
||
// 列出已加载的真实工具
|
||
defs := env.stageHost.GetToolDefs()
|
||
t.Logf("共有 %d 个已注册的真实工具:", len(defs))
|
||
for _, d := range defs {
|
||
t.Logf(" - %s", d.Name)
|
||
}
|
||
|
||
// 调用 healthcheck 进行全面检查（含 LLM 驱动阶段）
|
||
result, err := env.stageHost.ExecuteTool("healthcheck", map[string]interface{}{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
data, _ := json.Marshal(result)
|
||
var resp map[string]interface{}
|
||
json.Unmarshal(data, &resp)
|
||
|
||
t.Logf("===== Full Healthcheck Result =====")
|
||
t.Logf("status: %v", resp["status"])
|
||
t.Logf("summary: %v", resp["summary"])
|
||
t.Logf("total: %v", resp["total"])
|
||
t.Logf("passed: %v", resp["passed"])
|
||
t.Logf("failed: %v", resp["failed"])
|
||
|
||
checks := resp["checks"].([]interface{})
|
||
for _, c := range checks {
|
||
cr := c.(map[string]interface{})
|
||
prefix := "✅"
|
||
if cr["status"] == "fail" {
|
||
prefix = "❌"
|
||
}
|
||
t.Logf(" %s %s: %s %s", prefix, cr["name"], cr["status"], cr["detail"])
|
||
}
|
||
|
||
if resp["status"] != "ok" {
|
||
t.Fatalf("expected status=ok, got %v", resp["status"])
|
||
}
|
||
|
||
// 验证 LLM 发现阶段的存在
|
||
foundDiscovery := false
|
||
for _, c := range checks {
|
||
cr := c.(map[string]interface{})
|
||
if cr["name"] == "llm_discovery" {
|
||
foundDiscovery = true
|
||
if cr["status"] != "ok" {
|
||
t.Fatalf("LLM discovery failed: status=%s detail=%s", cr["status"], cr["detail"])
|
||
}
|
||
break
|
||
}
|
||
}
|
||
if !foundDiscovery {
|
||
t.Fatal("expected llm_discovery check in results")
|
||
}
|
||
|
||
t.Logf("✅ LLM-driven tool discovery test PASSED")
|
||
}
|