mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-15 12:15:16 +00:00
refactor: rename openclaw -> clawhubadapter, add ClawHub search/install, fix agent interrupt
- Rename internal/plugins/openclaw/ -> internal/plugins/clawhubadapter/ - Add RegistryDispatcher with 5 sub-registries (Tool, Provider, Channel, Stage, Cap) - Add clawhubadapter_search tool for ClawHub marketplace search - Add clawhub: prefix support for installing from ClawHub (ZIP/tgz auto-detect) - Add CallProvider RPC and provider/call routing - Fix QQ interrupt: check interceptCh before/after each tool execution
This commit is contained in:
379
internal/plugins/clawhubadapter/sidecar_test.go
Normal file
379
internal/plugins/clawhubadapter/sidecar_test.go
Normal file
@ -0,0 +1,379 @@
|
||||
package clawhubadapter
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||||
)
|
||||
|
||||
func TestLaunchSidecarNoMainJS(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
sp, err := launchSidecar(tmpDir, "nonexistent")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if sp != nil {
|
||||
t.Fatal("expected nil for dir without main.js")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLaunchSidecarAndListTools(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
src := filepath.Join("testdata", "echoplugin", "main.js")
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read test plugin: %v", err)
|
||||
}
|
||||
dst := filepath.Join(tmpDir, "main.js")
|
||||
if err := os.WriteFile(dst, data, 0755); err != nil {
|
||||
t.Fatalf("write test plugin: %v", err)
|
||||
}
|
||||
|
||||
sp, err := launchSidecar(tmpDir, "echoplugin")
|
||||
if err != nil {
|
||||
t.Fatalf("launch sidecar: %v", err)
|
||||
}
|
||||
defer sp.Close()
|
||||
|
||||
tools, err := sp.ListTools()
|
||||
if err != nil {
|
||||
t.Fatalf("list tools: %v", err)
|
||||
}
|
||||
|
||||
if len(tools) == 0 {
|
||||
t.Fatal("expected at least one tool")
|
||||
}
|
||||
|
||||
found := false
|
||||
for _, tool := range tools {
|
||||
if tool.Name == "echo" {
|
||||
found = true
|
||||
if tool.Description == "" {
|
||||
t.Error("expected non-empty description for echo tool")
|
||||
}
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Fatal("expected 'echo' tool in list")
|
||||
}
|
||||
t.Logf("tools: %+v", tools)
|
||||
}
|
||||
|
||||
func TestCallEchoTool(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
src := filepath.Join("testdata", "echoplugin", "main.js")
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read test plugin: %v", err)
|
||||
}
|
||||
dst := filepath.Join(tmpDir, "main.js")
|
||||
if err := os.WriteFile(dst, data, 0755); err != nil {
|
||||
t.Fatalf("write test plugin: %v", err)
|
||||
}
|
||||
|
||||
sp, err := launchSidecar(tmpDir, "echoplugin")
|
||||
if err != nil {
|
||||
t.Fatalf("launch sidecar: %v", err)
|
||||
}
|
||||
defer sp.Close()
|
||||
|
||||
result, err := sp.CallTool("echo", map[string]interface{}{
|
||||
"text": "hello world",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("call echo tool: %v", err)
|
||||
}
|
||||
|
||||
expected := "Echo: hello world"
|
||||
if result != expected {
|
||||
t.Fatalf("expected %q, got %q", expected, result)
|
||||
}
|
||||
t.Logf("echo result: %s", result)
|
||||
}
|
||||
|
||||
func TestCallAddTool(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
src := filepath.Join("testdata", "echoplugin", "main.js")
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read test plugin: %v", err)
|
||||
}
|
||||
dst := filepath.Join(tmpDir, "main.js")
|
||||
if err := os.WriteFile(dst, data, 0755); err != nil {
|
||||
t.Fatalf("write test plugin: %v", err)
|
||||
}
|
||||
|
||||
sp, err := launchSidecar(tmpDir, "echoplugin")
|
||||
if err != nil {
|
||||
t.Fatalf("launch sidecar: %v", err)
|
||||
}
|
||||
defer sp.Close()
|
||||
|
||||
result, err := sp.CallTool("add", map[string]interface{}{
|
||||
"a": 3.0,
|
||||
"b": 4.0,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("call add tool: %v", err)
|
||||
}
|
||||
|
||||
expected := "7"
|
||||
if result != expected {
|
||||
t.Fatalf("expected %q, got %q", expected, result)
|
||||
}
|
||||
t.Logf("add result: %s", result)
|
||||
}
|
||||
|
||||
func TestCallNonexistentTool(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
src := filepath.Join("testdata", "echoplugin", "main.js")
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read test plugin: %v", err)
|
||||
}
|
||||
dst := filepath.Join(tmpDir, "main.js")
|
||||
if err := os.WriteFile(dst, data, 0755); err != nil {
|
||||
t.Fatalf("write test plugin: %v", err)
|
||||
}
|
||||
|
||||
sp, err := launchSidecar(tmpDir, "echoplugin")
|
||||
if err != nil {
|
||||
t.Fatalf("launch sidecar: %v", err)
|
||||
}
|
||||
defer sp.Close()
|
||||
|
||||
_, err = sp.CallTool("nonexistent", nil)
|
||||
if err == nil {
|
||||
t.Fatal("expected error for nonexistent tool")
|
||||
}
|
||||
t.Logf("expected error: %v", err)
|
||||
}
|
||||
|
||||
func TestConcurrentCalls(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
src := filepath.Join("testdata", "echoplugin", "main.js")
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read test plugin: %v", err)
|
||||
}
|
||||
dst := filepath.Join(tmpDir, "main.js")
|
||||
if err := os.WriteFile(dst, data, 0755); err != nil {
|
||||
t.Fatalf("write test plugin: %v", err)
|
||||
}
|
||||
|
||||
sp, err := launchSidecar(tmpDir, "echoplugin")
|
||||
if err != nil {
|
||||
t.Fatalf("launch sidecar: %v", err)
|
||||
}
|
||||
defer sp.Close()
|
||||
|
||||
done := make(chan bool, 5)
|
||||
for i := 0; i < 5; i++ {
|
||||
go func(n int) {
|
||||
result, err := sp.CallTool("add", map[string]interface{}{
|
||||
"a": float64(n),
|
||||
"b": float64(n * 2),
|
||||
})
|
||||
if err != nil {
|
||||
t.Errorf("concurrent call %d: %v", n, err)
|
||||
}
|
||||
if result != fmt.Sprintf("%d", n + n*2) {
|
||||
t.Errorf("call %d: expected %d, got %s", n, n + n*2, result)
|
||||
}
|
||||
done <- true
|
||||
}(i)
|
||||
}
|
||||
for i := 0; i < 5; i++ {
|
||||
<-done
|
||||
}
|
||||
}
|
||||
|
||||
// ---- Simulator + OpenClaw plugin tests ----
|
||||
|
||||
func launchSimulator(t *testing.T, pluginDir, name string) *sidecarProcess {
|
||||
t.Helper()
|
||||
simPath, err := filepath.Abs(filepath.Join("simulator", "main.js"))
|
||||
if err != nil {
|
||||
t.Fatalf("abs simulator path: %v", err)
|
||||
}
|
||||
sp, err := launchProcess("node", simPath, pluginDir, name)
|
||||
if err != nil {
|
||||
t.Fatalf("launch simulator for %s: %v", name, err)
|
||||
}
|
||||
return sp
|
||||
}
|
||||
|
||||
func TestSimulatorWithOCSimple(t *testing.T) {
|
||||
pluginDir, err := filepath.Abs(filepath.Join("testdata", "oc-simple"))
|
||||
if err != nil {
|
||||
t.Fatalf("abs testdata: %v", err)
|
||||
}
|
||||
sp := launchSimulator(t, pluginDir, "oc-simple")
|
||||
defer sp.Close()
|
||||
|
||||
tools, err := sp.ListTools()
|
||||
if err != nil {
|
||||
t.Fatalf("list tools: %v", err)
|
||||
}
|
||||
if len(tools) != 2 {
|
||||
t.Fatalf("expected 2 tools, got %d: %+v", len(tools), tools)
|
||||
}
|
||||
|
||||
found := map[string]bool{"greet": false, "ping": false}
|
||||
for _, tool := range tools {
|
||||
found[tool.Name] = true
|
||||
}
|
||||
if !found["greet"] || !found["ping"] {
|
||||
t.Fatalf("expected greet and ping tools, got %+v", tools)
|
||||
}
|
||||
|
||||
result, err := sp.CallTool("greet", map[string]interface{}{"name": "Test"})
|
||||
if err != nil {
|
||||
t.Fatalf("call greet: %v", err)
|
||||
}
|
||||
if result != "Hello, Test!" {
|
||||
t.Fatalf("expected 'Hello, Test!', got %q", result)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSimulatorWithOCPackage(t *testing.T) {
|
||||
pluginDir, err := filepath.Abs(filepath.Join("testdata", "oc-pkg"))
|
||||
if err != nil {
|
||||
t.Fatalf("abs testdata: %v", err)
|
||||
}
|
||||
sp := launchSimulator(t, pluginDir, "oc-pkg")
|
||||
defer sp.Close()
|
||||
|
||||
tools, err := sp.ListTools()
|
||||
if err != nil {
|
||||
t.Fatalf("list tools: %v", err)
|
||||
}
|
||||
if len(tools) != 2 {
|
||||
t.Fatalf("expected 2 tools, got %d: %+v", len(tools), tools)
|
||||
}
|
||||
|
||||
found := map[string]bool{"add": false, "info": false}
|
||||
for _, tool := range tools {
|
||||
found[tool.Name] = true
|
||||
if tool.Name == "add" {
|
||||
if tool.InputSchema == nil {
|
||||
t.Error("add tool should have inputSchema")
|
||||
}
|
||||
}
|
||||
}
|
||||
if !found["add"] || !found["info"] {
|
||||
t.Fatalf("expected add and info tools, got %+v", tools)
|
||||
}
|
||||
|
||||
result, err := sp.CallTool("add", map[string]interface{}{"a": 10.0, "b": 20.0})
|
||||
if err != nil {
|
||||
t.Fatalf("call add: %v", err)
|
||||
}
|
||||
if result != "30" {
|
||||
t.Fatalf("expected '30', got %q", result)
|
||||
}
|
||||
|
||||
infoResult, err := sp.CallTool("info", nil)
|
||||
if err != nil {
|
||||
t.Fatalf("call info: %v", err)
|
||||
}
|
||||
if infoResult == "" {
|
||||
t.Fatal("expected non-empty info result")
|
||||
}
|
||||
t.Logf("info result: %s", infoResult)
|
||||
}
|
||||
|
||||
func TestLoadOCPluginViaPluginStart(t *testing.T) {
|
||||
skillsDir := t.TempDir()
|
||||
|
||||
ocSimpleDir := filepath.Join(skillsDir, "oc-simple")
|
||||
if err := os.MkdirAll(ocSimpleDir, 0755); err != nil {
|
||||
t.Fatalf("mkdir: %v", err)
|
||||
}
|
||||
for _, name := range []string{"openclaw.plugin.json", "index.js"} {
|
||||
src := filepath.Join("testdata", "oc-simple", name)
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read %s: %v", name, err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(ocSimpleDir, name), data, 0644); err != nil {
|
||||
t.Fatalf("write %s: %v", name, err)
|
||||
}
|
||||
}
|
||||
|
||||
ocPkgDir := filepath.Join(skillsDir, "oc-pkg")
|
||||
if err := os.MkdirAll(filepath.Join(ocPkgDir, "lib"), 0755); err != nil {
|
||||
t.Fatalf("mkdir: %v", err)
|
||||
}
|
||||
for _, name := range []string{"package.json", "lib/entry.js"} {
|
||||
src := filepath.Join("testdata", "oc-pkg", name)
|
||||
data, err := os.ReadFile(src)
|
||||
if err != nil {
|
||||
t.Fatalf("read %s: %v", name, err)
|
||||
}
|
||||
dst := filepath.Join(ocPkgDir, name)
|
||||
if err := os.WriteFile(dst, data, 0644); err != nil {
|
||||
t.Fatalf("write %s: %v", name, err)
|
||||
}
|
||||
}
|
||||
|
||||
SimulatorDir = filepath.Join(t.TempDir(), ".simulator")
|
||||
|
||||
p := New("openclaw", skillsDir)
|
||||
|
||||
var registeredTools []string
|
||||
registeredHandlers := make(map[string]sdk.ToolHandler)
|
||||
sdk := sdk.New("openclaw", sdk.SDKConfig{
|
||||
RegTool: func(name string, def sdk.ToolDef, handler sdk.ToolHandler) error {
|
||||
registeredTools = append(registeredTools, name)
|
||||
registeredHandlers[name] = handler
|
||||
return nil
|
||||
},
|
||||
})
|
||||
|
||||
if err := p.Start(sdk); err != nil {
|
||||
t.Fatalf("start plugin: %v", err)
|
||||
}
|
||||
defer p.Stop()
|
||||
|
||||
expected := []string{"oc-simple_greet", "oc-simple_ping", "oc-pkg_add", "oc-pkg_info"}
|
||||
for _, exp := range expected {
|
||||
found := false
|
||||
for _, name := range registeredTools {
|
||||
if name == exp {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Fatalf("expected tool %q not registered. got: %v", exp, registeredTools)
|
||||
}
|
||||
}
|
||||
|
||||
handler, ok := registeredHandlers["oc-simple_greet"]
|
||||
if !ok {
|
||||
t.Fatal("greet handler not registered")
|
||||
}
|
||||
result, err := handler(map[string]interface{}{"name": "OpenClaw"})
|
||||
if err != nil {
|
||||
t.Fatalf("exec greet: %v", err)
|
||||
}
|
||||
if result != "Hello, OpenClaw!" {
|
||||
t.Fatalf("expected 'Hello, OpenClaw!', got %v", result)
|
||||
}
|
||||
|
||||
handler, ok = registeredHandlers["oc-pkg_add"]
|
||||
if !ok {
|
||||
t.Fatal("add handler not registered")
|
||||
}
|
||||
result, err = handler(map[string]interface{}{"a": 7.0, "b": 8.0})
|
||||
if err != nil {
|
||||
t.Fatalf("exec add: %v", err)
|
||||
}
|
||||
if result != "15" {
|
||||
t.Fatalf("expected '15', got %v", result)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user