From 0e5f9afc3927cee14acab491c309080fd0330155 Mon Sep 17 00:00:00 2001
From: "Salvador E. Tropea" <salvador@inti.gob.ar>
Date: Sat, 1 Aug 2020 12:29:49 -0300
Subject: [PATCH] Added test for "digikey link" in the internal BoM.

---
 tests/test_plot/test_int_bom.py               | 30 +++++++++++++++++++
 .../int_bom_digikey_link.kiplot.yaml          | 20 +++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 tests/yaml_samples/int_bom_digikey_link.kiplot.yaml

diff --git a/tests/test_plot/test_int_bom.py b/tests/test_plot/test_int_bom.py
index 7236f28d..bc3bc7b4 100644
--- a/tests/test_plot/test_int_bom.py
+++ b/tests/test_plot/test_int_bom.py
@@ -200,3 +200,33 @@ def test_int_bom_datasheet_link():
         assert 'pdf' in c
         logging.debug(c + ' OK')
     ctx.clean_up()
+
+
+def test_int_bom_digikey_link():
+    prj = 'links'
+    ext = 'html'
+    ctx = context.TestContext('test_int_bom_digikey_link', prj, 'int_bom_digikey_link', BOM_DIR)
+    ctx.run(no_board_file=True, extra=['-e', os.path.join(ctx.get_board_dir(), prj+'.sch')])
+    out = prj + '.' + ext
+    rows, headers = ctx.load_html(out)
+    # Test we got the normal and DNF tables
+    assert len(rows) == 2
+    assert len(headers) == 2
+    # Test both tables has the same headings and they are the expected
+    assert headers[0] == headers[1]
+    head_no_comp = deepcopy(KIBOM_TEST_HEAD)
+    assert headers[0] == ['', 'References', 'Part', 'Value', 'Quantity Per PCB', 'digikey#', 'manf#']
+    # Look for reference and quantity columns
+    ref_column = headers[0].index(REF_COLUMN_NAME)
+    dk_column = headers[0].index('digikey#')
+    # Check the normal table
+    check_kibom_test_netlist(rows[0], ref_column, 2, ['C1'], ['J1', 'J2','R1'])
+    # Check the DNF table
+    check_kibom_test_netlist(rows[1], ref_column, 1, ['J1', 'J2','R1'], ['C1'])
+    # Check the digikey link
+    parts = get_column(rows[0]+rows[1], dk_column, False)
+    for c in parts:
+        assert c.strip().startswith('<a href')
+        assert 'digikey' in c
+        logging.debug(c + ' OK')
+    ctx.clean_up()
diff --git a/tests/yaml_samples/int_bom_digikey_link.kiplot.yaml b/tests/yaml_samples/int_bom_digikey_link.kiplot.yaml
new file mode 100644
index 00000000..91c8c671
--- /dev/null
+++ b/tests/yaml_samples/int_bom_digikey_link.kiplot.yaml
@@ -0,0 +1,20 @@
+# Example KiPlot config file
+kiplot:
+  version: 1
+
+outputs:
+  - name: 'bom_internal'
+    comment: "Bill of Materials in HTML format"
+    type: bom
+    dir: BoM
+    options:
+      format: HTML
+      digikey_link: 'digikey#'
+      output: '%f.%x'
+      columns:
+        - References
+        - Part
+        - Value
+        - Quantity Per PCB
+        - digikey#
+        - manf#