jorgasisten

from jorgasisten.bot.main import normalize_direct_prompt


def test_normalize_direct_prompt_strips_data_prefix() -> None:
    assert normalize_direct_prompt("tanya data stok oli mpx berapa?") == "stok oli mpx berapa?"
    assert normalize_direct_prompt("cari data: booking atas nama Budi") == "booking atas nama Budi"


def test_normalize_direct_prompt_strips_chat_prefix() -> None:
    assert normalize_direct_prompt("chat umum jelasin fungsi busi motor") == "jelasin fungsi busi motor"
    assert normalize_direct_prompt("chat: bantu bikin caption promo") == "bantu bikin caption promo"


def test_normalize_direct_prompt_keeps_plain_question() -> None:
    plain_question = "stok oli mpx berapa?"

    assert normalize_direct_prompt(plain_question) == plain_question