Finish 0.2.3

Author: chandr-andr

README.md

@@ -20,8 +20,7 @@ Let's see the example with the redis broker and redis async result:
 ```python
 import asyncio
 
-from taskiq_redis.redis_broker import ListQueueBroker
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
+from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend
 
 redis_async_result = RedisAsyncResultBackend(
     redis_url="redis://localhost:6379",

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "taskiq-redis"
-version = "0.2.2"
+version = "0.2.3"
 description = "Redis integration for taskiq"
 authors = ["taskiq-team <[email protected]>"]
 readme = "README.md"

taskiq_redis/__init__.py

@@ -1,5 +1,9 @@
 """Package for redis integration."""
 from taskiq_redis.redis_backend import RedisAsyncResultBackend
-from taskiq_redis.redis_broker import ListQueueBroker
+from taskiq_redis.redis_broker import ListQueueBroker, PubSubBroker
 
-__all__ = ["RedisAsyncResultBackend", "ListQueueBroker"]
+__all__ = [
+    "RedisAsyncResultBackend",
+    "ListQueueBroker",
+    "PubSubBroker",
+]

taskiq_redis/redis_broker.py

@@ -49,6 +49,7 @@ def __init__(
 
     async def shutdown(self) -> None:
         """Closes redis connection pool."""
+        await super().shutdown()
         await self.connection_pool.disconnect()
 
     async def listen(self) -> AsyncGenerator[BrokerMessage, None]:

tests/test_backend.py

@@ -5,11 +5,11 @@
 import pytest
 from taskiq import TaskiqResult
 
+from taskiq_redis import RedisAsyncResultBackend
 from taskiq_redis.exceptions import (
     DuplicateExpireTimeSelectedError,
     ExpireTimeMustBeMoreThanZeroError,
 )
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
 
 _ReturnType = TypeVar("_ReturnType")
 

tests/test_broker.py

@@ -4,7 +4,7 @@
 import pytest
 from taskiq import AsyncBroker, BrokerMessage
 
-from taskiq_redis.redis_broker import ListQueueBroker, PubSubBroker
+from taskiq_redis import ListQueueBroker, PubSubBroker
 
 
 async def get_message(broker: AsyncBroker) -> BrokerMessage:  # type: ignore

tests/test_result_backend.py

@@ -3,7 +3,7 @@
 import pytest
 from taskiq import TaskiqResult
 
-from taskiq_redis.redis_backend import RedisAsyncResultBackend
+from taskiq_redis import RedisAsyncResultBackend
 
 
 @pytest.mark.anyio