diff options
Diffstat (limited to 'app/tests/test_taskview.py')
| -rw-r--r-- | app/tests/test_taskview.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/app/tests/test_taskview.py b/app/tests/test_taskview.py index 78cc2f1..2b3e542 100644 --- a/app/tests/test_taskview.py +++ b/app/tests/test_taskview.py @@ -277,7 +277,9 @@ class StateTransitionTests(_FakeTui, unittest.TestCase): view.handle_event({"kind": "step_cancelled", "index": 0}) view.handle_event({"kind": "finish", "phase": "cancelled", "rc": 1}) self.assertEqual(view.phase, "cancelled") - self.assertEqual(view._result_rc(), 1) + # 130: the CLI's user-cancel code, so callers can flash + # "cancelled" instead of "failed" (see hub._download_models_action). + self.assertEqual(view._result_rc(), 130) # The step interrupted by cancel is marked cancelled, not failed. self.assertEqual(view._step_mark(0), ("[x]", "warn")) self.assertEqual(view._step_mark(1), ("[ ]", "dim")) @@ -677,7 +679,7 @@ class LanesViewTests(_FakeTui, unittest.TestCase): view._drain() self.assertEqual(view.phase, "cancelled") self.assertTrue(view.cancelled) - self.assertEqual(view._result_rc(), 1) + self.assertEqual(view._result_rc(), 130) def test_split_render_draws_both_lane_titles(self): view, screen = self.make_view(self._two_lanes()) |