aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/db
diff options
context:
space:
mode:
authorBonfaceKilz2021-07-29 22:15:36 +0300
committerGitHub2021-07-29 22:15:36 +0300
commit00278bc237c90b4ac276171a32bbe57e056644b6 (patch)
treef6ff2fd64cd288d75af4900cb7b2cd984f598ce0 /tests/unit/db
parent2ccbf1844afe352e23af7ff958ec2b0694cd87ea (diff)
parent8d83ba7298c33e54ab204effe2a4241ce848add9 (diff)
downloadgenenetwork3-00278bc237c90b4ac276171a32bbe57e056644b6.tar.gz
Merge pull request #30 from genenetwork/Feature/Update-db-from-csv-data
Feature/update db from csv data
Diffstat (limited to 'tests/unit/db')
-rw-r--r--tests/unit/db/test_traits.py33
1 files changed, 32 insertions, 1 deletions
diff --git a/tests/unit/db/test_traits.py b/tests/unit/db/test_traits.py
index e3c5c28..c8f28b5 100644
--- a/tests/unit/db/test_traits.py
+++ b/tests/unit/db/test_traits.py
@@ -11,7 +11,9 @@ from gn3.db.traits import (
retrieve_temp_trait_info,
retrieve_trait_dataset_name,
retrieve_publish_trait_info,
- retrieve_probeset_trait_info)
+ retrieve_probeset_trait_info,
+ update_sample_data)
+
class TestTraitsDBFunctions(TestCase):
"Test cases for traits functions"
@@ -112,3 +114,32 @@ class TestTraitsDBFunctions(TestCase):
trait_type, trait_name, trait_dataset_id,
trait_dataset_name, db_mock),
tuple())
+
+ def test_update_sample_data(self):
+ """Test that the SQL queries when calling update_sample_data are called with
+ the right calls.
+
+ """
+ db_mock = mock.MagicMock()
+
+ STRAIN_ID_SQL: str = "UPDATE Strain SET Name = %s WHERE Id = %s"
+ PUBLISH_DATA_SQL: str = ("UPDATE PublishData SET value = %s "
+ "WHERE StrainId = %s AND Id = %s")
+ PUBLISH_SE_SQL: str = ("UPDATE PublishSE SET error = %s "
+ "WHERE StrainId = %s AND DataId = %s")
+ N_STRAIN_SQL: str = ("UPDATE NStrain SET count = %s "
+ "WHERE StrainId = %s AND DataId = %s")
+
+ with db_mock.cursor() as cursor:
+ type(cursor).rowcount = 1
+ self.assertEqual(update_sample_data(
+ conn=db_mock, strain_name="BXD11",
+ strain_id=10, publish_data_id=8967049,
+ value=18.7, error=2.3, count=2),
+ (1, 1, 1, 1))
+ cursor.execute.assert_has_calls(
+ [mock.call(STRAIN_ID_SQL, ('BXD11', 10)),
+ mock.call(PUBLISH_DATA_SQL, (18.7, 10, 8967049)),
+ mock.call(PUBLISH_SE_SQL, (2.3, 10, 8967049)),
+ mock.call(N_STRAIN_SQL, (2, 10, 8967049))]
+ )