Initial import

tests/client/test_stdio.py

@@ -0,0 +1,38 @@
+import pytest
+
+from mcp_python.client.stdio import StdioServerParameters, stdio_client
+from mcp_python.types import JSONRPCMessage, JSONRPCRequest, JSONRPCResponse
+
+
+@pytest.mark.anyio
+async def test_stdio_client():
+    server_parameters = StdioServerParameters(command="/usr/bin/tee")
+
+    async with stdio_client(server_parameters) as (read_stream, write_stream):
+        # Test sending and receiving messages
+        messages = [
+            JSONRPCMessage(root=JSONRPCRequest(jsonrpc="2.0", id=1, method="ping")),
+            JSONRPCMessage(root=JSONRPCResponse(jsonrpc="2.0", id=2, result={})),
+        ]
+
+        async with write_stream:
+            for message in messages:
+                await write_stream.send(message)
+
+        read_messages = []
+        async with read_stream:
+            async for message in read_stream:
+                if isinstance(message, Exception):
+                    raise message
+
+                read_messages.append(message)
+                if len(read_messages) == 2:
+                    break
+
+        assert len(read_messages) == 2
+        assert read_messages[0] == JSONRPCMessage(
+            root=JSONRPCRequest(jsonrpc="2.0", id=1, method="ping")
+        )
+        assert read_messages[1] == JSONRPCMessage(
+            root=JSONRPCResponse(jsonrpc="2.0", id=2, result={})
+        )