chore: enable anyio_mode=auto

Author: danfimov

poetry.lock

@@ -34,7 +34,7 @@ pydantic = ">=1.0,<=3.0"
 importlib-metadata = "*"
 pycron = "^3.0.0"
 taskiq_dependencies = ">=1.3.1,<2"
-anyio = ">=3"
+anyio = ">=4"
 packaging = ">=19"
 # For prometheus metrics
 prometheus_client = { version = "^0", optional = true }
@@ -104,6 +104,7 @@ multi_line_output = 3
 
 [tool.pytest.ini_options]
 log_level = 'INFO'
+anyio_mode = "auto"
 
 [tool.coverage.run]
 omit = [
@@ -174,6 +175,7 @@ line-length = 88
     "SLF001", # Private member accessed
     "S311",   # Standard pseudo-random generators are not suitable for security/cryptographic purposes
     "D101",   # Missing docstring in public class
+    "D102",  # Missing docstring in public method
 ]
 
 [tool.ruff.lint.pydocstyle]

pyproject.toml

@@ -7,7 +7,6 @@
 from tests.utils import AsyncQueueBroker
 
 
-@pytest.mark.anyio
 async def test_successful() -> None:
     broker = AsyncQueueBroker()
     kicked = 0
@@ -28,7 +27,6 @@ def test_func() -> None:
     assert kicked == desired_kicked
 
 
-@pytest.mark.anyio
 async def test_cancelation() -> None:
     broker = AsyncQueueBroker()
     kicked = 0

tests/api/test_receiver_task.py

@@ -2,15 +2,12 @@
 import contextlib
 from datetime import datetime, timedelta, timezone
 
-import pytest
-
 from taskiq import TaskiqScheduler
 from taskiq.api import run_scheduler_task
 from taskiq.schedule_sources import LabelScheduleSource
 from tests.utils import AsyncQueueBroker
 
 
-@pytest.mark.anyio
 async def test_successful() -> None:
     broker = AsyncQueueBroker()
     scheduler = TaskiqScheduler(broker, sources=[LabelScheduleSource(broker)])
@@ -26,7 +23,6 @@ def _() -> None:
     scheduler_task.cancel()
 
 
-@pytest.mark.anyio
 async def test_cancelation() -> None:
     broker = AsyncQueueBroker()
     scheduler = TaskiqScheduler(broker, sources=[LabelScheduleSource(broker)])

tests/api/test_scheduler.py

@@ -8,7 +8,6 @@
 from taskiq.state import TaskiqState
 
 
-@pytest.mark.anyio
 async def test_inmemory_success() -> None:
     broker = InMemoryBroker()
     test_val = uuid.uuid4().hex
@@ -23,7 +22,6 @@ async def task() -> str:
     assert not broker._running_tasks
 
 
-@pytest.mark.anyio
 async def test_cannot_listen() -> None:
     broker = InMemoryBroker()
 
@@ -32,7 +30,6 @@ async def test_cannot_listen() -> None:
             pass
 
 
-@pytest.mark.anyio
 async def test_startup() -> None:
     broker = InMemoryBroker()
     test_value = uuid.uuid4().hex
@@ -51,7 +48,6 @@ async def _c_startup(state: TaskiqState) -> None:
     assert broker.state.from_client == test_value
 
 
-@pytest.mark.anyio
 async def test_shutdown() -> None:
     broker = InMemoryBroker()
     test_value = uuid.uuid4().hex
@@ -70,7 +66,6 @@ async def _c_startup(state: TaskiqState) -> None:
     assert broker.state.from_client == test_value
 
 
-@pytest.mark.anyio
 async def test_execution() -> None:
     broker = InMemoryBroker()
     test_value = uuid.uuid4().hex
@@ -87,7 +82,6 @@ async def test_task() -> str:
     assert result.return_value == test_value
 
 
-@pytest.mark.anyio
 async def test_inline_awaits() -> None:
     broker = InMemoryBroker(await_inplace=True)
     slept = False
@@ -104,7 +98,6 @@ async def test_task() -> None:
     assert not broker._running_tasks
 
 
-@pytest.mark.anyio
 async def test_wait_all() -> None:
     broker = InMemoryBroker()
     slept = False

tests/brokers/test_inmemory.py

@@ -1,8 +1,6 @@
 from datetime import datetime
 from typing import List, Union
 
-import pytest
-
 from taskiq import InMemoryBroker, ScheduleSource
 from taskiq.cli.scheduler.run import get_all_schedules
 from taskiq.scheduler.scheduled_task import ScheduledTask
@@ -20,7 +18,6 @@ async def get_schedules(self) -> List[ScheduledTask]:
         return self.schedules
 
 
-@pytest.mark.anyio
 async def test_get_schedules_success() -> None:
     """Tests that schedules are returned correctly."""
     schedules1 = [
@@ -62,7 +59,6 @@ async def test_get_schedules_success() -> None:
     ]
 
 
-@pytest.mark.anyio
 async def test_get_schedules_error() -> None:
     """Tests that if source returned an error, empty list will be returned."""
     source1 = DummySource(

tests/cli/scheduler/test_updater.py

@@ -56,7 +56,6 @@ def get_message(
     )
 
 
-@pytest.mark.anyio
 @pytest.mark.parametrize(
     "state,meta",
     [
@@ -82,7 +81,6 @@ async def test_func(tes_val: ProgressTracker[Any] = TaskiqDepends()) -> None:
     assert progress.state == state
 
 
-@pytest.mark.anyio
 async def test_progress_tracker_ctx_none() -> None:
     broker = InMemoryBroker()
 
@@ -98,7 +96,6 @@ async def test_func() -> None:
     assert progress is None
 
 
-@pytest.mark.anyio
 @pytest.mark.parametrize(
     "state,meta",
     [

tests/depends/test_progress_tracker.py

@@ -1,12 +1,9 @@
 import json
 
-import pytest
-
 from taskiq.formatters.json_formatter import JSONFormatter
 from taskiq.message import BrokerMessage, TaskiqMessage
 
 
-@pytest.mark.anyio
 async def test_json_dumps() -> None:
     fmt = JSONFormatter()
     msg = TaskiqMessage(
@@ -34,7 +31,6 @@ async def test_json_dumps() -> None:
     assert json.loads(dumped.message) == json.loads(expected.message)
 
 
-@pytest.mark.anyio
 async def test_json_loads() -> None:
     fmt = JSONFormatter()
     msg = (

tests/formatters/test_json_formatter.py

@@ -1,10 +1,7 @@
-import pytest
-
 from taskiq.brokers.inmemory_broker import InMemoryBroker
 from taskiq.message import BrokerMessage, TaskiqMessage
 
 
-@pytest.mark.anyio
 async def test_proxy_dumps() -> None:
     # uses json serializer by default
     broker = InMemoryBroker()
@@ -29,7 +26,6 @@ async def test_proxy_dumps() -> None:
     assert broker.formatter.dumps(msg) == expected
 
 
-@pytest.mark.anyio
 async def test_proxy_loads() -> None:
     # uses json serializer by default
     broker = InMemoryBroker()

tests/formatters/test_proxy_formatter.py

@@ -17,7 +17,6 @@ def broker() -> AsyncMock:
     return mocked_broker
 
 
-@pytest.mark.anyio
 async def test_successful_retry(broker: AsyncMock) -> None:
     middleware = SimpleRetryMiddleware()
     middleware.set_broker(broker)
@@ -39,7 +38,6 @@ async def test_successful_retry(broker: AsyncMock) -> None:
     assert resend.labels["_retries"] == "1"
 
 
-@pytest.mark.anyio
 async def test_no_retry(broker: AsyncMock) -> None:
     middleware = SimpleRetryMiddleware()
     middleware.set_broker(broker)
@@ -57,7 +55,6 @@ async def test_no_retry(broker: AsyncMock) -> None:
     broker.kick.assert_not_called()
 
 
-@pytest.mark.anyio
 async def test_max_retries(broker: AsyncMock) -> None:
     middleware = SimpleRetryMiddleware(default_retry_count=3)
     middleware.set_broker(broker)