MpakoImageAIbot/tests/test_settings.py

34 lines
903 B
Python

from __future__ import annotations
from bot import MpakoBot
from config import Settings
def test_settings_command_is_not_shadowed_by_settings_attribute() -> None:
bot = MpakoBot(Settings(bot_token=""))
assert callable(bot._settings_command)
def test_quality_keyboard_marks_standard() -> None:
keyboard = MpakoBot._quality_keyboard("standard")
rows = keyboard.inline_keyboard
assert [row[0].text for row in rows] == [
"Стандартная ✅",
"Улучшенная",
]
assert [row[0].callback_data for row in rows] == [
"quality:standard",
"quality:improved",
]
def test_quality_keyboard_marks_improved() -> None:
keyboard = MpakoBot._quality_keyboard("improved")
rows = keyboard.inline_keyboard
assert [row[0].text for row in rows] == [
"Стандартная",
"Улучшенная ✅",
]