| 1 | import { assert, describe, it } from "@effect/vitest" |
| 2 | import * as McpProtocol from "effect/ai/McpProtocol" |
| 3 | import * as McpSchema from "effect/ai/McpSchema" |
| 4 | import * as Effect from "effect/Effect" |
| 5 | import * as BaseProtocolTest from "./McpConformance/BaseProtocolTest.ts" |
| 6 | import * as CompletionTest from "./McpConformance/CompletionTest.ts" |
| 7 | import * as ElicitationTest from "./McpConformance/ElicitationTest.ts" |
| 8 | import * as LifecycleTest from "./McpConformance/LifecycleTest.ts" |
| 9 | import * as LoggingTest from "./McpConformance/LoggingTest.ts" |
| 10 | import { layer as makeMcpConformanceLayer, McpConformance } from "./McpConformance/McpConformance.ts" |
| 11 | import * as PromptsTest from "./McpConformance/PromptsTest.ts" |
| 12 | import * as ResourcesTest from "./McpConformance/ResourcesTest.ts" |
| 13 | import * as RootsTest from "./McpConformance/RootsTest.ts" |
| 14 | import * as SamplingTest from "./McpConformance/SamplingTest.ts" |
| 15 | import * as ToolsTest from "./McpConformance/ToolsTest.ts" |
| 16 | import * as TransportsTest from "./McpConformance/TransportsTest.ts" |
| 17 | import * as UtilitiesTest from "./McpConformance/UtilitiesTest.ts" |
| 18 | |
| 19 | const protocol = McpProtocol.v2025_11_25 |
| 20 | const testLayer = makeMcpConformanceLayer(protocol) |
| 21 | |
| 22 | LifecycleTest.suite(protocol, testLayer) |
| 23 | BaseProtocolTest.suite(protocol, testLayer) |
| 24 | BaseProtocolTest.statefulLegacySuite(protocol, testLayer) |
| 25 | TransportsTest.suite(protocol, testLayer) |
| 26 | UtilitiesTest.suite(protocol, testLayer) |
| 27 | UtilitiesTest.statefulLegacySuite(protocol, testLayer) |
| 28 | ToolsTest.suite(protocol, testLayer) |
| 29 | ToolsTest.statefulLegacySuite(protocol, testLayer) |
| 30 | ResourcesTest.suite(protocol, testLayer) |
| 31 | ResourcesTest.statefulLegacySuite(protocol, testLayer) |
| 32 | PromptsTest.suite(protocol, testLayer) |
| 33 | PromptsTest.statefulLegacySuite(protocol, testLayer) |
| 34 | CompletionTest.suite(protocol, testLayer) |
| 35 | LoggingTest.suite(protocol, testLayer) |
| 36 | RootsTest.suite(protocol, testLayer) |
| 37 | SamplingTest.suite(protocol, testLayer) |
| 38 | ElicitationTest.suite(protocol, testLayer) |
| 39 | |
| 40 | it.layer(testLayer)(`Mcp Conformance (${protocol.protocolVersion})`, (it) => { |
| 41 | describe("Tools", () => { |
| 42 | it.effect("does not expose experimental Tasks without an extension", () => |
| 43 | Effect.gen(function*() { |
| 44 | const test = yield* McpConformance |
| 45 | const initialized = yield* test.initialize({ server: "features" }) |
| 46 | yield* test.notifyInitialized(initialized) |
| 47 | const before = (yield* test.observations).toolInvocations |
| 48 | const response = yield* test.send(initialized, { |
| 49 | jsonrpc: "2.0", |
| 50 | id: 2, |
| 51 | method: "tasks/get", |
| 52 | params: { taskId: "task-1" } |
| 53 | }) |
| 54 | const error = yield* test.decodeError(response) |
| 55 | |
| 56 | assert.strictEqual(error.error.code, McpSchema.METHOD_NOT_FOUND_ERROR_CODE) |
| 57 | assert.strictEqual((yield* test.observations).toolInvocations, before) |
| 58 | })) |
| 59 | }) |
| 60 | }) |