From fa85d46af0b91477cf354947df628af0dc0d2800 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Thu, 28 Mar 2019 16:38:50 +0100
Subject: ee.xsd: o Renaming <part-uri> to <part-reference>. o Adding
 <supplier> on <part>, removing from <supplier-part-number>. A   part can have
 exactly one part. create-order: o Creating anonymous part objects, with two
 references, one schematic   reference and one part-uri reference to the
 selected part. o Redoing how the order is calculated with the new ObjDb
 structure. ee.part.Part: o Absorbing bom_file_utils into Part. Much better
 wrapper object around   the xml goop.

---
 test/test_db.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 test/test_db.py

(limited to 'test')

diff --git a/test/test_db.py b/test/test_db.py
new file mode 100644
index 0000000..7fc3cae
--- /dev/null
+++ b/test/test_db.py
@@ -0,0 +1,82 @@
+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
-- 
cgit v1.2.3