feat (#77): correct AppWrapper types

Author: Lancetnik

pyproject.toml

@@ -81,8 +81,7 @@ test = [
 
 dev = [
     "taskiq-faststream[test]",
-
-    "mypy>=1.8.0,<1.16.0",
+    "mypy==1.11.2",
     "ruff==0.11.8",
     "pre-commit >=3.6.0,<5.0.0",
 ]

taskiq_faststream/broker.py

@@ -3,7 +3,7 @@
 from typing import Any
 
 import anyio
-from faststream.app import FastStream
+from faststream._internal.application import Application
 from faststream.types import SendableMessage
 from taskiq import AsyncBroker
 from taskiq.acks import AckableMessage
@@ -66,7 +66,7 @@ async def listen(
             yield b""
             await anyio.sleep(60)
 
-    def task(
+    def task(  # type: ignore[override]
         self,
         message: typing.Union[
             None,
@@ -106,7 +106,7 @@ class AppWrapper(BrokerWrapper):
         task : Register FastStream scheduled task.
     """
 
-    def __init__(self, app: FastStream) -> None:
+    def __init__(self, app: Application) -> None:
         super(BrokerWrapper, self).__init__()
         self.formatter = PatchedFormatter()
         self.app = app

tests/test_kafka.py

@@ -23,3 +23,10 @@ class TestApp(TestBroker):
     def build_taskiq_broker(broker: KafkaBroker) -> AsyncBroker:
         """Build AppWrapper."""
         return AppWrapper(FastStream(broker))
+
+
+class TestAsgiApp(TestBroker):
+    @staticmethod
+    def build_taskiq_broker(broker: KafkaBroker) -> AsyncBroker:
+        """Build AppWrapper."""
+        return AppWrapper(FastStream(broker).as_asgi())

tests/test_nats.py

@@ -23,3 +23,10 @@ class TestApp(TestBroker):
     def build_taskiq_broker(broker: NatsBroker) -> AsyncBroker:
         """Build AppWrapper."""
         return AppWrapper(FastStream(broker))
+
+
+class TestAsgiApp(TestBroker):
+    @staticmethod
+    def build_taskiq_broker(broker: NatsBroker) -> AsyncBroker:
+        """Build AppWrapper."""
+        return AppWrapper(FastStream(broker).as_asgi())

tests/test_rabbit.py

@@ -23,3 +23,10 @@ class TestApp(TestBroker):
     def build_taskiq_broker(broker: RabbitBroker) -> AsyncBroker:
         """Build AppWrapper."""
         return AppWrapper(FastStream(broker))
+
+
+class TestAsgiApp(TestBroker):
+    @staticmethod
+    def build_taskiq_broker(broker: RabbitBroker) -> AsyncBroker:
+        """Build AppWrapper."""
+        return AppWrapper(FastStream(broker).as_asgi())

tests/test_redis.py

@@ -23,3 +23,10 @@ class TestApp(TestBroker):
     def build_taskiq_broker(broker: RedisBroker) -> AsyncBroker:
         """Build AppWrapper."""
         return AppWrapper(FastStream(broker))
+
+
+class TestAsgiApp(TestBroker):
+    @staticmethod
+    def build_taskiq_broker(broker: RedisBroker) -> AsyncBroker:
+        """Build AppWrapper."""
+        return AppWrapper(FastStream(broker).as_asgi())