1 files changed, 73 insertions, 0 deletions
diff --git a/tests/unit/db/test_db.py b/tests/unit/db/test_db.py
new file mode 100644
index 0000000..a3b1d71
--- /dev/null
+++ b/tests/unit/db/test_db.py
@@ -0,0 +1,73 @@
+"""Tests for db/phenotypes.py"""
+from unittest import TestCase
+from unittest import mock
+
+from gn3.db import fetchone
+from gn3.db import update
+from gn3.db import diff_from_dict
+from gn3.db.phenotypes import Phenotype
+
+
+class TestPhenotypes(TestCase):
+    """Test cases for fetching chromosomes"""
+
+    def test_update_phenotype_with_no_data(self):
+        """Test that a phenotype is updated correctly if an empty Phenotype dataclass
+        is provided
+
+        """
+        db_mock = mock.MagicMock()
+        self.assertEqual(update(
+            conn=db_mock, table="Phenotype",
+            data=Phenotype(), where=Phenotype()), None)
+
+    def test_update_phenotype_with_data(self):
+        """
+        Test that a phenotype is updated correctly if some
+        data is provided
+        """
+        db_mock = mock.MagicMock()
+        with db_mock.cursor() as cursor:
+            type(cursor).rowcount = 1
+            self.assertEqual(update(
+                conn=db_mock, table="Phenotype",
+                data=Phenotype(
+                    pre_pub_description="Test Pre Pub",
+                    submitter="Rob",
+                    post_pub_description="Test Post Pub"),
+                where=Phenotype(id_=1, owner="Rob")), 1)
+            cursor.execute.assert_called_once_with(
+                "UPDATE Phenotype SET "
+                "Pre_publication_description = %s, "
+                "Post_publication_description = %s, "
+                "Submitter = %s WHERE id = %s AND Owner = %s",
+                ('Test Pre Pub', 'Test Post Pub', 'Rob', 1, 'Rob'))
+
+    def test_fetch_phenotype(self):
+        """Test that a single phenotype is fetched properly
+
+        """
+        db_mock = mock.MagicMock()
+        with db_mock.cursor() as cursor:
+            test_data = (
+                35, "Test pre-publication", "Test post-publication",
+                "Original description A", "cm^2", "pre-abbrev",
+                "post-abbrev", "LAB001", "R. W.", "R. W.", "R. W."
+            )
+            cursor.fetchone.return_value = test_data
+            phenotype = fetchone(db_mock,
+                                 "Phenotype",
+                                 where=Phenotype(id_=35, owner="Rob"))
+            self.assertEqual(phenotype.id_, 35)
+            self.assertEqual(phenotype.pre_pub_description,
+                             "Test pre-publication")
+            cursor.execute.assert_called_once_with(
+                "SELECT * FROM Phenotype WHERE id = %s AND Owner = %s",
+                (35, 'Rob'))
+
+    def test_diff_from_dict(self):
+        """Test that a correct diff is generated"""
+        self.assertEqual(diff_from_dict({"id": 1, "data": "a"},
+                                        {"id": 2, "data": "b"}),
+                         {"id": {"old": 1, "new": 2},
+                          "data": {"old": "a", "new": "b"}})