def test_flash_finished():
"""
Ensure state is set back as expected when the flashing thread is finished.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mm.flash_thread = mock.MagicMock()
mm.flash_timer = mock.MagicMock()
mm.flash_finished()
view.button_bar.slots['flash'].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with("Finished flashing.")
assert mm.flash_thread is None
assert mm.flash_timer is None
def test_flash_finished():
"""
Ensure state is set back as expected when the flashing thread is finished.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mm.flash_thread = mock.MagicMock()
mm.flash_timer = mock.MagicMock()
mm.flash_finished()
view.button_bar.slots['flash'].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with("Finished flashing.")
assert mm.flash_thread is None
assert mm.flash_timer is None
def test_flash_failed():
"""
Ensure things are cleaned up if flashing failed.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mock_timer = mock.MagicMock()
mm.flash_timer = mock_timer
mm.flash_thread = mock.MagicMock()
mm.flash_failed('Boom')
assert view.show_message.call_count == 1
view.button_bar.slots['flash'].setEnabled.assert_called_once_with(True)
assert mm.flash_thread is None
assert mm.flash_timer is None
mock_timer.stop.assert_called_once_with()
def test_flash_failed():
"""
Ensure things are cleaned up if flashing failed.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mock_timer = mock.MagicMock()
mm.flash_timer = mock_timer
mm.flash_thread = mock.MagicMock()
mm.flash_failed('Boom')
assert view.show_message.call_count == 1
view.button_bar.slots['flash'].setEnabled.assert_called_once_with(True)
assert mm.flash_thread is None
assert mm.flash_timer is None
mock_timer.stop.assert_called_once_with()
def test_flash_failed():
"""
Ensure things are cleaned up if flashing failed.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mm.set_buttons = mock.MagicMock()
mock_timer = mock.MagicMock()
mm.flash_timer = mock_timer
mm.flash_thread = mock.MagicMock()
mm.flash_failed("Boom")
assert view.show_message.call_count == 1
mm.set_buttons.assert_called_once_with(flash=True)
assert mm.flash_thread is None
assert mm.flash_timer is None
mock_timer.stop.assert_called_once_with()
def test_flash_finished_copy_main():
"""
Ensure state is set back as expected when the flashing thread is finished.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mm.python_script = 'foo'
mm.copy_main = mock.MagicMock()
mm.set_buttons = mock.MagicMock()
mm.flash_thread = mock.MagicMock()
mm.flash_timer = mock.MagicMock()
mm.flash_finished()
mm.set_buttons.assert_called_once_with(flash=True)
editor.show_status_message.assert_called_once_with("Finished flashing.")
assert mm.flash_thread is None
assert mm.flash_timer is None
mm.copy_main.assert_called_once_with()
def test_flash_finished_copy_main_encounters_error():
"""
If copy_main encounters an error, flash_failed is called.
"""
view = mock.MagicMock()
editor = mock.MagicMock()
mm = MicrobitMode(editor, view)
mm.flash_failed = mock.MagicMock()
mm.python_script = 'foo'
error = IOError('boom')
mm.copy_main = mock.MagicMock(side_effect=error)
mm.set_buttons = mock.MagicMock()
mm.flash_thread = mock.MagicMock()
mm.flash_timer = mock.MagicMock()
mm.flash_finished()
mm.set_buttons.assert_called_once_with(flash=True)
editor.show_status_message.assert_called_once_with("Finished flashing.")
assert mm.flash_thread is None
assert mm.flash_timer is None
mm.copy_main.assert_called_once_with()
mm.flash_failed.assert_called_once_with(error)
