python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-10 17:53:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Rework polling and start covering

Browse source
This commit is contained in:
Alex Root Junior 2019-12-10 01:14:58 +02:00
parent 38c725db46
commit db397e3a05
6 changed files with 84 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_api/test_client/test_base_bot.py
View file

@ -48,10 +48,16 @@ class TestBaseBot:
mocked_close.assert_awaited()
@pytest.mark.asyncio
async def test_context_manager(self):
@pytest.mark.parametrize("close", [True, False])
async def test_context_manager(self, close: bool):
with patch(
"aiogram.api.client.session.aiohttp.AiohttpSession.close", new_callable=CoroutineMock
) as mocked_close:
async with BaseBot("42:TEST", session=AiohttpSession()) as bot:
async with BaseBot("42:TEST", session=AiohttpSession()).context(
auto_close=close
) as bot:
assert isinstance(bot, BaseBot)
mocked_close.assert_awaited()
if close:
mocked_close.assert_awaited()
else:
mocked_close.assert_not_awaited()