Rename BaseBot to Bot

tests/test_api/test_client/test_bot.py

@@ -0,0 +1,63 @@
+import pytest
+
+from aiogram import Bot
+from aiogram.api.client.session.aiohttp import AiohttpSession
+from aiogram.api.methods import GetMe
+
+try:
+    from asynctest import CoroutineMock, patch
+except ImportError:
+    from unittest.mock import AsyncMock as CoroutineMock, patch  # type: ignore
+
+
+class TestBot:
+    def test_init(self):
+        bot = Bot("42:TEST")
+        assert isinstance(bot.session, AiohttpSession)
+        assert bot.id == 42
+
+    def test_hashable(self):
+        bot = Bot("42:TEST")
+        assert hash(bot) == hash("42:TEST")
+
+    def test_equals(self):
+        bot = Bot("42:TEST")
+        assert bot == Bot("42:TEST")
+        assert bot != "42:TEST"
+
+    @pytest.mark.asyncio
+    async def test_emit(self):
+        bot = Bot("42:TEST")
+
+        method = GetMe()
+
+        with patch(
+            "aiogram.api.client.session.aiohttp.AiohttpSession.make_request",
+            new_callable=CoroutineMock,
+        ) as mocked_make_request:
+            await bot(method)
+            mocked_make_request.assert_awaited_with("42:TEST", method)
+
+    @pytest.mark.asyncio
+    async def test_close(self):
+        bot = Bot("42:TEST", session=AiohttpSession())
+        await bot.session.create_session()
+
+        with patch(
+            "aiogram.api.client.session.aiohttp.AiohttpSession.close", new_callable=CoroutineMock
+        ) as mocked_close:
+            await bot.close()
+            mocked_close.assert_awaited()
+
+    @pytest.mark.asyncio
+    @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 Bot("42:TEST", session=AiohttpSession()).context(auto_close=close) as bot:
+                assert isinstance(bot, Bot)
+            if close:
+                mocked_close.assert_awaited()
+            else:
+                mocked_close.assert_not_awaited()