from utils import (
    parse_tags,
    get_file_name_with_no_tags as gfnwt,
    get_file_extension as gfe,
)


def test_parse_tags():
    file_name = "Super Mario Bros. (World).nes"
    assert parse_tags(file_name) == ("World", "", [])

    file_name = "Super Mario Bros. (W) (Rev A).nes"
    assert parse_tags(file_name) == ("World", "A", [])

    file_name = "Super Mario Bros. (USA) (Rev A) (Beta).nes"
    assert parse_tags(file_name) == ("USA", "A", ["Beta"])

    file_name = "Super Mario Bros. (U) (Beta).nes"
    assert parse_tags(file_name) == ("USA", "", ["Beta"])

    file_name = "Super Mario Bros. (CH) [!].nes"
    assert parse_tags(file_name) == ("China", "", ["!"])

    file_name = "Super Mario Bros. (reg-T) (rev-1.2).nes"
    assert parse_tags(file_name) == ("Taiwan", "1.2", [])

    file_name = "Super Mario Bros. (Reg S) (Rev A).nes"
    assert parse_tags(file_name) == ("Spain", "A", [])


def test_get_file_name_with_no_tags():
    file_name = "Super Mario Bros. (World).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (W) (Rev A).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (USA) (Rev A) (Beta).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (U) (Beta).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (U) [!].nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (reg-T) (rev-1.2).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "Super Mario Bros. (Reg S) (Rev A).nes"
    assert gfnwt(file_name) == "Super Mario Bros."

    file_name = "007 - Agent Under Fire.nkit.iso"
    assert gfnwt(file_name) == "007 - Agent Under Fire"

    file_name = "Jimmy Houston's Bass Tournament U.S.A..zip"
    assert gfnwt(file_name) == "Jimmy Houston's Bass Tournament U.S.A."

    # This is expected behavior, since the regex is aggressive
    file_name = "Battle Stadium D.O.N.zip"
    assert gfnwt(file_name) == "Battle Stadium D"


def test_get_file_extension():
    rom = {"file_name": "Super Mario Bros. (World).nes", "multi": False}
    assert gfe(rom) == "nes"

    rom = {"file_name": "Super Mario Bros. (World).nes", "multi": True}
    assert gfe(rom) == ""

    rom = {"file_name": "007 - Agent Under Fire.nkit.iso", "multi": False}
    assert gfe(rom) == "nkit.iso"