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

Cover BaseBot context manager

Browse source
This commit is contained in:
jrootjunior 2019-11-26 11:13:48 +02:00
parent b5ab2fd153
commit cd12b54ec2
3 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,7 @@
import pytest
from asynctest import CoroutineMock, patch
from aiogram import Bot
from aiogram.api.client.base import BaseBot
from aiogram.api.client.session.aiohttp import AiohttpSession
from aiogram.api.methods import GetMe
@ -23,3 +24,23 @@ class TestBaseBot:
) as mocked_make_request:
await base_bot.emit(method)
mocked_make_request.assert_awaited_with("TOKEN", method)
@pytest.mark.asyncio
async def test_close(self):
base_bot = BaseBot("TOKEN", session=AiohttpSession())
await base_bot.session.create_session()
with patch(
"aiogram.api.client.session.aiohttp.AiohttpSession.close", new_callable=CoroutineMock
) as mocked_close:
await base_bot.close()
mocked_close.assert_awaited()
@pytest.mark.asyncio
async def test_context_manager(self):
with patch(
"aiogram.api.client.session.aiohttp.AiohttpSession.close", new_callable=CoroutineMock
) as mocked_close:
async with BaseBot("TOKEN", session=AiohttpSession()) as bot:
assert isinstance(bot, BaseBot)
mocked_close.assert_awaited()