From 3a43ffc5708fdf263956ccccda6b275b82a2a944 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Sun, 17 Nov 2019 23:37:24 +0200 Subject: [PATCH] Add tests for BaseBot --- tests/test_api/test_client/test_base_bot.py | 25 +++++++++++++++++++ .../test_session/test_base_session.py | 3 ++- 2 files changed, 27 insertions(+), 1 deletion(-) create mode 100644 tests/test_api/test_client/test_base_bot.py 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):