Fix test to use real asyncio tasks instead of mocks

Changed test_exception_during_guardrail_processing to use real
asyncio.Task objects instead of Mock objects. This allows the
production code to remain simple and clean, without needing
isinstance checks to handle test-specific mocks.

The test now creates actual tasks using asyncio.create_task() which
better reflects real-world usage and naturally works with the cleanup
logic that uses asyncio.gather().

Author: gn00295120

tests/realtime/test_guardrail_cleanup.py

@@ -165,9 +165,7 @@ async def failing_guardrail_func(context, agent, output):
             # Simulate an error during cleanup
             raise RuntimeError("Cleanup error") from e
 
-    guardrail = OutputGuardrail(
-        guardrail_function=failing_guardrail_func, name="failing_guardrail"
-    )
+    guardrail = OutputGuardrail(guardrail_function=failing_guardrail_func, name="failing_guardrail")
 
     run_config: RealtimeRunConfig = {
         "output_guardrails": [guardrail],

tests/test_session_exceptions.py

@@ -249,16 +249,21 @@ async def test_exception_during_guardrail_processing(
 
         session = RealtimeSession(fake_model, fake_agent, None)
 
-        # Add some fake guardrail tasks
-        fake_task1 = Mock()
-        fake_task1.done.return_value = False
-        fake_task1.cancel = Mock()
+        # Create real async tasks for testing cleanup
+        async def long_running_task():
+            await asyncio.sleep(10)
 
-        fake_task2 = Mock()
-        fake_task2.done.return_value = True
-        fake_task2.cancel = Mock()
+        async def completed_task():
+            pass
 
-        session._guardrail_tasks = {fake_task1, fake_task2}
+        # Create tasks
+        task1 = asyncio.create_task(long_running_task())
+        task2 = asyncio.create_task(completed_task())
+
+        # Wait for task2 to complete
+        await task2
+
+        session._guardrail_tasks = {task1, task2}
 
         fake_model.set_next_events([exception_event])
 
@@ -268,8 +273,8 @@ async def test_exception_during_guardrail_processing(
                     pass
 
         # Verify guardrail tasks were properly cleaned up
-        fake_task1.cancel.assert_called_once()
-        fake_task2.cancel.assert_not_called()  # Already done
+        assert task1.cancelled()  # Should be cancelled
+        assert task2.done()  # Was already done
         assert len(session._guardrail_tasks) == 0
 
     @pytest.mark.asyncio