Feature/BCON lite - Beam Control without deflection

citestfiles/CathodeHeating/beams_off_test.py

@@ -0,0 +1,143 @@
+import sys, os, unittest
+from unittest.mock import MagicMock
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
+
+from subsystem.cathode_heating.cathode_heating import CathodeHeatingSubsystem
+from utils import LogLevel
+
+class TestBeamsOff(unittest.TestCase):
+    def setUp(self):
+        # Bypass __init__ to avoid Tk and image loading
+        self.subsys = object.__new__(CathodeHeatingSubsystem)
+        # Inject only what turn_off_all_beams uses
+        self.subsys.power_supplies_initialized = True
+        self.subsys.power_supplies = [MagicMock(), None, MagicMock()]
+        self.subsys.power_supply_status = [True, False, True]
+        self.subsys.toggle_states = [True, True, True]
+        self.subsys.toggle_off_image = object()
+        self.subsys.toggle_buttons = [MagicMock(), MagicMock(), MagicMock()]
+        # Simple logger hook
+        self.subsys.logger = MagicMock()
+        self.subsys.log = lambda msg, lvl=LogLevel.INFO: None
+
+        # Alias method under test (name in your file)
+        self.turn_off_all_beams = self.subsys.turn_off_all_beams
+
+    def test_turns_off_only_initialized_ps_and_updates_ui_on_success(self):
+        # First and third supply return True; middle is uninitialized
+        self.subsys.power_supplies[0].set_output.return_value = True
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        self.turn_off_all_beams()
+
+        self.subsys.power_supplies[0].set_output.assert_called_once_with("0")
+        self.subsys.power_supplies[2].set_output.assert_called_once_with("0")
+        self.assertFalse(self.subsys.toggle_states[0])
+        self.assertFalse(self.subsys.toggle_states[2])
+        self.subsys.toggle_buttons[0].config.assert_called_once()
+        self.subsys.toggle_buttons[2].config.assert_called_once()
+        # Uninitialized index 1 untouched
+        self.subsys.toggle_buttons[1].config.assert_not_called()
+
+    def test_does_not_update_ui_when_off_fails(self):
+        # Simulate failure on index 0, success on index 2
+        self.subsys.power_supplies[0].set_output.return_value = False
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        self.turn_off_all_beams()
+
+        # UI should not change for failed OFF
+        self.assertTrue(self.subsys.toggle_states[0])
+        self.subsys.toggle_buttons[0].config.assert_not_called()
+
+        # UI should change for successful OFF
+        self.assertFalse(self.subsys.toggle_states[2])
+        self.subsys.toggle_buttons[2].config.assert_called_once()
+
+    def test_exceptions_are_caught_and_others_continue(self):
+        self.subsys.power_supplies[0].set_output.side_effect = RuntimeError("boom")
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        # Should not raise
+        self.turn_off_all_beams()
+
+        self.subsys.power_supplies[2].set_output.assert_called_once_with("0")
+        self.assertFalse(self.subsys.toggle_states[2])
+
+    def test_returns_early_when_not_initialized(self):
+        # Arrange: pretend subsystem not initialized
+        self.subsys.power_supplies_initialized = False
+        # Keep mocks to detect accidental calls
+        self.subsys.power_supplies = [MagicMock(), MagicMock(), MagicMock()]
+        self.subsys.power_supply_status = [True, True, True]
+
+        # Act
+        self.turn_off_all_beams()
+
+        # Assert: no calls made
+        for ps in self.subsys.power_supplies:
+            ps.set_output.assert_not_called()
+        for btn in self.subsys.toggle_buttons:
+            btn.config.assert_not_called()
+
+    def test_skips_when_status_false_even_if_ps_present(self):
+        # Arrange: ps exists but status is False
+        self.subsys.power_supplies = [MagicMock(), MagicMock(), MagicMock()]
+        self.subsys.power_supply_status = [True, False, True]
+        self.subsys.power_supplies[0].set_output.return_value = True
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        # Act
+        self.turn_off_all_beams()
+
+        # Assert
+        self.subsys.power_supplies[1].set_output.assert_not_called()
+
+    def test_updates_button_with_correct_image_on_success(self):
+        # Arrange
+        self.subsys.power_supplies[0].set_output.return_value = True
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        # Act
+        self.turn_off_all_beams()
+
+        # Assert exact image argument used
+        self.subsys.toggle_buttons[0].config.assert_called_once_with(image=self.subsys.toggle_off_image)
+        self.subsys.toggle_buttons[2].config.assert_called_once_with(image=self.subsys.toggle_off_image)
+
+    def test_true_status_but_none_power_supply_is_safely_skipped(self):
+        # Arrange: ps None but status True for index 1
+        self.subsys.power_supplies = [MagicMock(), None, MagicMock()]
+        self.subsys.power_supply_status = [True, True, True]
+        self.subsys.power_supplies[0].set_output.return_value = True
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        # Act (should not raise)
+        self.turn_off_all_beams()
+
+        # Assert: others still called, middle skipped
+        self.subsys.power_supplies[0].set_output.assert_called_once_with("0")
+        self.subsys.power_supplies[2].set_output.assert_called_once_with("0")
+
+        # Button 1 should not be touched since ps is None
+        self.subsys.toggle_buttons[1].config.assert_not_called()
+
+    def test_second_call_is_idempotent_and_keeps_off_state(self):
+        # Arrange first call success
+        self.subsys.power_supplies[0].set_output.return_value = True
+        self.subsys.power_supplies[2].set_output.return_value = True
+
+        # Act: call twice
+        self.turn_off_all_beams()
+        self.turn_off_all_beams()
+
+        # Assert: set_output called twice for active channels
+        self.assertEqual(self.subsys.power_supplies[0].set_output.call_count, 2)
+        self.assertEqual(self.subsys.power_supplies[2].set_output.call_count, 2)
+        # State remains off
+        self.assertFalse(self.subsys.toggle_states[0])
+        self.assertFalse(self.subsys.toggle_states[2])
+
+if __name__ == '__main__':
+    unittest.main()