|
| 1 | +import { env, SELF } from "cloudflare:test"; |
| 2 | +import { Client } from "@modelcontextprotocol/sdk/client/index.js"; |
| 3 | +import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js"; |
| 4 | +import { afterEach, describe, expect, it } from "vitest"; |
| 5 | + |
| 6 | +let client: Client | undefined; |
| 7 | + |
| 8 | +const authorizedFetch: typeof fetch = (input, init = {}) => { |
| 9 | + const headers = new Headers(init.headers); |
| 10 | + headers.set("authorization", "Bearer test-token"); |
| 11 | + return SELF.fetch(input, { ...init, headers }); |
| 12 | +}; |
| 13 | + |
| 14 | +afterEach(async () => { |
| 15 | + await client?.close(); |
| 16 | + client = undefined; |
| 17 | +}); |
| 18 | + |
| 19 | +describe("Computer Code Mode MCP", () => { |
| 20 | + it("serves public setup routes and keeps the container callback private", async () => { |
| 21 | + const home = await SELF.fetch("https://example.test/"); |
| 22 | + expect(home.status).toBe(200); |
| 23 | + expect(await home.text()).toContain("https://example.test/mcp"); |
| 24 | + |
| 25 | + const health = await SELF.fetch("https://example.test/health"); |
| 26 | + expect(health.status).toBe(200); |
| 27 | + expect(await health.text()).toBe("ok\n"); |
| 28 | + |
| 29 | + const internal = await SELF.fetch("https://example.test/ws"); |
| 30 | + expect(internal.status).toBe(404); |
| 31 | + }); |
| 32 | + |
| 33 | + it("requires the configured bearer token", async () => { |
| 34 | + const missing = await SELF.fetch("https://example.test/mcp", { method: "POST" }); |
| 35 | + expect(missing.status).toBe(401); |
| 36 | + expect(missing.headers.get("www-authenticate")).toBe("Bearer"); |
| 37 | + |
| 38 | + const wrong = await SELF.fetch("https://example.test/mcp", { |
| 39 | + method: "POST", |
| 40 | + headers: { authorization: "Bearer test-tokem" }, |
| 41 | + }); |
| 42 | + expect(wrong.status).toBe(401); |
| 43 | + |
| 44 | + const get = await authorizedFetch("https://example.test/mcp"); |
| 45 | + expect(get.status).toBe(405); |
| 46 | + expect(get.headers.get("allow")).toBe("POST"); |
| 47 | + |
| 48 | + const { COMPUTER_MCP } = env as unknown as { |
| 49 | + COMPUTER_MCP: DurableObjectNamespace; |
| 50 | + }; |
| 51 | + const id = COMPUTER_MCP.idFromName("direct-auth-test"); |
| 52 | + const direct = await COMPUTER_MCP.get(id).fetch("https://example.test/mcp", { |
| 53 | + method: "POST", |
| 54 | + }); |
| 55 | + expect(direct.status).toBe(401); |
| 56 | + }); |
| 57 | + |
| 58 | + it("exposes durable Computer tools through one Code Mode tool", async () => { |
| 59 | + client = new Client({ name: "computer-mcp-test", version: "1.0.0" }); |
| 60 | + const transport = new StreamableHTTPClientTransport(new URL("https://example.test/mcp"), { |
| 61 | + fetch: authorizedFetch, |
| 62 | + }); |
| 63 | + await client.connect(transport); |
| 64 | + |
| 65 | + const listed = await client.listTools(); |
| 66 | + expect(listed.tools.map((tool) => tool.name)).toEqual(["code"]); |
| 67 | + const description = listed.tools[0]?.description; |
| 68 | + expect(description).toContain("codemode.read"); |
| 69 | + expect(description).toContain('"worker-shell"'); |
| 70 | + expect(description).toContain("no ambient outbound network"); |
| 71 | + expect(description).toContain("HTTPS URLs"); |
| 72 | + expect(description).toContain("Cannot run npm"); |
| 73 | + expect(description).toContain('"container-shell"'); |
| 74 | + expect(description).toContain("Full Debian Linux"); |
| 75 | + expect(description).toContain("Cold starts more slowly"); |
| 76 | + |
| 77 | + const result = await client.callTool({ |
| 78 | + name: "code", |
| 79 | + arguments: { |
| 80 | + code: `async () => { |
| 81 | + await codemode.write({ path: "/workspace/message.txt", content: "hello" }); |
| 82 | + await codemode.edit({ |
| 83 | + path: "/workspace/message.txt", |
| 84 | + edits: [{ oldText: "hello", newText: "hello from Code Mode" }] |
| 85 | + }); |
| 86 | + const file = await codemode.read({ path: "/workspace/message.txt" }); |
| 87 | + const listing = await codemode.ls({ path: "/workspace" }); |
| 88 | + const shell = await codemode.exec({ command: "pwd" }); |
| 89 | + const git = await codemode.exec({ command: "git init && git status --short" }); |
| 90 | + return { |
| 91 | + content: file.content, |
| 92 | + listed: listing.entries.some((entry) => entry.name === "message.txt"), |
| 93 | + backend: shell.backend, |
| 94 | + cwd: shell.stdout.trim(), |
| 95 | + gitWorked: git.exitCode === 0 && git.stdout.includes("message.txt") |
| 96 | + }; |
| 97 | + }`, |
| 98 | + }, |
| 99 | + }); |
| 100 | + |
| 101 | + expect(result.isError, JSON.stringify(result)).not.toBe(true); |
| 102 | + expect(readTextResult(result)).toEqual({ |
| 103 | + content: "hello from Code Mode", |
| 104 | + listed: true, |
| 105 | + backend: "worker-shell", |
| 106 | + cwd: "/workspace", |
| 107 | + gitWorked: true, |
| 108 | + }); |
| 109 | + |
| 110 | + const persisted = await client.callTool({ |
| 111 | + name: "code", |
| 112 | + arguments: { |
| 113 | + code: `async () => { |
| 114 | + const file = await codemode.read({ path: "/workspace/message.txt" }); |
| 115 | + return file.content; |
| 116 | + }`, |
| 117 | + }, |
| 118 | + }); |
| 119 | + expect(readTextResult(persisted)).toBe("hello from Code Mode"); |
| 120 | + |
| 121 | + const outbound = await client.callTool({ |
| 122 | + name: "code", |
| 123 | + arguments: { |
| 124 | + code: `async () => { |
| 125 | + const response = await fetch("https://example.com"); |
| 126 | + return response.status; |
| 127 | + }`, |
| 128 | + }, |
| 129 | + }); |
| 130 | + expect(outbound.isError).toBe(true); |
| 131 | + }); |
| 132 | +}); |
| 133 | + |
| 134 | +function readTextResult(result: Awaited<ReturnType<Client["callTool"]>>) { |
| 135 | + const content = result.content as Array<{ type: string; text?: string }>; |
| 136 | + const text = content.find((item) => item.type === "text"); |
| 137 | + if (!text?.text) throw new Error("Expected a text MCP result."); |
| 138 | + try { |
| 139 | + return JSON.parse(text.text) as unknown; |
| 140 | + } catch { |
| 141 | + return text.text; |
| 142 | + } |
| 143 | +} |
0 commit comments