Add most basic Todo infrastructure.

[plomtask] / tests / processes.py

diff --git a/tests/processes.py b/tests/processes.py
index ac519c8fc1b1b75700a8442cfe55132b43dadda6..87b0b09809398ab1a5aa59af78bf1ac11d8bf1e4 100644 (file)
--- a/tests/processes.py
+++ b/tests/processes.py
@@ -64,42 +64,45 @@ class TestsWithDB(TestCaseWithDB):
         p_1.add_step(self.db_conn, None, p_2.id_, None)
         p_1_dict: dict[int, dict[str, Any]] = {1: {
             'process': p_2, 'parent_id': None,
-            'is_explicit': True, 'steps': {}
+            'is_explicit': True, 'steps': {}, 'seen': False
         }}
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
         s_b = p_1.add_step(self.db_conn, None, p_3.id_, None)
         p_1_dict[2] = {
             'process': p_3, 'parent_id': None,
-            'is_explicit': True, 'steps': {}
+            'is_explicit': True, 'steps': {}, 'seen': False
         }
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
         s_c = p_2.add_step(self.db_conn, None, p_3.id_, None)
         assert s_c.id_ is not None
         p_1_dict[1]['steps'] = {3: {
             'process': p_3, 'parent_id': None,
-            'is_explicit': False, 'steps': {}
+            'is_explicit': False, 'steps': {}, 'seen': False
         }}
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
         p_1.add_step(self.db_conn, None, p_2.id_, s_b.id_)
         p_1_dict[2]['steps'][4] = {
-            'process': p_2, 'parent_id': s_b.id_,
+            'process': p_2, 'parent_id': s_b.id_, 'seen': False,
             'is_explicit': True, 'steps': {3: {
                 'process': p_3, 'parent_id': None,
-                'is_explicit': False, 'steps': {}
+                'is_explicit': False, 'steps': {}, 'seen': True
                 }}}
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
         p_1.add_step(self.db_conn, None, p_3.id_, 999)
         p_1_dict[5] = {
             'process': p_3, 'parent_id': None,
-            'is_explicit': True, 'steps': {}
+            'is_explicit': True, 'steps': {}, 'seen': False
         }
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
         p_1.add_step(self.db_conn, None, p_3.id_, 3)
         p_1_dict[6] = {
             'process': p_3, 'parent_id': None,
-            'is_explicit': True, 'steps': {}
+            'is_explicit': True, 'steps': {}, 'seen': False
         }
         self.assertEqual(p_1.get_steps(self.db_conn, None), p_1_dict)
+        self.assertEqual(p_1.used_as_step_by(self.db_conn), [])
+        self.assertEqual(p_2.used_as_step_by(self.db_conn), [p_1])
+        self.assertEqual(p_3.used_as_step_by(self.db_conn), [p_1, p_2])
 
     def test_Process_by_id(self) -> None:
         """Test Process.by_id()."""