Files
romm/backend/handler/tests/test_fastapi.py
Georges-Antoine Assi 0139e3a13a fix tests
2024-07-27 16:56:13 -04:00

57 lines
1.6 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(
filename="Paper Mario (USA).z64",
size=1024,
crc_hash="9d0d1c6e",
md5_hash="f1b7f9e4f4d0e0b7b9faa1b1f2f8e4e9",
sha1_hash="c3c7f9f3d1d0e0b7b9faa1b1f2f8e4e9",
)
]
rom = await scan_rom(
platform,
{"file_name": "Paper Mario (USA).z64", "multi": False, "files": files},
ScanType.QUICK,
)
assert type(rom) is Rom
assert rom.file_name == "Paper Mario (USA).z64"
assert rom.name == "Paper Mario"
assert rom.igdb_id == 3340
assert rom.file_size_bytes == 1024
assert rom.files == files
assert rom.tags == []
assert not rom.multi