Add unit tests for SessionAssistant

canonical · Mar 15, 2024 · aa01dfb · aa01dfb
1 parent cd6ec93
commit aa01dfb
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/checkbox-ng/plainbox/impl/session/test_assistant.py b/checkbox-ng/plainbox/impl/session/test_assistant.py
@@ -203,3 +203,27 @@ def test_bootstrap(self, mock_tpu, mock_su, mock_get_providers):
             self_mock._context.state.update_desired_job_list.call_count,
             2
         )
+
+    @mock.patch("plainbox.impl.session.state.select_units")
+    def test_hand_pick_jobs(self, mock_su, mock_get_providers):
+        self_mock = mock.MagicMock()
+        SessionAssistant.hand_pick_jobs(self_mock, [])
+        self.assertEqual(
+            self_mock._context.state.update_desired_job_list.call_count,
+            1
+        )
+
+    @mock.patch("plainbox.impl.session.state.select_units")
+    @mock.patch("plainbox.impl.unit.testplan.TestPlanUnit")
+    def test_get_bootstrap_todo_list(
+        self,
+        mock_tpu,
+        mock_su,
+        mock_get_providers
+    ):
+        self_mock = mock.MagicMock()
+        SessionAssistant.get_bootstrap_todo_list(self_mock)
+        self.assertEqual(
+            self_mock._context.state.update_desired_job_list.call_count,
+            1
+        )