fix: add missing jsonrpc field to JSONRPCRequest in tests

All JSONRPCRequest instances must include the jsonrpc='2.0' field
as required by the Pydantic model validation.

Author: dgenio

tests/shared/test_proxy.py

@@ -36,7 +36,7 @@ async def test_proxy_forwards_client_to_server(create_streams):
     client_streams, server_streams, (client_read_writer, _), (_, server_write_reader) = create_streams()
 
     # Create a test message
-    request = JSONRPCRequest(id="1", method="test_method", params={"key": "value"})
+    request = JSONRPCRequest(jsonrpc="2.0", id="1", method="test_method", params={"key": "value"})
     message = SessionMessage(JSONRPCMessage(request))
 
     async with mcp_proxy(client_streams, server_streams):
@@ -56,7 +56,7 @@ async def test_proxy_forwards_server_to_client(create_streams):
     client_streams, server_streams, (_, client_write_reader), (server_read_writer, _) = create_streams()
 
     # Create a test message
-    request = JSONRPCRequest(id="2", method="server_method", params={"data": "test"})
+    request = JSONRPCRequest(jsonrpc="2.0", id="2", method="server_method", params={"data": "test"})
     message = SessionMessage(JSONRPCMessage(request))
 
     async with mcp_proxy(client_streams, server_streams):
@@ -84,8 +84,8 @@ async def test_proxy_bidirectional_forwarding(create_streams):
     ) = create_streams()
 
     # Create test messages
-    client_request = JSONRPCRequest(id="client_1", method="client_method", params={})
-    server_request = JSONRPCRequest(id="server_1", method="server_method", params={})
+    client_request = JSONRPCRequest(jsonrpc="2.0", id="client_1", method="client_method", params={})
+    server_request = JSONRPCRequest(jsonrpc="2.0", id="server_1", method="server_method", params={})
 
     client_msg = SessionMessage(JSONRPCMessage(client_request))
     server_msg = SessionMessage(JSONRPCMessage(server_request))
@@ -122,7 +122,7 @@ async def transform_client_message(msg: SessionMessage) -> SessionMessage | None
             msg.message.root.id = f"transformed_{original_id}"
         return msg
 
-    request = JSONRPCRequest(id="original", method="test", params={})
+    request = JSONRPCRequest(jsonrpc="2.0", id="original", method="test", params={})
     message = SessionMessage(JSONRPCMessage(request))
 
     async with mcp_proxy(client_streams, server_streams, on_client_message=transform_client_message):
@@ -143,7 +143,7 @@ async def drop_message(msg: SessionMessage) -> SessionMessage | None:
         """Drop all messages."""
         return None
 
-    request = JSONRPCRequest(id="dropped", method="test", params={})
+    request = JSONRPCRequest(jsonrpc="2.0", id="dropped", method="test", params={})
     message = SessionMessage(JSONRPCMessage(request))
 
     async with mcp_proxy(client_streams, server_streams, on_client_message=drop_message):
@@ -197,7 +197,7 @@ def error_handler(error: Exception) -> None:
         await client_read_writer.send(ValueError("Error 1"))
 
         # Send a valid message
-        request = JSONRPCRequest(id="after_error", method="test", params={})
+        request = JSONRPCRequest(jsonrpc="2.0", id="after_error", method="test", params={})
         message = SessionMessage(JSONRPCMessage(request))
         await client_read_writer.send(message)
 
@@ -224,7 +224,7 @@ async def failing_transform(msg: SessionMessage) -> SessionMessage | None:
         """Transform that always fails."""
         raise RuntimeError("Transform failed")
 
-    request = JSONRPCRequest(id="test", method="test", params={})
+    request = JSONRPCRequest(jsonrpc="2.0", id="test", method="test", params={})
     message = SessionMessage(JSONRPCMessage(request))
 
     async with mcp_proxy(client_streams, server_streams, onerror=error_handler, on_client_message=failing_transform):
@@ -262,13 +262,17 @@ async def test_proxy_cleans_up_streams(create_streams):
     # All streams should be closed
     # Attempting to send/receive should fail
     with pytest.raises(anyio.ClosedResourceError):
-        await client_read_writer.send(SessionMessage(JSONRPCMessage(JSONRPCRequest(id="1", method="test", params={}))))
+        await client_read_writer.send(
+            SessionMessage(JSONRPCMessage(JSONRPCRequest(jsonrpc="2.0", id="1", method="test", params={})))
+        )
 
     with pytest.raises((anyio.ClosedResourceError, anyio.EndOfStream)):
         await client_write_reader.receive()
 
     with pytest.raises(anyio.ClosedResourceError):
-        await server_read_writer.send(SessionMessage(JSONRPCMessage(JSONRPCRequest(id="2", method="test", params={}))))
+        await server_read_writer.send(
+            SessionMessage(JSONRPCMessage(JSONRPCRequest(jsonrpc="2.0", id="2", method="test", params={})))
+        )
 
     with pytest.raises((anyio.ClosedResourceError, anyio.EndOfStream)):
         await server_write_reader.receive()
@@ -282,7 +286,7 @@ async def test_proxy_multiple_messages(create_streams):
     async with mcp_proxy(client_streams, server_streams):
         # Send multiple messages
         for i in range(5):
-            request = JSONRPCRequest(id=str(i), method=f"method_{i}", params={})
+            request = JSONRPCRequest(jsonrpc="2.0", id=str(i), method=f"method_{i}", params={})
             message = SessionMessage(JSONRPCMessage(request))
             await client_read_writer.send(message)