fix await disconnect (#46)

DeebotUniverse · Dec 10, 2021 · 9118d7b · 9118d7b
1 parent f703479
commit 9118d7b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deebot_client/mqtt_client.py b/deebot_client/mqtt_client.py
@@ -81,7 +81,7 @@ def on_credentials_changed(credentials: Credentials) -> None:
     async def initialize(self) -> None:
         """Initialize MQTT."""
         if self._client:
-            self.disconnect()
+            await self.disconnect()
 
         credentials = await self._authenticator.authenticate()
         client_id = f"{credentials.user_id}@ecouser/{self._config.device_id}"
@@ -113,10 +113,10 @@ def unsubscribe(self, vacuum_bot: VacuumBot) -> None:
             for subscription in _get_subscriptions(device_info):
                 self._client.unsubscribe(subscription.topic)
 
-    def disconnect(self) -> None:
+    async def disconnect(self) -> None:
         """Disconnect from MQTT."""
         if self._client:
-            self._client.disconnect()
+            await self._client.disconnect()
         self._subscribers.clear()
 
     async def _handle_atr(self, topic_split: List[str], payload: bytes) -> None: