fixup! refactor: update test files to use ParsedMessage

modelcontextprotocol · Mar 3, 2025 · 2e3388b · 2e3388b
1 parent 05fc4f7
commit 2e3388b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 74 deletions.
diff --git a/tests/issues/test_201_client_hangs_on_logging.py b/tests/issues/test_201_client_hangs_on_logging.py
diff --git a/tests/server/test_lifespan.py b/tests/server/test_lifespan.py
@@ -121,7 +121,7 @@ async def run_server():
 
         # Get response and verify
         response = await receive_stream2.receive()
-        assert response.root.result["content"][0]["text"] == "true"
+        assert response.root.root.result["content"][0]["text"] == "true"
 
         # Cancel server task
         tg.cancel_scope.cancel()
@@ -220,7 +220,7 @@ async def run_server():
 
         # Get response and verify
         response = await receive_stream2.receive()
-        assert response.root.result["content"][0]["text"] == "true"
+        assert response.root.root.result["content"][0]["text"] == "true"
 
         # Cancel server task
         tg.cancel_scope.cancel()