def test_harmless_actions(self): noop = Mock() noop.cmd = 'harmless' dangerous = Mock() dangerous.cmd = 'status' action_list = [noop, dangerous] self.assertEqual(remove_harmless_actions(action_list), [])
def test_dangerous_reboot_action(self): noop = Mock() noop.cmd = 'harmless' dangerous = Mock() dangerous.cmd = 'reboot' action_list = [noop, dangerous] self.assertEqual(remove_harmless_actions(action_list), [dangerous])
def test_harmless_update_without_reboot_action(self): noop = Mock() noop.cmd = 'harmless' harmless = Mock() harmless.cmd = 'update' harmless.kwargs = {yadtshell.constants.REBOOT_REQUIRED: False} action_list = [noop, harmless] self.assertEqual(remove_harmless_actions(action_list), [])
def test_dangerous_update_with_reboot_action(self): noop = Mock() noop.cmd = 'harmless' dangerous = Mock() dangerous.cmd = 'update' dangerous.kwargs = {yadtshell.constants.REBOOT_REQUIRED: True} action_list = [noop, dangerous] self.assertEqual(remove_harmless_actions(action_list), [dangerous])