from typing import Optional

import pytest

from ee.db import *


class MyPartNumber(object):
    def __init__(self, number, supplier=None):
        self.supplier = supplier
        self.number = number

    def __repr__(self):
        if self.supplier:
            return "(supplier={}, number={})".format(self.supplier, self.number)
        else:
            return "(number={})".format(self.number)


class MyPart(object):
    def __init__(self, mpn: MyPartNumber, spn: Optional[MyPartNumber] = None):
        self.mpn = mpn
        self.spn = spn

    def __repr__(self):
        mpn = self.mpn if self.mpn else ""
        spn = self.spn if self.spn else ""
        return "(Part: mpn={}, spn={})".format(mpn, spn)


suppliers = [("supplier-a", "SA-"), ("supplier-b", "SB-"), ("supplier-c", None)]
mpns = ["RES" + value for value in ["100", "101", "103"]]


def test_basic():
    db = ObjDb()
    build_db(db)

    assert len(db) == 11


def build_db(db: ObjDb):
    db.add(MyPart(MyPartNumber("RES100")))
    db.add(MyPart(MyPartNumber("RES101")))
    db.add(MyPart(MyPartNumber("RES102")))
    db.add(MyPart(MyPartNumber("RES103")))
    db.add(MyPart(MyPartNumber("RES104")))

    db.add(MyPart(MyPartNumber("RES100"), MyPartNumber("SA-RES100", "supplier-a")))
    db.add(MyPart(MyPartNumber("RES101"), MyPartNumber("SA-RES101", "supplier-a")))
    db.add(MyPart(MyPartNumber("RES102"), MyPartNumber("SA-RES102", "supplier-a")))
    db.add(MyPart(MyPartNumber("RES103"), MyPartNumber("SA-RES103", "supplier-a")))

    db.add(MyPart(MyPartNumber("RES101"), MyPartNumber("SB-RES101", "supplier-b")))
    db.add(MyPart(MyPartNumber("RES102"), MyPartNumber("SB-RES102", "supplier-b")))


def test_index():
    db = ObjDb()
    mpn_idx = db.add_index("mpn", lambda part: part.mpn.number if part.mpn else None)
    spn_idx = db.add_unique_index("spn", lambda part: part.spn.number if part.spn else None)
    supplier_pn = db.add_multi_index("supplier_pn",
                                     lambda part: (part.spn.supplier, part.spn.number) if part.spn else None)
    build_db(db)

    assert len(mpn_idx.values()) == 5
    tmp = mpn_idx.get("RES100")
    assert len(mpn_idx.get("RES100")) == 2

    # for supplier, spn in spn_idx.items():
    #     print("SPN: {}={}".format(supplier, spn))
    assert len(spn_idx.items()) == (len(suppliers) - 1) * len(mpns)

    assert len((supplier_pn.items())) == 2
    assert "supplier-a" in supplier_pn
    assert "SA-RES103" in supplier_pn.get("supplier-a")
    assert "supplier-b" in supplier_pn


@pytest.mark.skipif('False')
def test_skip():
    pass