from __future__ import annotations

from bot import MpakoBot


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] == [
        "Стандартная",
        "Улучшенная ✅",
    ]