raw request propagation in tools - implementation, unit tests, types

Author: KKonstantinov

package-lock.json

@@ -1,6 +1,6 @@
 {
  "name": "@modelcontextprotocol/sdk",
- "version": "1.12.0",
+ "version": "1.12.2",
  "description": "Model Context Protocol implementation for TypeScript",
  "license": "MIT",
  "author": "Anthropic, PBC (https://anthropic.com)",

package.json

@@ -20,7 +20,14 @@ import {
 import { Transport } from "../shared/transport.js";
 import { Server } from "../server/index.js";
 import { InMemoryTransport } from "../inMemory.js";
-
+import { RequestInfo } from "../server/types/types.js";
+
+const mockRequestInfo: RequestInfo = {
+ headers: {
+ 'content-type': 'application/json',
+ 'accept': 'application/json',
+ },
+};
 /***
  * Test: Initialize with Matching Protocol Version
  */
@@ -42,7 +49,7 @@ test("should initialize with matching protocol version", async () => {
  },
  instructions: "test instructions",
  },
- });
+ }, { requestInfo: mockRequestInfo });
  }
  return Promise.resolve();
  }),
@@ -100,7 +107,7 @@ test("should initialize with supported older protocol version", async () => {
  version: "1.0",
  },
  },
- });
+ }, { requestInfo: mockRequestInfo });
  }
  return Promise.resolve();
  }),
@@ -150,7 +157,7 @@ test("should reject unsupported protocol version", async () => {
  version: "1.0",
  },
  },
- });
+ }, { requestInfo: mockRequestInfo });
  }
  return Promise.resolve();
  }),

src/client/index.test.ts

@@ -19,6 +19,14 @@ import {
 import { Transport } from "../shared/transport.js";
 import { InMemoryTransport } from "../inMemory.js";
 import { Client } from "../client/index.js";
+import { RequestInfo } from "./types/types.js";
+
+const mockRequestInfo: RequestInfo = {
+ headers: {
+ 'content-type': 'application/json',
+ 'traceparent': '00-4bf92f3577b34da6a3ce929d0e0e4736-00f067aa0ba902b7-01',
+ },
+};
 
 test("should accept latest protocol version", async () => {
  let sendPromiseResolve: (value: unknown) => void;
@@ -77,7 +85,7 @@ test("should accept latest protocol version", async () => {
  version: "1.0",
  },
  },
- });
+ }, { requestInfo: mockRequestInfo });
 
  await expect(sendPromise).resolves.toBeUndefined();
 });
@@ -138,7 +146,7 @@ test("should accept supported older protocol version", async () => {
  version: "1.0",
  },
  },
- });
+ }, { requestInfo: mockRequestInfo });
 
  await expect(sendPromise).resolves.toBeUndefined();
 });
@@ -198,7 +206,7 @@ test("should handle unsupported protocol version", async () => {
  version: "1.0",
  },
  },
- });
+ }, { requestInfo: mockRequestInfo });
 
  await expect(sendPromise).resolves.toBeUndefined();
 });