@lobu/worker 6.1.1 → 7.1.0

package/src/__tests__/embedded-mcp-cli-bash.test.ts

@@ -0,0 +1,179 @@
+import { afterEach, describe, expect, test } from "bun:test";
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import type { McpToolDef } from "@lobu/core";
+import {
+  buildMcpCliCommands,
+  type McpCliDeps,
+  type McpRuntimeRef,
+} from "../embedded/mcp-cli-commands";
+import type { GatewayParams } from "../shared/tool-implementations";
+
+const tempDirs: string[] = [];
+
+afterEach(() => {
+  for (const dir of tempDirs.splice(0)) {
+    fs.rmSync(dir, { recursive: true, force: true });
+  }
+});
+
+const gw: GatewayParams = {
+  gatewayUrl: "http://gateway",
+  workerToken: "worker-token",
+  channelId: "channel-1",
+  conversationId: "conversation-1",
+  platform: "telegram",
+};
+
+const searchKnowledge: McpToolDef = {
+  name: "search_memory",
+  description: "Search memory",
+  inputSchema: {
+    type: "object",
+    properties: { query: { type: "string" } },
+    required: ["query"],
+  },
+};
+
+async function buildBash(options: {
+  ref: McpRuntimeRef;
+  callTool: (
+    mcpId: string,
+    toolName: string,
+    payload: Record<string, unknown>
+  ) => Promise<{ content: Array<{ type: "text"; text: string }> }>;
+}) {
+  const { Bash, ReadWriteFs, defineCommand } = await import("just-bash");
+  const tmp = fs.mkdtempSync(path.join(os.tmpdir(), "lobu-mcp-cli-"));
+  tempDirs.push(tmp);
+
+  const callTool: McpCliDeps["callTool"] = async (
+    _gw,
+    mcpId,
+    toolName,
+    payload
+  ) => options.callTool(mcpId, toolName, payload);
+
+  const cliCommands = buildMcpCliCommands(options.ref, gw, { callTool });
+
+  const customCommands = cliCommands.map((c) =>
+    defineCommand(c.name, async (args: string[], ctx) => {
+      return c.execute(args, {
+        stdin: typeof ctx.stdin === "string" ? ctx.stdin : "",
+      });
+    })
+  );
+
+  const bash = new Bash({
+    fs: new ReadWriteFs({ root: tmp }),
+    cwd: "/",
+    env: { PATH: "/usr/bin:/bin" },
+    customCommands,
+  });
+
+  return bash;
+}
+
+describe("embedded MCP CLI through real just-bash", () => {
+  test("heredoc JSON on stdin reaches the handler as a parsed object", async () => {
+    const ref: McpRuntimeRef = {
+      current: {
+        mcpTools: { lobu: [searchKnowledge] },
+        mcpStatus: [],
+        mcpContext: {},
+      },
+    };
+    const calls: Array<{ payload: Record<string, unknown> }> = [];
+    const bash = await buildBash({
+      ref,
+      callTool: async (_mcpId, _toolName, payload) => {
+        calls.push({ payload });
+        return { content: [{ type: "text", text: "hit" }] };
+      },
+    });
+
+    const result = await bash.exec(
+      `lobu search_memory <<'EOF'
+{"query":"architecture"}
+EOF`,
+      { cwd: "/" }
+    );
+
+    expect(result.exitCode).toBe(0);
+    expect(result.stdout.trim()).toBe("hit");
+    expect(calls).toEqual([{ payload: { query: "architecture" } }]);
+  });
+
+  test("echo … | <server> <tool> routes the piped JSON into the handler", async () => {
+    const ref: McpRuntimeRef = {
+      current: {
+        mcpTools: { lobu: [searchKnowledge] },
+        mcpStatus: [],
+        mcpContext: {},
+      },
+    };
+    const calls: Array<{ payload: Record<string, unknown> }> = [];
+    const bash = await buildBash({
+      ref,
+      callTool: async (_mcpId, _toolName, payload) => {
+        calls.push({ payload });
+        return { content: [{ type: "text", text: "ok" }] };
+      },
+    });
+
+    const result = await bash.exec(
+      `echo '{"query":"piped"}' | lobu search_memory`,
+      { cwd: "/" }
+    );
+
+    expect(result.exitCode).toBe(0);
+    expect(calls).toEqual([{ payload: { query: "piped" } }]);
+  });
+
+  test("<server> --help lists the discovered tools", async () => {
+    const ref: McpRuntimeRef = {
+      current: {
+        mcpTools: { lobu: [searchKnowledge] },
+        mcpStatus: [
+          {
+            id: "lobu",
+            name: "Lobu",
+            requiresAuth: true,
+            requiresInput: false,
+            authenticated: true,
+            configured: true,
+          },
+        ],
+        mcpContext: {},
+      },
+    };
+    const bash = await buildBash({
+      ref,
+      callTool: async () => ({ content: [] }),
+    });
+
+    const result = await bash.exec("lobu --help", { cwd: "/" });
+    expect(result.exitCode).toBe(0);
+    expect(result.stdout).toContain("search_memory");
+    expect(result.stdout).toContain("auth login|check|logout");
+  });
+
+  test("unknown subcommand exits with stderr diagnostic", async () => {
+    const ref: McpRuntimeRef = {
+      current: {
+        mcpTools: { lobu: [] },
+        mcpStatus: [],
+        mcpContext: {},
+      },
+    };
+    const bash = await buildBash({
+      ref,
+      callTool: async () => ({ content: [] }),
+    });
+
+    const result = await bash.exec("lobu mystery", { cwd: "/" });
+    expect(result.exitCode).toBe(2);
+    expect(result.stderr).toContain("unknown tool");
+  });
+});