added tests for the new method

Author: anngoroshi

test2text/services/db/tables/requirements.py

@@ -64,6 +64,7 @@ def insert(
         else:
             return None
 
+    @property
     def count(self) -> int:
         """
         Returns the number of entries in the Requirements table.

tests/test_db/test_tables/test_annos_to_reqs.py

@@ -50,3 +50,11 @@ def test_insert_wrong_requirement(self):
         count_after = self.db.annos_to_reqs.count
         self.assertFalse(inserted)  # Should fail due to foreign key constraint
         self.assertEqual(count_before, count_after)
+
+    def test_count(self):
+        count_before = self.db.annos_to_reqs.count
+        self.db.annos_to_reqs.insert(self.anno1, self.req1, 1)
+        self.db.annos_to_reqs.insert(self.anno2, self.req2, 1)
+        count_after = self.db.annos_to_reqs.count
+        self.assertEqual(count_after, count_before + 2)
+

tests/test_db/test_tables/test_annotations.py

@@ -81,3 +81,9 @@ def test_set_embedding(self):
             read_embedding = unpack_float32(result[0])
             self.assertEqual(len(read_embedding), self.db.annotations.embedding_size)
             self.assertEqual(round_vector(read_embedding), round_vector(new_embedding))
+
+    def test_count(self):
+        count_before = self.db.annotations.count
+        self.db.annotations.insert("Test Summary 11")
+        count_after = self.db.annotations.count
+        self.assertEqual(count_after, count_before + 1)

tests/test_db/test_tables/test_cases_to_annos.py

@@ -50,3 +50,10 @@ def test_insert_wrong_annotation(self):
         count_after = self.db.cases_to_annos.count
         self.assertFalse(inserted)  # Should fail due to foreign key constraint
         self.assertEqual(count_before, count_after)
+
+    def test_count(self):
+        count_before = self.db.cases_to_annos.count
+        self.db.cases_to_annos.insert(self.case1, self.anno1)
+        self.db.cases_to_annos.insert(self.case2, self.anno2)
+        count_after = self.db.cases_to_annos.count
+        self.assertEqual(count_after, count_before + 2)

tests/test_db/test_tables/test_requirements.py

@@ -60,3 +60,10 @@ def test_insert_long_embedding(self):
         long_embedding = [0.1] * (self.db.requirements.embedding_size + 1)
         id1 = self.db.requirements.insert("Test Requirement 7", long_embedding)
         self.assertIsNone(id1)
+
+    def test_count(self):
+        count_before = self.db.requirements.count
+        self.db.requirements.insert("Test Requirement 8")
+        count_after = self.db.requirements.count
+        self.assertEqual(count_after, count_before + 1)
+

tests/test_db/test_tables/test_test_cases.py

@@ -61,3 +61,9 @@ def test_insert_long_embedding(self):
             "Test Script 14", "Test Case 14", long_embedding
         )
         self.assertIsNone(id1)
+
+    def test_count(self):
+        count_before = self.db.test_cases.count
+        self.db.test_cases.insert("Test Script 15", "Test Case 15")
+        count_after = self.db.test_cases.count
+        self.assertEqual(count_after, count_before + 1)