from io import BytesIO

from openpyxl import Workbook

from bls.parser import parse_components_xlsx, parse_foods_xlsx


def _xlsx(rows):
    wb = Workbook()
    ws = wb.active
    for row in rows:
        ws.append(row)
    buf = BytesIO()
    wb.save(buf)
    return buf.getvalue()


def test_parse_components_dynamic():
    data = _xlsx([
        ["Code", "Name_DE", "Name_EN", "Einheit"],
        ["ENERCC", "Energie", "Energy", "kcal/100g"],
        ["NA", "Natrium", "Sodium", "mg/100g"],
    ])
    attrs = parse_components_xlsx(data)
    keys = {a["attr_key"] for a in attrs}
    assert "ENERCC" in keys
    assert "NA" in keys


def test_parse_foods_keeps_bls_code():
    data = _xlsx([
        ["BLS Code", "Name", "Food name", "ENERCC Energie [kcal/100g]", "ENERCC Herkunft", "ENERCC Referenz"],
        ["C131000", "Hafer roh", "Oats raw", 350, "Analyse", "MRI"],
    ])
    parsed = parse_foods_xlsx(data)
    assert parsed["foods"][0]["bls_code"] == "C131000"
    assert parsed["foods"][0]["name_de"] == "Hafer roh"
    vals = {v["attr_key"]: v["value_num"] for v in parsed["foods"][0]["values"]}
    assert vals.get("ENERCC") == 350