test: restore coverage for stdio invalid UTF-8 regression

Author: raashish1601

tests/issues/test_2328_stdio_invalid_utf8.py

@@ -2,6 +2,7 @@
 
 import io
 from io import TextIOWrapper
+from typing import cast
 
 import anyio
 import pytest
@@ -43,6 +44,16 @@ async def handle_call_tool(ctx: ServerRequestContext, params: types.CallToolRequ
         url_adapter.validate_python(arguments["url"])
         return types.CallToolResult(content=[types.TextContent(type="text", text="ok")])
 
+    ctx = cast(ServerRequestContext, None)
+    list_tools_result = await handle_list_tools(ctx, None)
+    assert list_tools_result.tools[0].name == "fetch"
+
+    valid_tool_call_result = await handle_call_tool(
+        ctx,
+        types.CallToolRequestParams(name="fetch", arguments={"url": "https://example.com"}),
+    )
+    assert valid_tool_call_result.content == [types.TextContent(type="text", text="ok")]
+
     server = Server("test-server", on_list_tools=handle_list_tools, on_call_tool=handle_call_tool)
 
     raw_stdin = io.BytesIO(
@@ -61,8 +72,7 @@ async def handle_call_tool(ctx: ServerRequestContext, params: types.CallToolRequ
 
     stdout.flush()
     responses = [
-        jsonrpc_message_adapter.validate_json(line)
-        for line in raw_stdout.getvalue().decode("utf-8").splitlines()
+        jsonrpc_message_adapter.validate_json(line) for line in raw_stdout.getvalue().decode("utf-8").splitlines()
     ]
 
     assert len(responses) == 2
@@ -86,6 +96,8 @@ async def fetch(url: str) -> str:
         await anyio.sleep(0.1)
         return str(TypeAdapter(AnyHttpUrl).validate_python(url))
 
+    assert await fetch("https://example.com") == "https://example.com/"
+
     raw_stdin = io.BytesIO(
         b'{"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{},"clientInfo":{"name":"test","version":"1.0"}}}\n'
         b'{"jsonrpc":"2.0","method":"notifications/initialized"}\n'
@@ -107,8 +119,7 @@ async def fetch(url: str) -> str:
 
     stdout.flush()
     responses = [
-        jsonrpc_message_adapter.validate_json(line)
-        for line in raw_stdout.getvalue().decode("utf-8").splitlines()
+        jsonrpc_message_adapter.validate_json(line) for line in raw_stdout.getvalue().decode("utf-8").splitlines()
     ]
 
     assert responses