X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=tests%2Fconditions.py;h=9c95206aab63cbdd045aaedf1f415ba6779d8afa;hb=c4ccb784bb3a83c1c614c9bab7fc007ee17f6615;hp=51f7cc2d3e5b7a657e5baa96ab290b28cf4fe372;hpb=9ad40c43627334d7294c07bf55d196dd6760cfde;p=plomtask diff --git a/tests/conditions.py b/tests/conditions.py index 51f7cc2..9c95206 100644 --- a/tests/conditions.py +++ b/tests/conditions.py @@ -1,55 +1,47 @@ """Test Conditions module.""" -from unittest import TestCase -from tests.utils import TestCaseWithDB, TestCaseWithServer +from tests.utils import TestCaseWithDB, TestCaseWithServer, TestCaseSansDB from plomtask.conditions import Condition from plomtask.processes import Process from plomtask.todos import Todo -from plomtask.exceptions import NotFoundException, HandledException +from plomtask.exceptions import HandledException -class TestsSansDB(TestCase): +class TestsSansDB(TestCaseSansDB): """Tests requiring no DB setup.""" + checked_class = Condition def test_Condition_id_setting(self) -> None: """Test .id_ being set and its legal range being enforced.""" - with self.assertRaises(HandledException): - Condition(0) - condition = Condition(5) - self.assertEqual(condition.id_, 5) + self.check_id_setting() + + def test_Condition_versioned_defaults(self) -> None: + """Test defaults of VersionedAttributes.""" + self.check_versioned_defaults({ + 'title': 'UNNAMED', + 'description': ''}) class TestsWithDB(TestCaseWithDB): """Tests requiring DB, but not server setup.""" + checked_class = Condition - def check_storage(self, content: list[Condition]) -> None: - """Test cache and DB equal content.""" - expected_cache = {} - for item in content: - expected_cache[item.id_] = item - self.assertEqual(Condition.get_cache(), expected_cache) - db_found: list[Condition] = [] - for item in content: - assert isinstance(item.id_, int) - for row in self.db_conn.row_where(Condition.table_name, 'id', - item.id_): - db_found += [Condition.from_table_row(self.db_conn, row)] - self.assertEqual(sorted(content), sorted(db_found)) - - def test_Condition_saving_and_caching(self) -> None: - """Test .save/.save_core.""" - c = Condition(None, False) + def versioned_condition(self) -> Condition: + """Create Condition with some VersionedAttribute values.""" + c = Condition(None) c.title.set('title1') c.title.set('title2') c.description.set('desc1') c.description.set('desc2') - # check object init itself doesn't store anything yet - self.check_storage([]) - # check saving stores in cache and DB + return c + + def test_Condition_saving_and_caching(self) -> None: + """Test .save/.save_core.""" + kwargs = {'id_': 1, 'is_active': False} + self.check_saving_and_caching(**kwargs) + # check .id_ set if None, and versioned attributes too + c = self.versioned_condition() c.save(self.db_conn) - self.check_storage([c]) - # check attributes set properly (and not unset by saving) - self.assertEqual(c.id_, 1) - self.assertEqual(c.is_active, False) + self.assertEqual(c.id_, 2) self.assertEqual(sorted(c.title.history.values()), ['title1', 'title2']) self.assertEqual(sorted(c.description.history.values()), @@ -57,18 +49,12 @@ class TestsWithDB(TestCaseWithDB): def test_Condition_from_table_row(self) -> None: """Test .from_table_row() properly reads in class from DB""" - c = Condition(1, True) - c.title.set('title1') - c.title.set('title2') - c.description.set('desc1') - c.description.set('desc2') + self.check_from_table_row() + c = self.versioned_condition() c.save(self.db_conn) assert isinstance(c.id_, int) for row in self.db_conn.row_where(Condition.table_name, 'id', c.id_): retrieved = Condition.from_table_row(self.db_conn, row) - assert isinstance(retrieved, Condition) - self.assertEqual(c, retrieved) - self.assertEqual({c.id_: c}, Condition.get_cache()) # pylint: disable=no-member self.assertEqual(sorted(retrieved.title.history.values()), ['title1', 'title2']) @@ -78,49 +64,24 @@ class TestsWithDB(TestCaseWithDB): def test_Condition_by_id(self) -> None: """Test .by_id(), including creation.""" - # check failure if not yet saved - c = Condition(3, False) - with self.assertRaises(NotFoundException): - Condition.by_id(self.db_conn, 3) - # check identity of saved and retrieved - c.save(self.db_conn) - self.assertEqual(c, Condition.by_id(self.db_conn, 3)) - # check create=True acts like normal instantiation (sans saving) - by_id_created = Condition.by_id(self.db_conn, 4, create=True) - self.assertEqual(Condition(4), by_id_created) - self.check_storage([c]) + self.check_by_id() def test_Condition_all(self) -> None: """Test .all().""" - c_1 = Condition(None, False) - # check pre-save .all() returns empty list - self.assertEqual(Condition.all(self.db_conn), []) - # check .save() fills .all() result - c_1.save(self.db_conn) - self.assertEqual(Condition.all(self.db_conn), [c_1]) - c_2 = Condition(None, True) - c_2.save(self.db_conn) - self.assertEqual(sorted(Condition.all(self.db_conn)), - sorted([c_1, c_2])) + self.check_all() def test_Condition_singularity(self) -> None: """Test pointers made for single object keep pointing to it.""" - c = Condition(None, False) - c.save(self.db_conn) - c.is_active = True - retrieved = Condition.by_id(self.db_conn, 1) - self.assertEqual(True, retrieved.is_active) + self.check_singularity('is_active', True) + + def test_Condition_versioned_attributes_singularity(self) -> None: + """Test behavior of VersionedAttributes on saving (with .title).""" + self.check_versioned_singularity() def test_Condition_remove(self) -> None: """Test .remove() effects on DB and cache.""" - # check only saved item can be removed - c = Condition(None, False) - with self.assertRaises(HandledException): - c.remove(self.db_conn) - c.save(self.db_conn) - c.remove(self.db_conn) - self.check_storage([]) - # check guard against deleting dependencies of other classes + self.check_remove() + c = Condition(None) proc = Process(None) todo = Todo(None, proc, False, '2024-01-01') for depender in (proc, todo):