refactor(tests): update annotations

Author: vldmrdev

tests/conftest.py

@@ -1,5 +1,4 @@
 import os
-from typing import List, Tuple
 
 import pytest
 
@@ -44,7 +43,7 @@ def redis_cluster_url() -> str:
 
 
 @pytest.fixture
-def redis_sentinels() -> List[Tuple[str, int]]:
+def redis_sentinels() -> list[tuple[str, int]]:
     """
     List of redis sentinel hosts.
 

tests/test_broker.py

@@ -1,6 +1,5 @@
 import asyncio
 import uuid
-from typing import List, Tuple, Union
 
 import pytest
 from redis.asyncio import Redis
@@ -26,7 +25,7 @@ def test_no_url_should_raise_typeerror() -> None:
 
 async def get_message(
     broker: AsyncBroker,
-) -> Union[bytes, AckableMessage]:
+) -> bytes | AckableMessage:
     """
     Get a message from the broker.
 
@@ -253,7 +252,7 @@ async def test_stream_cluster_broker(
 @pytest.mark.anyio
 async def test_pub_sub_sentinel_broker(
     valid_broker_message: BrokerMessage,
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -284,7 +283,7 @@ async def test_pub_sub_sentinel_broker(
 @pytest.mark.anyio
 async def test_list_queue_sentinel_broker(
     valid_broker_message: BrokerMessage,
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -313,7 +312,7 @@ async def test_list_queue_sentinel_broker(
 @pytest.mark.anyio
 async def test_streams_sentinel_broker(
     valid_broker_message: BrokerMessage,
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -405,7 +404,7 @@ async def test_maxlen_in_cluster_stream_broker(
 @pytest.mark.anyio
 async def test_maxlen_in_sentinel_stream_broker(
     redis_sentinel_master_name: str,
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     valid_broker_message: BrokerMessage,
 ) -> None:
     """

tests/test_result_backend.py

@@ -1,6 +1,5 @@
 import asyncio
 import uuid
-from typing import List, Tuple
 
 import pytest
 from taskiq import TaskiqResult
@@ -25,7 +24,7 @@ async def test_set_result_success(redis_url: str) -> None:
         redis_url=redis_url,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -58,7 +57,7 @@ async def test_fetch_without_logs(redis_url: str) -> None:
         redis_url=redis_url,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -92,7 +91,7 @@ async def test_remove_results_after_reading(redis_url: str) -> None:
         keep_results=False,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -122,7 +121,7 @@ async def test_keep_results_after_reading(redis_url: str) -> None:
         keep_results=True,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -153,7 +152,7 @@ async def test_set_result_max_connections(redis_url: str) -> None:
     )
 
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -182,7 +181,7 @@ async def test_set_result_success_cluster(redis_cluster_url: str) -> None:
         redis_url=redis_cluster_url,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -215,7 +214,7 @@ async def test_fetch_without_logs_cluster(redis_cluster_url: str) -> None:
         redis_url=redis_cluster_url,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -249,7 +248,7 @@ async def test_remove_results_after_reading_cluster(redis_cluster_url: str) -> N
         keep_results=False,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -279,7 +278,7 @@ async def test_keep_results_after_reading_cluster(redis_cluster_url: str) -> Non
         keep_results=True,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -298,7 +297,7 @@ async def test_keep_results_after_reading_cluster(redis_cluster_url: str) -> Non
 
 @pytest.mark.anyio
 async def test_set_result_success_sentinel(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -312,7 +311,7 @@ async def test_set_result_success_sentinel(
         master_name=redis_sentinel_master_name,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -336,7 +335,7 @@ async def test_set_result_success_sentinel(
 
 @pytest.mark.anyio
 async def test_fetch_without_logs_sentinel(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -350,7 +349,7 @@ async def test_fetch_without_logs_sentinel(
         master_name=redis_sentinel_master_name,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -374,7 +373,7 @@ async def test_fetch_without_logs_sentinel(
 
 @pytest.mark.anyio
 async def test_remove_results_after_reading_sentinel(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -389,7 +388,7 @@ async def test_remove_results_after_reading_sentinel(
         keep_results=False,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -409,7 +408,7 @@ async def test_remove_results_after_reading_sentinel(
 
 @pytest.mark.anyio
 async def test_keep_results_after_reading_sentinel(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """
@@ -424,7 +423,7 @@ async def test_keep_results_after_reading_sentinel(
         keep_results=True,
     )
     task_id = uuid.uuid4().hex
-    result: "TaskiqResult[int]" = TaskiqResult(
+    result: TaskiqResult[int] = TaskiqResult(
         is_err=True,
         log="My Log",
         return_value=11,
@@ -521,7 +520,7 @@ async def test_set_progress_cluster(redis_cluster_url: str) -> None:
 
 @pytest.mark.anyio
 async def test_set_progress_sentinel(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     """

tests/test_schedule_source.py

@@ -1,7 +1,6 @@
 import asyncio
 import datetime as dt
 import uuid
-from typing import List, Tuple
 
 import pytest
 from taskiq import ScheduledTask
@@ -243,7 +242,7 @@ async def test_cluster_get_schedules(redis_cluster_url: str) -> None:
 
 @pytest.mark.anyio
 async def test_sentinel_set_schedule(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     prefix = uuid.uuid4().hex
@@ -267,7 +266,7 @@ async def test_sentinel_set_schedule(
 
 @pytest.mark.anyio
 async def test_sentinel_delete_schedule(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     prefix = uuid.uuid4().hex
@@ -295,7 +294,7 @@ async def test_sentinel_delete_schedule(
 
 @pytest.mark.anyio
 async def test_sentinel_post_run_cron(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     prefix = uuid.uuid4().hex
@@ -320,7 +319,7 @@ async def test_sentinel_post_run_cron(
 
 @pytest.mark.anyio
 async def test_sentinel_post_run_time(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     prefix = uuid.uuid4().hex
@@ -345,7 +344,7 @@ async def test_sentinel_post_run_time(
 
 @pytest.mark.anyio
 async def test_sentinel_buffer(
-    redis_sentinels: List[Tuple[str, int]],
+    redis_sentinels: list[tuple[str, int]],
     redis_sentinel_master_name: str,
 ) -> None:
     prefix = uuid.uuid4().hex