Use case attribute id inside brackets if present during insertions

* gn3/db/sample_data.py (insert_sample_data): If an id is present in the column header, use it.
* tests/unit/db/test_sample_data.py (test_insert_sample_data): Update tests to
capture the above.

1 files changed, 35 insertions, 24 deletions

diff --git a/tests/unit/db/test_sample_data.py b/tests/unit/db/test_sample_data.py
index af00393..e7d7fb0 100644
--- a/tests/unit/db/test_sample_data.py
+++ b/tests/unit/db/test_sample_data.py
@@ -21,6 +21,7 @@ def test_insert_sample_data(mocker):
                 19,
             ],
             0,
+            0,
         )
         mocker.patch(
             "gn3.db.sample_data.get_sample_data_ids",
@@ -29,45 +30,55 @@ def test_insert_sample_data(mocker):
         insert_sample_data(
             conn=mock_conn,
             trait_name=35,
-            data="BXD1,18,3,0,M",
-            csv_header="Strain Name,Value,SE,Count,Sex",
+            data="BXD1,18,3,0,Red,M",
+            csv_header="Strain Name,Value,SE,Count,Color,Sex (13)",
             phenotype_id=10007,
         )
         calls = [
             mocker.call(
-                "SELECT Id FROM PublishData where Id = %s "
-                "AND StrainId = %s",
-                (data_id, strain_id),
+                "SELECT Id FROM PublishData where Id = %s AND StrainId = %s",
+                (17373, 1),
+            ),
+            mocker.call(
+                "INSERT INTO PublishData (StrainId, Id, value) "
+                "VALUES (%s, %s, %s)",
+                (1, 17373, "18"),
+            ),
+            mocker.call(
+                "INSERT INTO PublishSE (StrainId, DataId, error) VALUES "
+                "(%s, %s, %s)",
+                (1, 17373, "3"),
             ),
             mocker.call(
-                "INSERT INTO PublishData "
-                "(StrainId, Id, value) VALUES (%s, %s, %s)",
-                (strain_id, data_id, "18"),
+                "INSERT INTO NStrain (StrainId, DataId, count) VALUES "
+                "(%s, %s, %s)",
+                (1, 17373, "0"),
             ),
             mocker.call(
-                "INSERT INTO PublishSE "
-                "(StrainId, DataId, error) VALUES (%s, %s, %s)",
-                (strain_id, data_id, "3"),
+                "SELECT Id FROM CaseAttribute WHERE Name = %s", ("Color",)
             ),
             mocker.call(
-                "INSERT INTO NStrain "
-                "(StrainId, DataId, count) VALUES (%s, %s, %s)",
-                (strain_id, data_id, "0"),
+                "SELECT StrainId FROM CaseAttributeXRefNew WHERE "
+                "StrainId = %s AND CaseAttributeId = %s AND InbredSetId = %s",
+                (1, 19, 20),
             ),
             mocker.call(
-                "SELECT Id FROM CaseAttribute WHERE Name = %s", ("Sex",)
+                "INSERT INTO CaseAttributeXRefNew (StrainId, "
+                "CaseAttributeId, Value, InbredSetId) VALUES "
+                "(%s, %s, %s, %s)",
+                (1, 19, "Red", 20),
             ),
             mocker.call(
-                "SELECT StrainId FROM CaseAttributeXRefNew "
-                "WHERE StrainId = %s AND "
-                "CaseAttributeId = %s AND InbredSetId = %s",
-                (strain_id, 19, inbredset_id),
+                "SELECT StrainId FROM CaseAttributeXRefNew WHERE "
+                "StrainId = %s AND CaseAttributeId = %s AND "
+                "InbredSetId = %s",
+                (1, "13", 20),
             ),
             mocker.call(
-                "INSERT INTO CaseAttributeXRefNew "
-                "(StrainId, CaseAttributeId, Value, "
-                "InbredSetId) VALUES (%s, %s, %s, %s)",
-                (strain_id, 19, "M", inbredset_id),
+                "INSERT INTO CaseAttributeXRefNew (StrainId, "
+                "CaseAttributeId, Value, InbredSetId) VALUES "
+                "(%s, %s, %s, %s)",
+                (1, "13", "M", 20),
             ),
         ]
         cursor.execute.assert_has_calls(calls, any_order=False)
@@ -206,7 +217,7 @@ def test_update_sample_data(mocker):
                     "CaseAttribute WHERE Name = %s) "
                     "AND InbredSetId = %s",
                     ("Green", strain_id, "Color", inbredset_id),
-                )
+                ),
             ],
             any_order=False,
         )