from m3tpostprocess import process_line, process_scn_line, process_xml_line, replace_quote_str, process_circus_line


def test_process_line(src, expected):
    processed = process_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


def test_process_scn_line(src, expected):
    processed = process_scn_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")

def test_process_xml_line(src, expected):
    processed = process_xml_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


def test_replace_quote_str(src, expected):
    processed = replace_quote_str(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


def test_process_circus_line(src, expected):
    processed = process_circus_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


if __name__ == "__main__":
    test_process_line("Hello....", "Hello……")
    test_process_line("……。", "……")
    test_process_line("，。……", "……")
    test_process_line("…—~", "……")
    test_process_line("----~...", "——")
    test_process_line("~~--", "～～～～")
    test_process_line("!?!?", "？！？！")
    test_process_line("これはテストです测试。", "测试。")
    test_process_line("「地形模型。」", "「地形模型」")
    test_process_line("（测试。。、）", "（测试）")
    test_process_line('"狗屎\'引号\'就是"一坨""', "「狗屎『引号』就是「一坨」」")
    test_process_scn_line("「地形模型。%command;」", "「地形模型%command;」")
    test_process_scn_line("%cmd;25%的可能。", "%cmd;25％的可能。")
    test_process_xml_line("<tag>（测试。。、）</tag>", "<tag>（测试）</tag>")
    test_process_xml_line('"狗屎\'引号\'就是"一坨""<tag ok="this is ok">', '「狗屎『引号』就是「一坨」」<tag ok="this is ok">')
    test_process_xml_line("「地形模型。<command>」", "「地形模型<command>」")
    test_replace_quote_str("「狗屎『引号』就是「一坨」」", "「狗屎『引号』就是『一坨』」")
    test_process_circus_line("这是一个@special标记的测试。", "这是一个@special标记的测试。")
    test_process_circus_line("其他@测@123试", "其他＠测@123试")