Skip to content

Commit

Permalink
Add unit test for cancel_callback
Browse files Browse the repository at this point in the history
  • Loading branch information
fosterseth authored and bigfootjon committed Sep 3, 2024
1 parent 847f735 commit 70a4cfc
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions tests/test_utils.py
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 70a4cfc

Please sign in to comment.