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();
 });