Skip to content

Commit 30d527b

Browse files
author
Anton
committed
tests: more tests
1 parent c3e051b commit 30d527b

1 file changed

Lines changed: 36 additions & 0 deletions

File tree

‎tests/receiver/test_receiver.py‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -501,3 +501,39 @@ async def task_no_result() -> str:
501501
assert resp.return_value == "some value"
502502
assert not broker._running_tasks
503503
assert wrapper_call is True
504+
505+
506+
@pytest.mark.anyio
507+
async def test_sync_decorator_on_async_function_with_timeout() -> None:
508+
wrapper_call = False
509+
510+
def wrapper(f: Any) -> Any:
511+
@wraps(f)
512+
def wrapper_impl(*args: Any, **kwargs: Any) -> Any:
513+
nonlocal wrapper_call
514+
515+
wrapper_call = True
516+
return f(*args, **kwargs)
517+
518+
return wrapper_impl
519+
520+
@wrapper
521+
async def test_func() -> None:
522+
await asyncio.sleep(2)
523+
524+
receiver = get_receiver()
525+
526+
result = await receiver.run_task(
527+
test_func,
528+
TaskiqMessage(
529+
task_id="",
530+
task_name="",
531+
labels={"timeout": "0.3"},
532+
args=[],
533+
kwargs={},
534+
),
535+
)
536+
assert result.return_value is None
537+
assert result.execution_time < 2
538+
assert result.is_err
539+
assert wrapper_call is True

0 commit comments

Comments
 (0)