mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
from handler.db_handler import DBHandler
|
|
from models.platform import Platform
|
|
from models.rom import Rom
|
|
|
|
dbh = DBHandler()
|
|
|
|
|
|
def test_platforms():
|
|
platform = Platform(
|
|
name="test_platform", slug="test_platform_slug", fs_slug="test_platform"
|
|
)
|
|
dbh.add_platform(platform)
|
|
|
|
platforms = dbh.get_platforms()
|
|
assert len(platforms) == 1
|
|
|
|
platform = dbh.get_platform("test_platform_slug")
|
|
assert platform.name == "test_platform"
|
|
|
|
dbh.purge_platforms(["test_platform_slug"])
|
|
platforms = dbh.get_platforms()
|
|
assert len(platforms) == 0
|
|
|
|
|
|
def test_roms(rom):
|
|
dbh.add_rom(
|
|
Rom(
|
|
r_name="test_rom_2",
|
|
r_slug="test_rom_slug_2",
|
|
p_name="test_platform",
|
|
p_slug="test_platform_slug",
|
|
file_name="test_rom_2",
|
|
file_name_no_tags="test_rom_2",
|
|
)
|
|
)
|
|
|
|
with dbh.session.begin() as session:
|
|
roms = session.scalars(dbh.get_roms("test_platform_slug")).all()
|
|
assert len(roms) == 2
|
|
|
|
rom = dbh.get_rom(roms[0].id)
|
|
assert rom.file_name == "test_rom"
|
|
|
|
dbh.update_rom(roms[1].id, {"file_name": "test_rom_2_updated"})
|
|
rom_2 = dbh.get_rom(roms[1].id)
|
|
assert rom_2.file_name == "test_rom_2_updated"
|
|
|
|
dbh.delete_rom(rom.id)
|
|
|
|
with dbh.session.begin() as session:
|
|
roms = session.scalars(dbh.get_roms(rom.p_slug)).all()
|
|
assert len(roms) == 1
|
|
|
|
dbh.purge_roms(rom_2.p_slug, [rom_2.r_slug])
|
|
|
|
with dbh.session.begin() as session:
|
|
roms = session.scalars(dbh.get_roms("test_platform_slug")).all()
|
|
assert len(roms) == 0
|
|
|
|
|
|
def test_utils(rom):
|
|
with dbh.session.begin() as session:
|
|
roms = session.scalars(dbh.get_roms("test_platform_slug")).all()
|
|
assert dbh.rom_exists("test_platform_slug", "test_rom") == roms[0].id
|