Add tests for BaseBot

This commit is contained in:
Alex Root Junior 2019-11-17 23:37:24 +02:00
parent a46cd6d001
commit 3a43ffc570
2 changed files with 27 additions and 1 deletions

View file

@ -0,0 +1,25 @@
import pytest
from asynctest import CoroutineMock, patch
from aiogram.api.client.base import BaseBot
from aiogram.api.client.session.aiohttp import AiohttpSession
from aiogram.api.methods import GetMe
class TestBaseBot:
def test_init(self):
base_bot = BaseBot("TOKEN")
assert isinstance(base_bot.session, AiohttpSession)
@pytest.mark.asyncio
async def test_emit(self):
base_bot = BaseBot("TOKEN")
method = GetMe()
with patch(
"aiogram.api.client.session.aiohttp.AiohttpSession.make_request",
new_callable=CoroutineMock,
) as mocked_make_request:
await base_bot.emit(method)
mocked_make_request.assert_awaited_with("TOKEN", method)