33from typing import Any
44
55import pytest
6+ from typing_extensions import override
67
78from taskiq .abc .schedule_source import ScheduleSource
89from 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
5455async 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
9191async 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
125125async 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
149149async 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
183183async 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