Files
romm/backend/handler/tests/test_fastapi.py
2024-12-21 12:06:37 -05:00

56 lines
1.5 KiB
Python

import pytest
from handler.scan_handler import ScanType, scan_platform, scan_rom
from models.platform import Platform
from models.rom import Rom, RomFile
from utils.context import initialize_context
@pytest.mark.vcr
async def test_scan_platform():
async with initialize_context():
platform = await scan_platform("n64", ["n64"])
assert type(platform) is Platform
assert platform.fs_slug == "n64"
assert platform.slug == "n64"
assert platform.name == "Nintendo 64"
assert platform.igdb_id == 4
async with initialize_context():
platform = await scan_platform("", [])
assert platform.fs_slug == ""
assert platform.slug == ""
assert platform.name == ""
assert platform.igdb_id is None
@pytest.mark.vcr
async def test_scan_rom():
platform = Platform(fs_slug="n64", igdb_id=4)
async with initialize_context():
files = [
RomFile(
file_name="Paper Mario (USA).z64",
file_path="Paper Mario (USA)",
file_size_bytes=1024,
last_modified=1620000000,
)
]
rom = await scan_rom(
platform,
{"fs_name": "Paper Mario (USA).z64", "multi": False, "files": files},
ScanType.QUICK,
)
assert type(rom) is Rom
assert rom.fs_name == "Paper Mario (USA).z64"
assert rom.name == "Paper Mario"
assert rom.igdb_id == 3340
assert rom.fs_size_bytes == 1024
assert rom.files == files
assert rom.tags == []
assert not rom.multi