fix: ensure proper stream cleanup and formatting

- Add try/finally to test_proxy_bidirectional_forwarding for guaranteed cleanup

- Fix missing newlines in example files (README.md, pyproject.toml)

Author: dgenio

examples/servers/simple-proxy/README.md

@@ -124,5 +124,3 @@ The proxy creates two transport connections:
 - **Server-facing**: Communicates with the backend MCP server via stdio_client
 
 All messages are forwarded bidirectionally through the `mcp_proxy()` function.
-
-

examples/servers/simple-proxy/pyproject.toml

@@ -45,5 +45,3 @@ target-version = "py310"
 
 [dependency-groups]
 dev = ["pyright>=1.1.378", "pytest>=8.3.3", "ruff>=0.6.9"]
-
-

tests/shared/test_proxy.py

@@ -102,33 +102,34 @@ async def test_proxy_bidirectional_forwarding(create_streams):
         ),
     ) = create_streams()
 
-    # Create test messages
-    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))
-
-    async with mcp_proxy(client_streams, server_streams):
-        # Send messages from both sides
-        await client_read_writer.send(client_msg)
-        await server_read_writer.send(server_msg)
-
-        # Verify both arrive at their destinations
-        with anyio.fail_after(1):
-            # Client message should arrive at server
-            received_at_server = await server_write_reader.receive()
-            assert received_at_server.message.root.id == "client_1"
-
-            # Server message should arrive at client
-            received_at_client = await client_write_reader.receive()
-            assert received_at_client.message.root.id == "server_1"
-
-    # Clean up test streams
-    await client_read_writer.aclose()
-    await client_write_reader.aclose()
-    await server_read_writer.aclose()
-    await server_write_reader.aclose()
+    try:
+        # Create test messages
+        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))
+
+        async with mcp_proxy(client_streams, server_streams):
+            # Send messages from both sides
+            await client_read_writer.send(client_msg)
+            await server_read_writer.send(server_msg)
+
+            # Verify both arrive at their destinations
+            with anyio.fail_after(1):
+                # Client message should arrive at server
+                received_at_server = await server_write_reader.receive()
+                assert received_at_server.message.root.id == "client_1"
+
+                # Server message should arrive at client
+                received_at_client = await client_write_reader.receive()
+                assert received_at_client.message.root.id == "server_1"
+    finally:
+        # Clean up test streams
+        await client_read_writer.aclose()
+        await client_write_reader.aclose()
+        await server_read_writer.aclose()
+        await server_write_reader.aclose()
 
 
 @pytest.mark.anyio