Fix execute_many tests for async response timing

Co-authored-by: P4x-ng <P4X-ng@users.noreply.github.com>

Author: cursoragent

test/test_connection.py

@@ -216,18 +216,21 @@ async def test_execute_many_success():
     """Test execute_many returns ordered command results."""
     mock_ws = MockWebSocket()
 
-    # Queue responses in reverse order to verify request multiplexing.
-    mock_ws.queue_message({"id": 2, "result": {"result": {"type": "number", "value": 2}}})
-    mock_ws.queue_message({"id": 1, "result": {"result": {"type": "number", "value": 1}}})
-
     with patch('cdp.connection.websockets.connect', new_callable=AsyncMock) as mock_connect:
         mock_connect.return_value = mock_ws
 
         async with CDPConnection("ws://localhost:9222/test") as conn:
-            results = await conn.execute_many([
+            task = asyncio.create_task(conn.execute_many([
                 runtime.evaluate(expression="1"),
                 runtime.evaluate(expression="2"),
-            ])
+            ]))
+            await asyncio.sleep(0.05)
+
+            # Queue responses in reverse order to verify request multiplexing.
+            mock_ws.queue_message({"id": 2, "result": {"result": {"type": "number", "value": 2}}})
+            mock_ws.queue_message({"id": 1, "result": {"result": {"type": "number", "value": 1}}})
+
+            results = await task
 
             assert len(results) == 2
             assert isinstance(results[0][0], runtime.RemoteObject)
@@ -242,21 +245,24 @@ async def test_execute_many_return_exceptions():
     """Test execute_many can collect exceptions instead of raising."""
     mock_ws = MockWebSocket()
 
-    # First command succeeds, second command fails.
-    mock_ws.queue_message({"id": 1, "result": {"result": {"type": "number", "value": 1}}})
-    mock_ws.queue_message({"id": 2, "error": {"code": -32602, "message": "Invalid params"}})
-
     with patch('cdp.connection.websockets.connect', new_callable=AsyncMock) as mock_connect:
         mock_connect.return_value = mock_ws
 
         async with CDPConnection("ws://localhost:9222/test") as conn:
-            results = await conn.execute_many(
+            task = asyncio.create_task(conn.execute_many(
                 [
                     runtime.evaluate(expression="1"),
                     page.navigate(url="https://example.com"),
                 ],
                 return_exceptions=True,
-            )
+            ))
+            await asyncio.sleep(0.05)
+
+            # First command succeeds, second command fails.
+            mock_ws.queue_message({"id": 1, "result": {"result": {"type": "number", "value": 1}}})
+            mock_ws.queue_message({"id": 2, "error": {"code": -32602, "message": "Invalid params"}})
+
+            results = await task
 
             assert len(results) == 2
             assert isinstance(results[0][0], runtime.RemoteObject)
@@ -269,14 +275,18 @@ async def test_execute_many_return_exceptions():
 async def test_execute_many_raises_first_exception_by_default():
     """Test execute_many raises command errors by default."""
     mock_ws = MockWebSocket()
-    mock_ws.queue_message({"id": 1, "error": {"code": -32000, "message": "Command failed"}})
 
     with patch('cdp.connection.websockets.connect', new_callable=AsyncMock) as mock_connect:
         mock_connect.return_value = mock_ws
 
         async with CDPConnection("ws://localhost:9222/test") as conn:
             with pytest.raises(CDPCommandError) as exc_info:
-                await conn.execute_many([page.navigate(url="https://example.com")])
+                task = asyncio.create_task(
+                    conn.execute_many([page.navigate(url="https://example.com")])
+                )
+                await asyncio.sleep(0.05)
+                mock_ws.queue_message({"id": 1, "error": {"code": -32000, "message": "Command failed"}})
+                await task
             assert exc_info.value.code == -32000