From 68c63fa0ff44f554b8496cf8a3641ecf332ec34f 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 | 18 ++++++++++++++++++ 1 file changed, 18 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..f2266db2 --- /dev/null +++ b/tests/test_utils.py @@ -0,0 +1,18 @@ +from channels.utils import await_many_dispatch +import asyncio +import async_timeout +from unittest import mock +import pytest + + +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