import pytest
import os.path
import sys

from ee.kicad.bom import *
from ee.kicad.bom.io import read_bom

basedir = os.path.dirname(os.path.abspath(__file__))


@pytest.mark.parametrize("s, ref, val, rest", [
    ("C12", "C", 12, ""),
    ("C12n", "C", 12, "n"),
    ("C", "C", sys.maxsize, ""),
    ("Foo", "Foo", sys.maxsize, ""),
    ("+3.0VA1", "+3.0VA1", sys.maxsize, ""),
])
def test_split_ref(s, ref, val, rest):
    assert split_ref(s) == (ref, val, rest)


def test_read_bom_1():
    bom = read_bom(basedir + '/../demo/kicad/bom/A64-OlinuXino_Rev_C.xml')
    assert len(bom.get_components()) == 425


def test_read_bom_2():
    bom = read_bom(basedir + '/../demo/kicad/bom/gw.xml')
    assert len(bom.get_components()) == 165

    r5 = bom.get_component("R5")
    assert r5.ref == "R5"
    assert r5.value == "R0402_100R"
    assert r5["value"] == "R0402_100R"
    assert r5.footprint == "Resistors_SMD:R_0402"
    assert r5.library.name == "gw-cache"
    assert len(r5.fields) == 4
    assert r5.fields["Part Number"] == "CRCW0402100RFKED"
    assert r5["Part Number"] == "CRCW0402100RFKED"
    assert set(
        ['ref', 'value', 'Capacitance', 'Color', 'Description', 'Frequency', 'Impedance', 'Inductance', 'Manufacturer',
         'Part Number', 'Resistance']) == bom.all_field_names()

    assert not "foo" in r5
    with pytest.raises(KeyError):
        r5["foo"]


def test_read_bom_2_pandas():
    bom = read_bom(basedir + '/../demo/kicad/bom/gw.xml').to_pandas()
    assert len(bom) == 165

    print("bom")
    print(str(bom))
    r5 = bom.loc["R5"]
    print(str(r5.index))
    #  assert r5.index == "R5"
    assert r5["ref"] == "R5"
    assert r5["value"] == "R0402_100R"
    assert r5["value"] == "R0402_100R"
    #  assert r5["footprint"] == "Resistors_SMD:R_0402"
    #  assert r5.library.name == "gw-cache"
    #  assert len(r5.fields) == 4
    #  assert r5.fields["Part Number"] == "CRCW0402100RFKED"
    assert r5["Part Number"] == "CRCW0402100RFKED"
    #  assert set(['ref', 'value', 'Capacitance', 'Color', 'Description', 'Frequency', 'Impedance', 'Inductance', 'Manufacturer', 'Part Number', 'Resistance']) == bom.all_field_names()

    assert not "foo" in r5
    with pytest.raises(KeyError):
        r5["foo"]