poistojen testailua

TheJiahao · Dec 8, 2023 · 88579fa · 88579fa
1 parent f8b508b
commit 88579fa
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/tests/repositories/test_cite_repository.py b/src/tests/repositories/test_cite_repository.py
@@ -220,3 +220,24 @@ def test_remove_cite_only_removes_one_cite(self):
         self.assertEqual(len(cite_data), 1)
         self.assertEqual(len(authors), 1)
         self.assertEqual(len(fields), 1)
+
+    def test_remove_all_cites(self):
+        self.repository.add_cite(
+            Cite("amazingBook", "book", ["Ama Zing"], {"title": "An amazing book"})
+        )
+        self.repository.add_cite(
+            Cite("goodBook", "book", ["Good Book"], {"title": "A good book"})
+        )
+        self.repository.remove_cite("amazingBook")
+
+        self.repository.remove_all_cites()
+
+        cite_data = database.cursor.execute("SELECT * FROM Cites").fetchall()
+        authors = database.cursor.execute("SELECT * FROM Authors").fetchall()
+        fields = database.cursor.execute("SELECT * FROM Fields").fetchall()
+
+        self.assertEqual(len(cite_data), 0)
+        self.assertEqual(len(authors), 0)
+        self.assertEqual(len(fields), 0)
+
+
diff --git a/src/tests/services/test_logic.py b/src/tests/services/test_logic.py
@@ -43,3 +43,7 @@ def test_filter_cites_calls_filter_by_name_when_name_searched(self):
     def test_remove_cite(self):
         self.logic.remove_cite("123")
         self.repository_mock.remove_cite.assert_called()
+
+    def test_remova_all_cites(self):
+        self.logic.remove_all_cites()
+        self.repository_mock.remove_all_cites.assert_called()