danbusse
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""
|
|
Tests for archidekt.py — CSV and JSON parsing with various column name formats.
|
|
"""
|
|
import pytest
|
|
from app.services.imports.archidekt import parse
|
|
|
|
|
|
class TestArchidektCSV:
|
|
def _csv(self, header, rows):
|
|
lines = [header] + rows
|
|
return "\n".join(lines).encode("utf-8")
|
|
|
|
def test_standard_columns(self):
|
|
data = self._csv(
|
|
"Quantity,Foil Quantity,Card Name,Set Code,Collector Number",
|
|
["2,0,Sol Ring,cmm,1"]
|
|
)
|
|
result = parse(data)
|
|
assert len(result) == 1
|
|
assert result[0].card_name == "Sol Ring"
|
|
assert result[0].quantity == 2
|
|
assert result[0].foil_quantity == 0
|
|
assert result[0].set_code == "cmm"
|
|
|
|
def test_alternate_column_names(self):
|
|
data = self._csv(
|
|
"qty,foil qty,name,set,number",
|
|
["1,1,Black Lotus,lea,232"]
|
|
)
|
|
result = parse(data)
|
|
assert result[0].card_name == "Black Lotus"
|
|
assert result[0].quantity == 1
|
|
assert result[0].foil_quantity == 1
|
|
|
|
def test_skips_empty_rows(self):
|
|
data = self._csv(
|
|
"Quantity,Card Name",
|
|
["1,Sol Ring", ",", "2,Command Tower"]
|
|
)
|
|
result = parse(data)
|
|
assert len(result) == 2
|
|
|
|
def test_empty_csv_raises(self):
|
|
with pytest.raises(ValueError, match="empty"):
|
|
parse(b"")
|
|
|
|
def test_no_card_rows_raises(self):
|
|
data = self._csv("Quantity,Card Name", [])
|
|
with pytest.raises(ValueError, match="no card rows"):
|
|
parse(data)
|
|
|
|
|
|
class TestArchidektJSON:
|
|
def test_standard_json_array(self):
|
|
data = b'''[
|
|
{"quantity": 1, "foilQuantity": 0, "card": {"name": "Sol Ring", "set": {"code": "cmm"}, "collectorNumber": "1"}}
|
|
]'''
|
|
result = parse(data)
|
|
assert len(result) == 1
|
|
assert result[0].card_name == "Sol Ring"
|
|
assert result[0].set_code == "cmm"
|
|
|
|
def test_cards_wrapper(self):
|
|
data = b'{"cards": [{"quantity": 1, "card": {"name": "Sol Ring"}}]}'
|
|
result = parse(data)
|
|
assert result[0].card_name == "Sol Ring"
|
|
|
|
def test_foil_quantity(self):
|
|
data = b'[{"quantity": 0, "foilQuantity": 2, "card": {"name": "Sol Ring"}}]'
|
|
result = parse(data)
|
|
assert result[0].foil_quantity == 2
|
|
|
|
def test_empty_array_raises(self):
|
|
with pytest.raises(ValueError, match="no card entries"):
|
|
parse(b"[]")
|
|
|
|
def test_malformed_json_raises(self):
|
|
with pytest.raises(ValueError, match="malformed"):
|
|
parse(b"{broken json")
|