cprime-supergateway 3.4.3

package/tests/stdioToStatefulStreamableHttp.test.ts

@@ -0,0 +1,70 @@
+import { test } from 'node:test'
+import assert from 'node:assert/strict'
+import { spawn, ChildProcess } from 'child_process'
+import { Client } from '@modelcontextprotocol/sdk/client/index.js'
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'
+
+const PORT = 11004
+const MCP_URL = `http://localhost:${PORT}/mcp`
+
+let gatewayProc: ChildProcess
+
+test.before(() => {
+  gatewayProc = spawn(
+    'npm',
+    [
+      'run',
+      'start',
+      '--',
+      '--stdio',
+      'node tests/helpers/mock-mcp-server.js stdio',
+      '--outputTransport',
+      'streamableHttp',
+      '--stateful',
+      '--port',
+      String(PORT),
+      '--streamableHttpPath',
+      '/mcp',
+    ],
+    { stdio: 'ignore', shell: false },
+  )
+  gatewayProc.unref()
+})
+
+test.after(async () => {
+  gatewayProc.kill('SIGINT')
+  await new Promise((resolve) => gatewayProc.once('exit', resolve))
+})
+
+test('stdioToStatefulStreamableHttp listTools and callTool', async () => {
+  const transport = new StreamableHTTPClientTransport(new URL(MCP_URL))
+  const client = new Client({ name: 'stateful-test', version: '1.0.0' })
+  await new Promise((r) => setTimeout(r, 2000))
+  await client.connect(transport)
+
+  assert.ok(transport.sessionId, 'sessionId should be set after connect')
+
+  const { tools } = await client.listTools()
+  assert.ok(tools.some((t) => t.name === 'add'))
+
+  type Reply = { content: Array<{ text: string }> }
+  const reply1 = (await client.callTool({
+    name: 'add',
+    arguments: { a: 1, b: 2 },
+  })) as Reply
+
+  assert.strictEqual(reply1.content[0].text, 'The sum of 1 and 2 is 3.')
+
+  const reply2 = (await client.callTool({
+    name: 'add',
+    arguments: { a: 3, b: 4 },
+  })) as Reply
+
+  assert.strictEqual(reply2.content[0].text, 'The sum of 3 and 4 is 7.')
+
+  await transport.terminateSession()
+  assert.strictEqual(transport.sessionId, undefined)
+
+  await client.close()
+  transport.close()
+})

package/tests/stdioToStatelessStreamableHttp.test.ts

@@ -0,0 +1,71 @@
+import { test } from 'node:test'
+import assert from 'node:assert/strict'
+import { spawn, ChildProcess } from 'child_process'
+import { Client } from '@modelcontextprotocol/sdk/client/index.js'
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'
+
+const PORT = 11005
+const MCP_URL = `http://localhost:${PORT}/mcp`
+
+let gatewayProc: ChildProcess
+
+test.before(() => {
+  gatewayProc = spawn(
+    'npm',
+    [
+      'run',
+      'start',
+      '--',
+      '--stdio',
+      'node tests/helpers/mock-mcp-server.js stdio',
+      '--outputTransport',
+      'streamableHttp',
+      '--port',
+      String(PORT),
+      '--streamableHttpPath',
+      '/mcp',
+    ],
+    { stdio: 'ignore', shell: false },
+  )
+  gatewayProc.unref()
+})
+
+test.after(async () => {
+  gatewayProc.kill('SIGINT')
+  await new Promise((resolve) => gatewayProc.once('exit', resolve))
+})
+
+test('stdioToStatelessStreamableHttp listTools and callTool', async () => {
+  const transport = new StreamableHTTPClientTransport(new URL(MCP_URL))
+  const client = new Client({ name: 'stateless-test', version: '1.0.0' })
+  await new Promise((r) => setTimeout(r, 2000))
+  await client.connect(transport)
+
+  assert.strictEqual(transport.sessionId, undefined)
+
+  const { tools } = await client.listTools()
+  assert.ok(tools.some((t) => t.name === 'add'))
+
+  type Reply = { content: Array<{ text: string }> }
+  const reply1 = (await client.callTool({
+    name: 'add',
+    arguments: { a: 4, b: 5 },
+  })) as Reply
+
+  assert.strictEqual(reply1.content[0].text, 'The sum of 4 and 5 is 9.')
+
+  const reply2 = (await client.callTool({
+    name: 'add',
+    arguments: { a: 2, b: 7 },
+  })) as Reply
+
+  assert.strictEqual(reply2.content[0].text, 'The sum of 2 and 7 is 9.')
+
+  await client.close()
+  transport.close()
+})
+
+test('GET returns 405', async () => {
+  const res = await fetch(MCP_URL)
+  assert.strictEqual(res.status, 405)
+})

package/tests/streamableHttpCli.test.ts

@@ -0,0 +1,24 @@
+import { test } from 'node:test'
+import assert from 'node:assert/strict'
+import yargs from 'yargs'
+import { hideBin } from 'yargs/helpers'
+
+test('yargs parses streamableHttp outputTransport', () => {
+  const argv = yargs(
+    hideBin([
+      'node',
+      '',
+      '--stdio',
+      'true',
+      '--outputTransport',
+      'streamableHttp',
+    ]),
+  )
+    .option('stdio', { type: 'string' })
+    .option('outputTransport', {
+      type: 'string',
+      choices: ['stdio', 'sse', 'ws', 'streamableHttp'],
+    })
+    .parseSync()
+  assert.strictEqual(argv.outputTransport, 'streamableHttp')
+})

package/tests/streamableHttpToStdio.test.ts

@@ -0,0 +1,64 @@
+import { test } from 'node:test'
+import assert from 'node:assert/strict'
+import { spawn, ChildProcess } from 'child_process'
+
+import { Client } from '@modelcontextprotocol/sdk/client/index.js'
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js'
+
+const MCP_PORT = 11002
+const MCP_URL = `http://localhost:${MCP_PORT}/mcp`
+
+let serverProc: ChildProcess | undefined
+
+function spawnMcpServer(): Promise<ChildProcess> {
+  return new Promise((res, rej) => {
+    const proc = spawn(
+      'node',
+      ['tests/helpers/mock-mcp-server.js', 'streamableHttp'],
+      {
+        env: { ...process.env, PORT: String(MCP_PORT) },
+        shell: false,
+        stdio: ['inherit', 'pipe', 'inherit'],
+      },
+    )
+
+    proc.stdout.setEncoding('utf8')
+    proc.stdout.on('data', (chunk: string) => {
+      if (chunk.includes('MCP Streamable HTTP Server listening')) {
+        res(proc)
+      }
+    })
+
+    proc.on('error', rej)
+  })
+}
+
+test.before(async () => {
+  serverProc = await spawnMcpServer()
+})
+
+test.after(() => serverProc?.kill('SIGINT'))
+
+test('streamableHttpToStdio listTools and callTool', async () => {
+  const gatewayCmd = ['npm', 'run', 'start', '--', '--streamableHttp', MCP_URL]
+
+  const transport = new StdioClientTransport({
+    command: gatewayCmd[0],
+    args: gatewayCmd.slice(1),
+  })
+
+  const client = new Client({ name: 'gateway-test', version: '1.0.0' })
+  await client.connect(transport)
+
+  const { tools } = await client.listTools()
+  assert.ok(tools.some((t) => t.name === 'add'))
+
+  type Reply = { content: Array<{ text: string }> }
+  const reply = (await client.callTool({
+    name: 'add',
+    arguments: { a: 2, b: 3 },
+  })) as Reply
+
+  assert.strictEqual(reply.content[0].text, 'The sum of 2 and 3 is 5.')
+  await client.close()
+})

package/tsconfig.build.json

@@ -0,0 +1,8 @@
+{
+  "extends": "./tsconfig.json",
+  "compilerOptions": {
+    "rootDir": "src",
+    "outDir": "dist"
+  },
+  "include": ["src/**/*"]
+}

package/tsconfig.json

@@ -0,0 +1,12 @@
+{
+  "compilerOptions": {
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "target": "ES2022",
+    "strict": true,
+    "esModuleInterop": true,
+    "forceConsistentCasingInFileNames": true,
+    "outDir": "dist"
+  },
+  "include": ["src/**/*", "tests/**/*"]
+}

package/tsconfig.test.json

@@ -0,0 +1,10 @@
+{
+  "extends": "./tsconfig.json",
+  "compilerOptions": {
+    "rootDir": "tests",
+    "outDir": "dist-tests",
+    "types": ["node"],
+    "noEmitOnError": false
+  },
+  "include": ["tests/**/*.ts"]
+}