🎨 Add explicit async methods for fastapi app lifespan

docs/index.md

@@ -5,7 +5,7 @@
 MQTT is a lightweight publish/subscribe messaging protocol designed for M2M (machine to machine) telemetry in low bandwidth environments.
 Fastapi-mqtt is the client for working with MQTT.
 
-For more information about MQQT, please refer to here: [MQTT](mqtt.md)
+For more information about MQTT, please refer to here: [MQTT](mqtt.md)
 
 Fatapi-mqtt wraps around [gmqtt](https://github.com/wialon/gmqtt) module. Gmqtt Python async client for MQTT client implementation.
 The module has the support of MQTT version 5.0 protocol

fastapi_mqtt/fastmqtt.py

@@ -1,6 +1,5 @@
 import asyncio
 import uuid
-from functools import partial
 from itertools import zip_longest
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
@@ -213,21 +212,30 @@ def unsubscribe(self, topic: str, **kwargs):
 
         topic: topic name
         """
-        partial(self.client.unsubscribe, topic, **kwargs)
         log_info.debug("unsubscribe")
         if topic in self.subscriptions:
             del self.subscriptions[topic]
 
         return self.client.unsubscribe(topic, **kwargs)
 
+    async def mqtt_startup(self):
+        """Initial connection for MQTT client, for lifespan startup."""
+        await self.connection()
+
+    async def mqtt_shutdown(self):
+        """Final disconnection for MQTT client, for lifespan shutdown."""
+        await self.client.disconnect()
+
     def init_app(self, app: FastAPI) -> None:  # pragma: no cover
+        """Add startup and shutdown event handlers for app without lifespan."""
+
         @app.on_event("startup")
         async def startup():
-            await self.connection()
+            await self.mqtt_startup()
 
         @app.on_event("shutdown")
         async def shutdown():
-            await self.client.disconnect()
+            await self.mqtt_shutdown()
 
     def subscribe(
         self,

tests/conftest.py

@@ -33,10 +33,10 @@ def test_app():
 
     @asynccontextmanager
     async def _lifespan(application: FastAPI):
-        await fast_mqtt.connection()
+        await fast_mqtt.mqtt_startup()
         logging.info("connection done, starting fastapi app now")
         yield
-        await fast_mqtt.client.disconnect()
+        await fast_mqtt.mqtt_shutdown()
 
     app = FastAPI(lifespan=_lifespan)