from tests.utils import TestCaseWithDB, TestCaseWithServer
from plomtask.processes import Process, ProcessStep, ProcessStepsNode
from plomtask.conditions import Condition
+from plomtask.todos import Todo
from plomtask.exceptions import NotFoundException, HandledException
self.assertEqual(self.proc1.title.history, p_loaded.title.history)
def test_Process_removal(self) -> None:
- """Test removal of Processes."""
+ """Test removal of Processes and ProcessSteps."""
assert isinstance(self.proc3.id_, int)
self.proc1.remove(self.db_conn)
self.assertEqual({self.proc2.id_, self.proc3.id_},
set(p.id_ for p in Process.all(self.db_conn)))
self.proc2.set_steps(self.db_conn, [(None, self.proc3.id_, None)])
+ with self.assertRaises(HandledException):
+ self.proc3.remove(self.db_conn)
self.proc2.explicit_steps[0].remove(self.db_conn)
retrieved = Process.by_id(self.db_conn, self.proc2.id_)
self.assertEqual(retrieved.explicit_steps, [])
self.proc2.remove(self.db_conn)
with self.assertRaises(NotFoundException):
ProcessStep.by_id(self.db_conn, step.id_)
+ todo = Todo(None, self.proc3, False, '2024-01-01')
+ todo.save(self.db_conn)
+ with self.assertRaises(HandledException):
+ self.proc3.remove(self.db_conn)
+ todo.remove(self.db_conn)
+ self.proc3.remove(self.db_conn)
class TestsWithServer(TestCaseWithServer):