feat: Fix test_repository

FaceTheFacts · Oct 20, 2023 · 5712990 · 5712990
1 parent 9a403d5
commit 5712990
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/unit/api/test_repository.py b/tests/unit/api/test_repository.py
@@ -118,8 +118,7 @@ def test_repository_can_list_all_party_donations(self, session):
         assert retrieved_list[0].date == date(2020, 1, 6)
         assert retrieved_list[-1].date == date(2020, 1, 1)
         # Cleanup
-        session.execute(text("DELETE FROM party_donation"))
-        session.commit()
+        delete_party_donation(session)
 
     # list
     def test_repository_can_list_all_party_donations_with_party_ids(self, session):
@@ -133,5 +132,4 @@ def test_repository_can_list_all_party_donations_with_party_ids(self, session):
         assert retrieved_list[0].date == date(2020, 1, 4)
         assert retrieved_list[-1].date == date(2020, 1, 1)
         # Cleanup
-        session.execute(text("DELETE FROM party_donation"))
-        session.commit()
+        delete_party_donation(session)