Files
romm/backend/utils/tests/test_utils.py
2023-11-13 00:09:38 -05:00

67 lines
2.2 KiB
Python

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.O.N"
def test_get_file_extension():
assert gfe("Super Mario Bros. (World).nes") == "nes"
assert gfe("007 - Agent Under Fire.nkit.iso") == "nkit.iso"