"""Test Conditions module."""
+from unittest import TestCase
from tests.utils import TestCaseWithDB, TestCaseWithServer
from plomtask.conditions import Condition
from plomtask.processes import Process
-from plomtask.exceptions import NotFoundException, HandledException
+from plomtask.todos import Todo
+from plomtask.exceptions import HandledException
+
+
+class TestsSansDB(TestCase):
+ """Tests requiring no DB setup."""
+
+ 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)
class TestsWithDB(TestCaseWithDB):
"""Tests requiring DB, but not server setup."""
+ checked_class = Condition
+ default_ids = (1, 2, 3)
+
+ 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')
+ 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.assertEqual(c.id_, 2)
+ self.assertEqual(sorted(c.title.history.values()),
+ ['title1', 'title2'])
+ self.assertEqual(sorted(c.description.history.values()),
+ ['desc1', 'desc2'])
+
+ def test_Condition_from_table_row(self) -> None:
+ """Test .from_table_row() properly reads in class from DB"""
+ self.check_from_table_row(1)
+ 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)
+ # pylint: disable=no-member
+ self.assertEqual(sorted(retrieved.title.history.values()),
+ ['title1', 'title2'])
+ # pylint: disable=no-member
+ self.assertEqual(sorted(retrieved.description.history.values()),
+ ['desc1', 'desc2'])
def test_Condition_by_id(self) -> None:
- """Test creation and findability."""
- condition = Condition(None, False)
- condition.save(self.db_conn)
- self.assertEqual(Condition.by_id(self.db_conn, 1), condition)
- with self.assertRaises(NotFoundException):
- self.assertEqual(Condition.by_id(self.db_conn, 0), condition)
- with self.assertRaises(NotFoundException):
- self.assertEqual(Condition.by_id(self.db_conn, 2), condition)
+ """Test .by_id(), including creation."""
+ self.check_by_id()
def test_Condition_all(self) -> None:
"""Test .all()."""
- self.assertEqual(Condition.all(self.db_conn), [])
- condition_1 = Condition(None, False)
- condition_1.save(self.db_conn)
- self.assertEqual(Condition.all(self.db_conn), [condition_1])
- condition_2 = Condition(None, False)
- condition_2.save(self.db_conn)
- self.assertEqual(Condition.all(self.db_conn), [condition_1,
- condition_2])
+ self.check_all()
def test_Condition_singularity(self) -> None:
"""Test pointers made for single object keep pointing to it."""
- condition_1 = Condition(None, False)
- condition_1.save(self.db_conn)
- condition_1.is_active = True
- condition_retrieved = Condition.by_id(self.db_conn, 1)
- self.assertEqual(True, condition_retrieved.is_active)
-
- def test_Condition_removal(self) -> None:
- """Test removal of Condition."""
- cond = Condition(None, False)
- cond.save(self.db_conn)
- assert isinstance(cond.id_, int)
+ self.check_singularity('is_active', True)
+
+ def test_Condition_remove(self) -> None:
+ """Test .remove() effects on DB and cache."""
+ self.check_remove()
+ c = Condition(None)
proc = Process(None)
- proc.save(self.db_conn)
- proc.set_conditions(self.db_conn, [cond.id_], 'conditions')
- proc.save(self.db_conn)
- with self.assertRaises(HandledException):
- cond.remove(self.db_conn)
- proc.set_conditions(self.db_conn, [], 'conditions')
- proc.save(self.db_conn)
- cond.remove(self.db_conn)
- self.assertEqual(Condition.all(self.db_conn), [])
+ todo = Todo(None, proc, False, '2024-01-01')
+ for depender in (proc, todo):
+ assert hasattr(depender, 'save')
+ assert hasattr(depender, 'set_conditions')
+ c.save(self.db_conn)
+ depender.save(self.db_conn)
+ depender.set_conditions(self.db_conn, [c.id_], 'conditions')
+ depender.save(self.db_conn)
+ with self.assertRaises(HandledException):
+ c.remove(self.db_conn)
+ depender.set_conditions(self.db_conn, [], 'conditions')
+ depender.save(self.db_conn)
+ c.remove(self.db_conn)
class TestsWithServer(TestCaseWithServer):