mirror of
https://github.com/Balshgit/mosgortrans.git
synced 2025-09-11 13:00:40 +03:00
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import time
|
|
|
|
import pytest
|
|
from aiogram import Bot, types
|
|
from aiogram.dispatcher.filters.builtin import Command
|
|
from app.core.bot import dispatcher
|
|
from tests.conftest import FakeTelegram
|
|
from tests.data.factories import UserFactory
|
|
|
|
pytestmark = [
|
|
pytest.mark.asyncio,
|
|
]
|
|
|
|
|
|
async def test_parse_site(bot: Bot) -> None:
|
|
tg_user = UserFactory().as_dict()
|
|
user = types.User(**tg_user)
|
|
|
|
async with FakeTelegram(message_data=tg_user):
|
|
result = await bot.me
|
|
|
|
assert result == user
|
|
|
|
|
|
async def test_command1(bot: Bot) -> None:
|
|
|
|
dispatcher.bot = bot
|
|
handlers = dispatcher.message_handlers.handlers
|
|
for handler in handlers:
|
|
handl = list(
|
|
filter(lambda obj: isinstance(obj.filter, Command), handler.filters)
|
|
)
|
|
if handl:
|
|
command = handl[0].filter.commands[0]
|
|
assert command
|
|
|
|
|
|
async def test_update() -> None:
|
|
data = {
|
|
"update_id": 957250703,
|
|
"message": {
|
|
"message_id": 417070387,
|
|
"from": {
|
|
"id": 417070387,
|
|
"is_bot": False,
|
|
"first_name": "Dmitry",
|
|
"last_name": "Afanasyev",
|
|
"username": "Balshtg",
|
|
"language_code": "en",
|
|
},
|
|
"chat": {
|
|
"id": 417070387,
|
|
"first_name": "Dmitry",
|
|
"last_name": "Afanasyev",
|
|
"username": "Balshtg",
|
|
"type": "private",
|
|
},
|
|
"date": time.time(),
|
|
"text": "/chatid",
|
|
"entities": [{"type": "bot_command", "offset": 0, "length": 7}],
|
|
},
|
|
}
|
|
assert data
|