diff --git a/tests/test_api/test_client/test_base_bot.py b/tests/test_api/test_client/test_base_bot.py new file mode 100644 index 00000000..e0ca18b6 --- /dev/null +++ b/tests/test_api/test_client/test_base_bot.py @@ -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) diff --git a/tests/test_api/test_client/test_session/test_base_session.py b/tests/test_api/test_client/test_session/test_base_session.py index eb260ba5..d6cbf931 100644 --- a/tests/test_api/test_client/test_session/test_base_session.py +++ b/tests/test_api/test_client/test_session/test_base_session.py @@ -2,11 +2,12 @@ import datetime from unittest.mock import patch import pytest +from asynctest import CoroutineMock + from aiogram.api.client.session.base import BaseSession from aiogram.api.client.telegram import PRODUCTION, TelegramAPIServer from aiogram.api.methods import GetMe, Response from aiogram.utils.mixins import DataMixin -from asynctest import CoroutineMock class TestBaseSession(DataMixin):