feat: initial commit of vps-commander release v1.0.0

2026-09-28 07:48:01 +08:00
commit 467e28a6a1
48 changed files with 3851 additions and 0 deletions
@@ -0,0 +1,112 @@
+package mcp
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+)
+
+type mockBackend struct{}
+
+func (m *mockBackend) ListDevices(ctx context.Context) (any, error) {
+	return []map[string]any{{"name": "local", "status": "online"}}, nil
+}
+func (m *mockBackend) ExecCommand(ctx context.Context, clientIP, device, command, workdir string, timeout int) (any, error) {
+	return map[string]any{"device": device, "exit_code": 0, "stdout": "ok"}, nil
+}
+func (m *mockBackend) ReadFile(ctx context.Context, clientIP, device, path string, offset, limit int64) (any, error) {
+	return map[string]any{"device": device, "path": path, "content": "hello"}, nil
+}
+func (m *mockBackend) WriteFile(ctx context.Context, clientIP, device, path, content string) (any, error) {
+	return map[string]any{"device": device, "path": path, "bytes": len(content)}, nil
+}
+
+func TestMCPInitializeAndListTools(t *testing.T) {
+	srv := NewServer(&mockBackend{}, nil)
+
+	// Test Initialize
+	initReq := JSONRPCRequest{
+		JSONRPC: "2.0",
+		ID:      1,
+		Method:  "initialize",
+	}
+	body, _ := json.Marshal(initReq)
+	req := httptest.NewRequest("POST", "/mcp/message", bytes.NewReader(body))
+	w := httptest.NewRecorder()
+	srv.HandleMessage(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Fatalf("expected 200, got %d", w.Code)
+	}
+
+	var resp JSONRPCResponse
+	if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
+		t.Fatal(err)
+	}
+	if resp.Error != nil {
+		t.Fatalf("unexpected error: %v", resp.Error)
+	}
+
+	// Test tools/list
+	listReq := JSONRPCRequest{
+		JSONRPC: "2.0",
+		ID:      2,
+		Method:  "tools/list",
+	}
+	body, _ = json.Marshal(listReq)
+	req = httptest.NewRequest("POST", "/mcp/message", bytes.NewReader(body))
+	w = httptest.NewRecorder()
+	srv.HandleMessage(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Fatalf("expected 200, got %d", w.Code)
+	}
+
+	var listResp JSONRPCResponse
+	if err := json.Unmarshal(w.Body.Bytes(), &listResp); err != nil {
+		t.Fatal(err)
+	}
+	resMap, ok := listResp.Result.(map[string]any)
+	if !ok {
+		t.Fatalf("expected map result")
+	}
+	tools, ok := resMap["tools"].([]any)
+	if !ok || len(tools) != 4 {
+		t.Fatalf("expected 4 tools, got %d", len(tools))
+	}
+}
+
+func TestMCPCallTool(t *testing.T) {
+	srv := NewServer(&mockBackend{}, nil)
+
+	callReq := JSONRPCRequest{
+		JSONRPC: "2.0",
+		ID:      3,
+		Method:  "tools/call",
+		Params:  json.RawMessage(`{"name":"exec_command","arguments":{"command":"echo test"}}`),
+	}
+	body, _ := json.Marshal(callReq)
+	req := httptest.NewRequest("POST", "/mcp/message", bytes.NewReader(body))
+	w := httptest.NewRecorder()
+	srv.HandleMessage(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Fatalf("expected 200, got %d", w.Code)
+	}
+
+	var resp JSONRPCResponse
+	if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
+		t.Fatal(err)
+	}
+	if resp.Error != nil {
+		t.Fatalf("unexpected error: %v", resp.Error)
+	}
+	contentStr := string(w.Body.Bytes())
+	if !strings.Contains(contentStr, "stdout") {
+		t.Fatalf("expected stdout in result, got %s", contentStr)
+	}
+}