From 70a4cfc1e0a9e3d55515e5cab93f32100dbcd8ba Mon Sep 17 00:00:00 2001 From: Seth Foster Date: Thu, 24 Nov 2022 15:11:33 -0500 Subject: [PATCH] Add unit test for cancel_callback --- tests/test_utils.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/test_utils.py diff --git a/tests/test_utils.py b/tests/test_utils.py new file mode 100644 index 00000000..e7654da3 --- /dev/null +++ b/tests/test_utils.py @@ -0,0 +1,20 @@ +import asyncio +from unittest import mock + +import async_timeout +import pytest + +from channels.utils import await_many_dispatch + + +async def sleep_task(*args): + await asyncio.sleep(10) + + +@pytest.mark.asyncio +async def test_cancel_callback_called(): + cancel_callback = mock.AsyncMock() + with pytest.raises(asyncio.TimeoutError): + async with async_timeout.timeout(0): + await await_many_dispatch([sleep_task], sleep_task, cancel_callback) + assert cancel_callback.called