From c73c4d4edf88d2af5636962f1e4710be17516ea1 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Mon, 17 May 2021 14:34:25 +0300
Subject: tests: test_phenotypes: New test cases for loading phenotypes

---
 tests/unit/db/test_phenotypes.py | 41 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 tests/unit/db/test_phenotypes.py

diff --git a/tests/unit/db/test_phenotypes.py b/tests/unit/db/test_phenotypes.py
new file mode 100644
index 0000000..8b810fe
--- /dev/null
+++ b/tests/unit/db/test_phenotypes.py
@@ -0,0 +1,41 @@
+"""Tests for db/phenotypes.py"""
+from unittest import TestCase
+from unittest import mock
+
+from gn3.db.phenotypes import Phenotype
+from gn3.db.phenotypes import update_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_phenotype(
+            db_mock, 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_phenotype(
+                db_mock, data=Phenotype(
+                    pre_pub_description="Test Pre Pub",
+                    submitter="Rob",
+                    post_pub_description="Test Post Pub"),
+                where=Phenotype(id_=1)), 1)
+            cursor.execute.assert_called_once_with(
+                "UPDATE Phenotype SET "
+                "Pre_publication_description = "
+                "'Test Pre Pub', "
+                "Post_publication_description = "
+                "'Test Post Pub', Submitter = 'Rob' "
+                "WHERE id = '1'"
+            )
-- 
cgit v1.2.3