aiogram/tests/test_api/test_methods/test_get_me.py
2022-12-30 16:10:06 +02:00

31 lines
1.1 KiB
Python

from aiogram.methods import GetMe, Request
from aiogram.types import User
from tests.mocked_bot import MockedBot
class TestGetMe:
async def test_method(self, bot: MockedBot):
prepare_result = bot.add_result_for(
GetMe, ok=True, result=User(id=42, is_bot=False, first_name="User")
)
response: User = await GetMe()
request: Request = bot.get_request()
assert request.method == "getMe"
assert request.data == {}
assert response == prepare_result.result
async def test_bot_method(self, bot: MockedBot):
prepare_result = bot.add_result_for(
GetMe, ok=True, result=User(id=42, is_bot=False, first_name="User")
)
response: User = await bot.get_me()
request: Request = bot.get_request()
assert request.method == "getMe"
assert request.data == {}
assert response == prepare_result.result
async def test_me_property(self, bot: MockedBot):
response: User = await bot.me()
assert isinstance(response, User)
assert response == bot._me