Minor tests refactoring.

diff --git a/tests/conditions.py b/tests/conditions.py
index f825ba460400be22e650608bf0ecafe51e94e990..dabcf06f767d02732b972d5b4bf40972f7d9fe66 100644 (file)
--- a/tests/conditions.py
+++ b/tests/conditions.py
@@ -47,7 +47,7 @@ class TestsWithDB(TestCaseWithDB):
 
     def test_Condition_from_table_row(self) -> None:
         """Test .from_table_row() properly reads in class from DB"""
-        self.check_from_table_row(1)
+        self.check_from_table_row()
         c = self.versioned_condition()
         c.save(self.db_conn)
         assert isinstance(c.id_, int)

diff --git a/tests/days.py b/tests/days.py
index c143f985b22b7bdd41d65e21409c2016274de54c..1f0e55d871b282881dc0fee2f7e2fc91a591d243 100644 (file)
--- a/tests/days.py
+++ b/tests/days.py
@@ -44,12 +44,12 @@ class TestsWithDB(TestCaseWithDB):
 
     def test_Day_saving_and_caching(self) -> None:
         """Test .save/.save_core."""
-        kwargs = {'date': '2024-01-01', 'comment': 'foo'}
+        kwargs = {'date': self.default_ids[0], 'comment': 'foo'}
         self.check_saving_and_caching(**kwargs)
 
     def test_Day_from_table_row(self) -> None:
         """Test .from_table_row() properly reads in class from DB"""
-        self.check_from_table_row('2024-01-01')
+        self.check_from_table_row()
 
     def test_Day_by_id(self) -> None:
         """Test .by_id()."""

diff --git a/tests/utils.py b/tests/utils.py
index a826c16a5adb1d24222ef1f4b42f5cefc711baf8..61dbb36b949ee3fdfd5a38dcada155a5ab18a924 100644 (file)
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -79,8 +79,9 @@ class TestCaseWithDB(TestCase):
         self.assertEqual(self.checked_class(id2), by_id_created)
         self.check_storage([obj])
 
-    def check_from_table_row(self, id_: int | str) -> None:
+    def check_from_table_row(self) -> None:
         """Test .from_table_row() properly reads in class from DB"""
+        id_ = self.default_ids[0]
         obj = self.checked_class(id_)  # pylint: disable=not-callable
         obj.save(self.db_conn)
         assert isinstance(obj.id_, (str, int))