changing to asyncio.wait_for since feature added in 3.11, don't yet h…

…ave support for

tests/integration/async_push_test_client.py

@@ -252,8 +252,7 @@ async def get_notification(self, timeout=1):
             raise WebSocketException("WebSocket client not available as expected")
 
         try:
-            async with asyncio.timeout(timeout):
-                d = await self.ws.recv()
+            d = await asyncio.wait_for(self.ws.recv(), timeout)
             log.debug(f"Recv: {d!r}")
             return json.loads(d)
         except Exception:
@@ -265,8 +264,7 @@ async def get_broadcast(self, timeout=1):  # pragma: no cover
             raise WebSocketException("WebSocket client not available as expected")
 
         try:
-            async with asyncio.timeout(timeout):
-                d = await self.ws.recv()
+            d = await asyncio.wait_for(self.ws.recv(), timeout)
             log.debug(f"Recv: {d}")
             result = json.loads(d)
             assert result.get("messageType") == ClientMessageType.BROADCAST.value