aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/db/test_sample_data.py
diff options
context:
space:
mode:
authorBonfaceKilz2022-03-10 13:27:06 +0300
committerBonfaceKilz2022-03-12 15:33:09 +0300
commit7641343e90f9a7f03cf0baf30818675efac6d7f3 (patch)
tree723b04515a5ea206a385b699a54d685ba3cd26a3 /tests/unit/db/test_sample_data.py
parent9b714166cf0ebbbb098a52b97d7c9b4e3ec13a2e (diff)
downloadgenenetwork3-7641343e90f9a7f03cf0baf30818675efac6d7f3.tar.gz
Create action dict that's created when updating data
* gn3/db/sample_data.py (__extract_actions): An update on a vector of data can contain: inserts, deletes and updates. This functions extracts these actions during an update. * tests/unit/db/test_sample_data.py (test_extract_actions): Add test-case for the above.
Diffstat (limited to 'tests/unit/db/test_sample_data.py')
-rw-r--r--tests/unit/db/test_sample_data.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/tests/unit/db/test_sample_data.py b/tests/unit/db/test_sample_data.py
index 4c03dbd..4015ba5 100644
--- a/tests/unit/db/test_sample_data.py
+++ b/tests/unit/db/test_sample_data.py
@@ -2,6 +2,7 @@ import pytest
from gn3.db.sample_data import insert_sample_data
from gn3.db.sample_data import delete_sample_data
+from gn3.db.sample_data import __extract_actions
@pytest.mark.unit_test
@@ -73,3 +74,23 @@ def test_delete_sample_data(mocker):
(strain_id, "Sex", inbredset_id)),
]
cursor.execute.assert_has_calls(calls, any_order=False)
+
+
+@pytest.mark.unit_test
+def test_extract_actions():
+ assert (__extract_actions(original_data="BXD1,18,x,0,x",
+ updated_data="BXD1,x,2,1,F",
+ csv_header="Strain Name,Value,SE,Count,Sex") ==
+ {
+ "delete": {"data": "18", "csv_header": "Value"},
+ "insert": {"data": "2,F", "csv_header": "SE,Sex"},
+ "update": {"data": "1", "csv_header": "Count"},
+ })
+ assert(__extract_actions(original_data="BXD1,18,x,0,x",
+ updated_data="BXD1,19,2,1,F",
+ csv_header="Strain Name,Value,SE,Count,Sex") ==
+ {
+ "delete": None,
+ "insert": {"data": "2,F", "csv_header": "SE,Sex"},
+ "update": {"data": "19,1", "csv_header": "Value,Count"},
+ })