fix: deprecation warning in asyncio.wait (#2383)

sanic-org · Jan 19, 2022 · bb517dd · bb517dd
1 parent b8d9914
commit bb517dd
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/sanic/server/websockets/impl.py b/sanic/server/websockets/impl.py
@@ -518,8 +518,12 @@ async def recv(self, timeout: Optional[float] = None) -> Optional[Data]:
             )
         try:
             self.recv_cancel = asyncio.Future()
+            tasks = (
+                self.recv_cancel,
+                asyncio.ensure_future(self.assembler.get(timeout)),
+            )
             done, pending = await asyncio.wait(
-                (self.recv_cancel, self.assembler.get(timeout)),
+                tasks,
                 return_when=asyncio.FIRST_COMPLETED,
             )
             done_task = next(iter(done))
@@ -570,8 +574,12 @@ async def recv_burst(self, max_recv=256) -> Sequence[Data]:
             self.can_pause = False
             self.recv_cancel = asyncio.Future()
             while True:
+                tasks = (
+                    self.recv_cancel,
+                    asyncio.ensure_future(self.assembler.get(timeout=0)),
+                )
                 done, pending = await asyncio.wait(
-                    (self.recv_cancel, self.assembler.get(timeout=0)),
+                    tasks,
                     return_when=asyncio.FIRST_COMPLETED,
                 )
                 done_task = next(iter(done))