mirror of
https://github.com/github/awesome-copilot.git
synced 2026-07-16 19:03:26 +00:00
chore: publish from main
This commit is contained in:
@@ -0,0 +1,58 @@
|
||||
import assert from "node:assert/strict";
|
||||
import { createServer } from "node:http";
|
||||
import test from "node:test";
|
||||
import { probe } from "./test/mcp-probe.mjs";
|
||||
|
||||
test("native HTTP probe carries API key and MCP session through an SSE handshake", async (t) => {
|
||||
const apiKeys = [];
|
||||
const sessionIds = [];
|
||||
const methods = [];
|
||||
const server = createServer(async (req, res) => {
|
||||
const chunks = [];
|
||||
for await (const chunk of req) chunks.push(chunk);
|
||||
const message = JSON.parse(Buffer.concat(chunks).toString("utf8"));
|
||||
apiKeys.push(req.headers["x-api-key"]);
|
||||
sessionIds.push(req.headers["mcp-session-id"] || null);
|
||||
methods.push(message.method);
|
||||
|
||||
if (message.method === "notifications/initialized") {
|
||||
res.writeHead(202);
|
||||
res.end();
|
||||
return;
|
||||
}
|
||||
|
||||
let result;
|
||||
if (message.method === "initialize") {
|
||||
res.setHeader("Mcp-Session-Id", "session-1");
|
||||
result = {
|
||||
protocolVersion: "2025-06-18",
|
||||
serverInfo: { name: "test-server", version: "1.0.0" },
|
||||
capabilities: { tools: {} },
|
||||
};
|
||||
} else if (message.method === "tools/list") {
|
||||
result = { tools: [{ name: "ListTeams", inputSchema: { type: "object" } }] };
|
||||
} else {
|
||||
result = { content: [{ type: "text", text: "ok" }] };
|
||||
}
|
||||
|
||||
res.setHeader("Content-Type", "text/event-stream");
|
||||
res.end(`event: message\ndata: ${JSON.stringify({ jsonrpc: "2.0", id: message.id, result })}\n\n`);
|
||||
});
|
||||
await new Promise((resolve) => server.listen(0, "127.0.0.1", resolve));
|
||||
t.after(() => new Promise((resolve) => server.close(resolve)));
|
||||
|
||||
const { port } = server.address();
|
||||
const result = await probe({
|
||||
apiName: "WorkIQTeams",
|
||||
displayName: "WorkIQ Teams",
|
||||
url: `http://127.0.0.1:${port}/mcp`,
|
||||
key: "secret",
|
||||
});
|
||||
|
||||
assert.equal(result.ok, true);
|
||||
assert.equal(result.toolCount, 1);
|
||||
assert.equal(result.toolCalled, "ListTeams");
|
||||
assert.deepEqual(methods, ["initialize", "notifications/initialized", "tools/list", "tools/call"]);
|
||||
assert.deepEqual(apiKeys, ["secret", "secret", "secret", "secret"]);
|
||||
assert.deepEqual(sessionIds, [null, "session-1", "session-1", "session-1"]);
|
||||
});
|
||||
Reference in New Issue
Block a user