Skip to content

Commit f9e075f

Browse files
committed
chore: remove pytest.mark.anyio (we have anyio_mode=auto) and type ignore[override] (added override)
1 parent 67336bb commit f9e075f

2 files changed

Lines changed: 11 additions & 12 deletions

File tree

‎tests/abc/test_broker.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ async def test_task() -> None: ...
8282
assert test_task.labels == old_labels
8383

8484

85-
@pytest.mark.anyio
8685
@pytest.mark.parametrize(
8786
("is_worker_process", "startup", "shutdown"),
8887
[
@@ -120,7 +119,6 @@ async def track_shutdown(state: TaskiqState) -> None:
120119
assert shutdown_called is True
121120

122121

123-
@pytest.mark.anyio
124122
@pytest.mark.parametrize(
125123
("is_worker_process", "startup", "shutdown"),
126124
[

‎tests/cli/scheduler/test_send_with_timeout.py‎

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from typing import Any
44

55
import pytest
6+
from typing_extensions import override
67

78
from taskiq.abc.schedule_source import ScheduleSource
89
from taskiq.brokers.inmemory_broker import InMemoryBroker
@@ -30,7 +31,8 @@ def __init__(self, hang_seconds: float = 60.0) -> None:
3031
self.hang_seconds = hang_seconds
3132
self.on_ready_calls = 0
3233

33-
async def on_ready( # type: ignore[override]
34+
@override
35+
async def on_ready(
3436
self,
3537
source: ScheduleSource,
3638
task: ScheduledTask,
@@ -50,7 +52,6 @@ def _task() -> ScheduledTask:
5052
)
5153

5254

53-
@pytest.mark.anyio
5455
async def test_send_with_timeout_returns_on_timeout_without_raising(
5556
caplog: pytest.LogCaptureFixture,
5657
) -> None:
@@ -87,7 +88,6 @@ async def test_send_with_timeout_returns_on_timeout_without_raising(
8788
assert "dummy" in msg
8889

8990

90-
@pytest.mark.anyio
9191
async def test_send_with_timeout_does_not_log_on_success(
9292
caplog: pytest.LogCaptureFixture,
9393
) -> None:
@@ -99,7 +99,8 @@ def __init__(self) -> None:
9999
super().__init__(broker=InMemoryBroker(), sources=[_StubSource()])
100100
self.calls = 0
101101

102-
async def on_ready( # type: ignore[override]
102+
@override
103+
async def on_ready(
103104
self,
104105
source: ScheduleSource,
105106
task: ScheduledTask,
@@ -121,7 +122,6 @@ async def on_ready( # type: ignore[override]
121122
assert warnings == []
122123

123124

124-
@pytest.mark.anyio
125125
async def test_send_with_timeout_propagates_non_timeout_exceptions() -> None:
126126
"""Errors inside on_ready that AREN'T a timeout must still propagate.
127127
@@ -134,7 +134,8 @@ class _BoomScheduler(TaskiqScheduler):
134134
def __init__(self) -> None:
135135
super().__init__(broker=InMemoryBroker(), sources=[_StubSource()])
136136

137-
async def on_ready( # type: ignore[override]
137+
@override
138+
async def on_ready(
138139
self,
139140
source: ScheduleSource,
140141
task: ScheduledTask,
@@ -145,7 +146,6 @@ async def on_ready( # type: ignore[override]
145146
await send_with_timeout(_BoomScheduler(), _StubSource(), _task(), timeout=5.0)
146147

147148

148-
@pytest.mark.anyio
149149
async def test_send_with_timeout_cancels_inner_send() -> None:
150150
"""The inner ``send`` coroutine must actually be cancelled on timeout.
151151
@@ -158,7 +158,8 @@ class _CancelObservingScheduler(TaskiqScheduler):
158158
def __init__(self) -> None:
159159
super().__init__(broker=InMemoryBroker(), sources=[_StubSource()])
160160

161-
async def on_ready( # type: ignore[override]
161+
@override
162+
async def on_ready(
162163
self,
163164
source: ScheduleSource,
164165
task: ScheduledTask,
@@ -179,7 +180,6 @@ async def on_ready( # type: ignore[override]
179180
await asyncio.wait_for(cancelled.wait(), timeout=2.0)
180181

181182

182-
@pytest.mark.anyio
183183
async def test_plain_send_still_works_unchanged() -> None:
184184
"""The original ``send`` function must remain unchanged in behavior.
185185
@@ -191,7 +191,8 @@ def __init__(self) -> None:
191191
super().__init__(broker=InMemoryBroker(), sources=[_StubSource()])
192192
self.calls: list[Any] = []
193193

194-
async def on_ready( # type: ignore[override]
194+
@override
195+
async def on_ready(
195196
self,
196197
source: ScheduleSource,
197198
task: ScheduledTask,

0 commit comments

Comments
 (0)