Merge pull request #1 from ic-dev21/test

Fix pour dvd-dev/hilo#282
dvd-dev · Oct 1, 2023 · cdbf950 · cdbf950
2 parents 838db0a + 19e13af
commit cdbf950
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhilo/websocket.py b/pyhilo/websocket.py
@@ -298,7 +298,7 @@ async def _clean_queue(self) -> None:
 
     async def async_disconnect(self) -> None:
         """Disconnect from the websocket server."""
-        self._clean_queue()
+        await self._clean_queue()
         if not self.connected:
             return
 
@@ -323,7 +323,7 @@ async def async_listen(self) -> None:
         finally:
             LOG.info("Websocket: Listen completed; cleaning up")
             self._watchdog.cancel()
-            self._clean_queue()
+            await self._clean_queue()
 
             for callback in self._disconnect_callbacks:
                 schedule_callback(callback)