try to check notification event received

Author: davemssavage

tests/shared/test_session.py

@@ -50,6 +50,7 @@ async def test_request_cancellation():
 
     ev_tool_called = anyio.Event()
     ev_cancelled = anyio.Event()
+    ev_cancel_notified = anyio.Event()
     request_id = None
 
     # Start the request in a separate task so we can cancel it
@@ -66,6 +67,11 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list:
                 await anyio.sleep(10)  # Long enough to ensure we can cancel
                 return []
             raise ValueError(f"Unknown tool: {name}")
+        
+        @server.cancel_notification()
+        async def handle_cancel(requestId: str | int, reason: str | None):
+            nonlocal ev_cancel_notified
+            ev_cancel_notified.set()
 
         # Register the tool so it shows up in list_tools
         @server.list_tools()
@@ -121,6 +127,9 @@ async def make_request(client_session):
                 )
             )
 
+            with anyio.fail_after(1):
+                await ev_cancel_notified.wait()
+
             # Give cancellation time to process
             with anyio.fail_after(1):
                 await ev_cancelled.wait()