python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: wait for initial_sleep

Browse source
This commit is contained in:
Oleg A 2022-09-21 08:12:17 +03:00
parent 1e32c7aeb8
commit e137c2882f
No known key found for this signature in database
GPG key ID: 5FE046817A9657C5
2 changed files with 17 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_utils/test_chat_action.py
View file

@ -20,12 +20,25 @@ pytestmarm = pytest.mark.asyncio
class TestChatActionSender:
async def test_wait(self, bot: Bot, loop: asyncio.BaseEventLoop):
initial_sleep = 0.2
async def test_wait_with_event(self, bot: Bot, loop: asyncio.BaseEventLoop):
sender = ChatActionSender.typing(bot=bot, chat_id=42)
loop.call_soon(sender._close_event.set)
start = time.monotonic()
await sender._wait(1)
assert time.monotonic() - start >= 1
await sender._wait(self.initial_sleep)
assert time.monotonic() - start < self.initial_sleep
async def test_wait_without_event(self, bot: Bot):
sender = ChatActionSender.typing(bot=bot, chat_id=42)
start = time.monotonic()
await sender._wait(self.initial_sleep)
assert time.monotonic() - start >= self.initial_sleep
async def test_initial_sleep(self, bot: Bot):
start = time.monotonic()
async with ChatActionSender.typing(bot=bot, chat_id=42, initial_sleep=self.initial_sleep):
assert time.monotonic() - start >= self.initial_sleep
@pytest.mark.parametrize(
"action",