def test_python_run_toggle_on():
"""
Check the handler for clicking run starts the new process and updates the
UI state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.button_bar.slots = {
'debug': mock.MagicMock(),
'modes': mock.MagicMock(),
'run': mock.MagicMock(),
}
pm = PythonMode(editor, view)
pm.runner = None
def runner(pm=pm):
pm.runner = True
pm.run_script = mock.MagicMock(side_effect=runner)
pm.run_toggle(None)
pm.run_script.assert_called_once_with()
slot = pm.view.button_bar.slots['run']
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with('Stop')
slot.setToolTip.assert_called_once_with('Stop your Python script.')
pm.view.button_bar.slots['debug'].setEnabled.assert_called_once_with(False)
pm.view.button_bar.slots['modes'].setEnabled.assert_called_once_with(False)
def test_python_mode():
"""
Sanity check for setting up of the mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
assert pm.name == "Python 3"
assert pm.description is not None
assert pm.icon == "python"
assert pm.is_debugger is False
assert pm.editor == editor
assert pm.view == view
with mock.patch("mu.modes.python3.CHARTS", True):
actions = pm.actions()
assert len(actions) == 4
assert actions[0]["name"] == "run"
assert actions[0]["handler"] == pm.run_toggle
assert actions[1]["name"] == "debug"
assert actions[1]["handler"] == pm.debug
assert actions[2]["name"] == "repl"
assert actions[2]["handler"] == pm.toggle_repl
assert actions[3]["name"] == "plotter"
assert actions[3]["handler"] == pm.toggle_plotter
def test_python_add_repl():
"""
Check the REPL's kernal manager is configured correctly before being handed
to the Jupyter widget in the view.
"""
mock_qthread = mock.MagicMock()
mock_kernel_runner = mock.MagicMock()
editor = mock.MagicMock()
editor.envars = [['name', 'value'], ]
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.stop_kernel = mock.MagicMock()
with mock.patch('mu.modes.python3.QThread', mock_qthread), \
mock.patch('mu.modes.python3.KernelRunner', mock_kernel_runner):
pm.add_repl()
mock_qthread.assert_called_once_with()
mock_kernel_runner.assert_called_once_with(cwd=pm.workspace_dir(),
envars=editor.envars)
assert pm.kernel_thread == mock_qthread()
assert pm.kernel_runner == mock_kernel_runner()
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(False)
pm.kernel_runner.moveToThread.assert_called_once_with(pm.kernel_thread)
pm.kernel_runner.kernel_started.connect.\
assert_called_once_with(pm.on_kernel_start)
pm.kernel_runner.kernel_finished.connect.\
assert_called_once_with(pm.kernel_thread.quit)
pm.stop_kernel.connect.\
assert_called_once_with(pm.kernel_runner.stop_kernel)
pm.kernel_thread.started.connect.\
assert_called_once_with(pm.kernel_runner.start_kernel)
pm.kernel_thread.finished.connect.\
assert_called_once_with(pm.on_kernel_stop)
pm.kernel_thread.start.assert_called_once_with()
def test_python_add_repl():
"""
Check the REPL's kernal manager is configured correctly before being handed
to the Jupyter widget in the view.
"""
mock_qthread = mock.MagicMock()
mock_kernel_runner = mock.MagicMock()
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.stop_kernel = mock.MagicMock()
with mock.patch('mu.modes.python3.QThread', mock_qthread), \
mock.patch('mu.modes.python3.KernelRunner', mock_kernel_runner):
pm.add_repl()
mock_qthread.assert_called_once_with()
mock_kernel_runner.assert_called_once_with()
assert pm.kernel_thread == mock_qthread()
assert pm.kernel_runner == mock_kernel_runner()
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(False)
pm.kernel_runner.moveToThread.assert_called_once_with(pm.kernel_thread)
pm.kernel_runner.kernel_started.connect.\
assert_called_once_with(pm.on_kernel_start)
pm.kernel_runner.kernel_finished.connect.\
assert_called_once_with(pm.kernel_thread.quit)
pm.stop_kernel.connect.\
assert_called_once_with(pm.kernel_runner.stop_kernel)
pm.kernel_thread.started.connect.\
assert_called_once_with(pm.kernel_runner.start_kernel)
pm.kernel_thread.finished.connect.\
assert_called_once_with(pm.on_kernel_stop)
pm.kernel_thread.start.assert_called_once_with()
def test_python_api():
"""
Make sure the API definition is as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
result = pm.api()
assert result == SHARED_APIS + PYTHON3_APIS + PI_APIS
def test_python_api():
"""
Make sure the API definition is as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
result = pm.api()
assert isinstance(result, list)
def test_python_api():
"""
Make sure the API definition is as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
result = pm.api()
assert result == SHARED_APIS + PYTHON3_APIS + PI_APIS
def test_python_remove_plotter_reset_focus():
"""
Ensure the button states are returned to normal before calling super
method.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.remove_plotter()
view.current_tab.setFocus.assert_called_once_with()
def test_python_remove_plotter_reset_focus():
"""
Ensure the button states are returned to normal before calling super
method.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.remove_plotter()
view.current_tab.setFocus.assert_called_once_with()
def test_python_stop_script_no_runner():
"""
If the script is cancelled before the child process is created ensure
nothing breaks and the UI is reset.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = None
pm.stop_script()
view.remove_python_runner.assert_called_once_with()
def test_python_remove_repl():
"""
Make sure the REPL is removed properly.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.stop_kernel = mock.MagicMock()
pm.remove_repl()
pm.stop_kernel.emit.assert_called_once_with()
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(False)
def test_python_debug():
"""
Ensure Python3 mode hands over running of the script to the debug mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.debug(None)
editor.change_mode.assert_called_once_with("debugger")
assert editor.mode == "debugger"
editor.modes["debugger"].start.assert_called_once_with()
def test_python_remove_repl():
"""
Make sure the REPL is removed properly.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.stop_kernel = mock.MagicMock()
pm.remove_repl()
pm.stop_kernel.emit.assert_called_once_with()
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(False)
def test_python_debug():
"""
Ensure Python3 mode hands over running of the script to the debug mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.debug(None)
editor.change_mode.assert_called_once_with('debugger')
assert editor.mode == 'debugger'
editor.modes['debugger'].start.assert_called_once_with()
def test_python_remove_repl():
"""
Make sure the REPL is removed properly.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.repl = True
pm.remove_repl()
assert pm.repl is None
view.remove_repl.assert_called_once_with()
def test_python_stop_script_no_runner():
"""
If the script is cancelled before the child process is created ensure
nothing breaks and the UI is reset.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = None
pm.stop_script()
view.remove_python_runner.assert_called_once_with()
def test_python_run_script_no_editor():
"""
If there's no active tab, there can be no runner either.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab = None
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
assert pm.runner is None
pm.stop_script.assert_called_once_with()
def test_python_stop_resets_focus():
"""
Check that, when a child process is killed, the current
tab regains focus.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_runner = mock.MagicMock()
pm.runner = mock_runner
pm.stop_script()
view.current_tab.setFocus.assert_called_once_with()
def test_python_remove_repl():
"""
Make sure the REPL is removed properly.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
pm.stop_kernel = mock.MagicMock()
pm.remove_repl()
pm.stop_kernel.emit.assert_called_once_with()
pm.set_buttons.assert_called_once_with(repl=False)
def test_python_run_script_needs_saving():
"""
If the file hasn't been saved yet (it's unnamed), prompt the user to save
it.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab.path = None
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
editor.save.assert_called_once_with()
def test_python_stop_resets_focus():
"""
Check that, when a child process is killed, the current
tab regains focus.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_runner = mock.MagicMock()
pm.runner = mock_runner
pm.stop_script()
view.current_tab.setFocus.assert_called_once_with()
def test_python_run_script_no_editor():
"""
If there's no active tab, there can be no runner either.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab = None
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
assert pm.runner is None
pm.stop_script.assert_called_once_with()
def test_python_run_script_needs_saving():
"""
If the file hasn't been saved yet (it's unnamed), prompt the user to save
it.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab.path = None
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
editor.save.assert_called_once_with()
def test_python_run_script_uses_editor_save():
"""The run code uses the common editor save code, invoking
encoding checks and useful messages
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab.IsModified.return_value = True
view.current_tab.path = "foo"
view.current_tab.text = mock.MagicMock(return_value="foo")
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
editor.save_tab_to_file.assert_called_once_with(view.current_tab)
def test_python_on_kernel_stop():
"""
Ensure everything REPL based is cleaned up when this handler is called.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.button_bar.slots = {"repl": mock.MagicMock()}
pm = PythonMode(editor, view)
pm.on_kernel_stop()
assert pm.repl_kernel_manager is None
view.button_bar.slots["repl"].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with("REPL stopped.")
assert pm.kernel_runner is None
def test_python_stop_script():
"""
Check that the child process is killed, the runner cleaned up and UI
is reset.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_runner = mock.MagicMock()
pm.runner = mock_runner
pm.stop_script()
mock_runner.stop_process.assert_called_once_with()
assert pm.runner is None
def test_python_run_script_uses_editor_save():
"""The run code uses the common editor save code, invoking
encoding checks and useful messages
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab.IsModified.return_value = True
view.current_tab.path = "foo"
view.current_tab.text = mock.MagicMock(return_value="foo")
pm = PythonMode(editor, view)
pm.stop_script = mock.MagicMock()
pm.run_script()
editor.save_tab_to_file.assert_called_once_with(view.current_tab)
def test_python_run_toggle_on_cancelled():
"""
Ensure the button states are correct if running an unsaved script is
cancelled before the process is allowed to start. See issue #338.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = None
pm.run_script = mock.MagicMock()
pm.run_toggle(None)
pm.run_script.assert_called_once_with()
slot = pm.view.button_bar.slots['run']
assert slot.setIcon.call_count == 0
def test_python_remove_plotter():
"""
Ensure the button states are returned to normal before calling super
method.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
with mock.patch('builtins.super') as mock_super:
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
mock_super.reset_mock()
pm.remove_plotter()
pm.set_buttons.assert_called_once_with(run=True, repl=True, debug=True)
mock_super().remove_plotter.assert_called_once_with()
def test_python_remove_plotter():
"""
Ensure the button states are returned to normal before calling super
method.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
with mock.patch("builtins.super") as mock_super:
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
mock_super.reset_mock()
pm.remove_plotter()
pm.set_buttons.assert_called_once_with(run=True, repl=True, debug=True)
mock_super().remove_plotter.assert_called_once_with()
def test_python_stop_script():
"""
Check that the child process is killed, the runner cleaned up and UI
is reset.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_runner = mock.MagicMock()
pm.runner = mock_runner
pm.stop_script()
mock_runner.process.kill.assert_called_once_with()
mock_runner.process.waitForFinished.assert_called_once_with()
assert pm.runner is None
def test_python_on_kernel_stop():
"""
Ensure everything REPL based is cleaned up when this handler is called.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.button_bar.slots = {
'repl': mock.MagicMock(),
}
pm = PythonMode(editor, view)
pm.on_kernel_stop()
assert pm.repl_kernel_manager is None
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with('REPL stopped.')
assert pm.kernel_runner is None
def test_python_add_repl():
"""
Check the REPL's kernal manager is configured correctly before being handed
to the Jupyter widget in the view.
"""
mock_kernel_manager = mock.MagicMock()
mock_manager_class = mock.MagicMock(return_value=mock_kernel_manager)
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
with mock.patch('mu.modes.python3.QtInProcessKernelManager',
mock_manager_class):
pm.add_repl()
mock_kernel_manager.start_kernel.assert_called_once_with(show_banner=False)
view.add_jupyter_repl.assert_called_once_with(mock_kernel_manager)
assert pm.repl == mock_kernel_manager
def test_python_on_kernel_start():
"""
Ensure the handler for when the kernel has started updates the UI such that
the kernel manager and kernel client are used to add the Jupyter widget to
the UI, the REPL button is re-enabled and a status update is shown.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_kernel_manager = mock.MagicMock()
mock_client = mock.MagicMock()
pm.on_kernel_start(mock_kernel_manager, mock_client)
view.add_jupyter_repl.assert_called_once_with(mock_kernel_manager,
mock_client)
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with('REPL started.')
def test_python_on_kernel_start():
"""
Ensure the handler for when the kernel has started updates the UI such that
the kernel manager and kernel client are used to add the Jupyter widget to
the UI, the REPL button is re-enabled and a status update is shown.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
mock_kernel_manager = mock.MagicMock()
mock_client = mock.MagicMock()
pm.on_kernel_start(mock_kernel_manager, mock_client)
view.add_jupyter_repl.assert_called_once_with(mock_kernel_manager,
mock_client)
view.button_bar.slots['repl'].setEnabled.assert_called_once_with(True)
editor.show_status_message.assert_called_once_with('REPL started.')
def test_python_run_toggle_off():
"""
Check the handler for clicking run stops the process and reverts the UI
state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = True
pm.stop_script = mock.MagicMock()
pm.run_toggle(None)
pm.stop_script.assert_called_once_with()
slot = pm.view.button_bar.slots['run']
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with('Run')
slot.setToolTip.assert_called_once_with('Run your Python script.')
pm.view.button_bar.slots['debug'].setEnabled.assert_called_once_with(True)
def test_python_toggle_plotter():
"""
Ensure toggling the plotter causes it to be added/removed.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.add_plotter = mock.MagicMock()
pm.remove_plotter = mock.MagicMock()
pm.toggle_plotter()
pm.add_plotter.assert_called_once_with()
pm.plotter = True
pm.toggle_plotter()
pm.remove_plotter.assert_called_once_with()
def test_python_toggle_repl():
"""
Ensure the REPL handling works as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.add_repl = mock.MagicMock()
pm.remove_repl = mock.MagicMock()
pm.toggle_repl(None)
pm.add_repl.assert_called_once_with()
pm.kernel_runner = True
pm.toggle_repl(None)
pm.remove_repl.assert_called_once_with()
def test_python_add_repl():
"""
Check the REPL's kernal manager is configured correctly before being handed
to the Jupyter widget in the view.
"""
mock_qthread = mock.MagicMock()
mock_kernel_runner = mock.MagicMock()
editor = mock.MagicMock()
editor.envars = [["name", "value"]]
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
pm.stop_kernel = mock.MagicMock()
with mock.patch("mu.modes.python3.QThread",
mock_qthread), mock.patch("mu.modes.python3.KernelRunner",
mock_kernel_runner):
pm.add_repl()
mock_qthread.assert_called_once_with()
mock_kernel_runner.assert_called_once_with(cwd=pm.workspace_dir(),
envars=editor.envars)
assert pm.kernel_thread == mock_qthread()
assert pm.kernel_runner == mock_kernel_runner()
pm.set_buttons.assert_called_once_with(repl=False)
pm.kernel_runner.moveToThread.assert_called_once_with(pm.kernel_thread)
pm.kernel_runner.kernel_started.connect.assert_called_once_with(
pm.on_kernel_start)
pm.kernel_runner.kernel_finished.connect.assert_called_once_with(
pm.kernel_thread.quit)
pm.stop_kernel.connect.assert_called_once_with(
pm.kernel_runner.stop_kernel)
pm.kernel_thread.started.connect.assert_called_once_with(
pm.kernel_runner.start_kernel)
pm.kernel_thread.finished.connect.assert_called_once_with(
pm.on_kernel_stop)
pm.kernel_thread.start.assert_called_once_with()
def test_python_run_script():
"""
Ensure that running the script launches the process as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.current_tab.path = '/foo'
view.current_tab.isModified.return_value = True
mock_runner = mock.MagicMock()
view.add_python3_runner.return_value = mock_runner
pm = PythonMode(editor, view)
pm.workspace_dir = mock.MagicMock(return_value='/bar')
with mock.patch('builtins.open') as oa, \
mock.patch('mu.modes.python3.write_and_flush'):
pm.run_script()
oa.assert_called_once_with('/foo', 'w', newline='')
view.add_python3_runner.assert_called_once_with('/foo', '/bar',
interactive=True)
mock_runner.process.waitForStarted.assert_called_once_with()
def test_python_mode():
"""
Sanity check for setting up of the mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
assert pm.name == 'Python 3'
assert pm.description is not None
assert pm.icon == 'python'
assert pm.is_debugger is False
assert pm.editor == editor
assert pm.view == view
actions = pm.actions()
assert len(actions) == 2
assert actions[0]['name'] == 'run'
assert actions[0]['handler'] == pm.run
assert actions[1]['name'] == 'repl'
assert actions[1]['handler'] == pm.toggle_repl
def test_python_toggle_plotter():
"""
Ensure toggling the plotter causes it to be added/removed.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.add_plotter = mock.MagicMock()
pm.remove_plotter = mock.MagicMock()
pm.toggle_plotter()
pm.add_plotter.assert_called_once_with()
pm.plotter = True
pm.toggle_plotter()
pm.remove_plotter.assert_called_once_with()
def test_python_toggle_repl():
"""
Ensure the REPL handling works as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.add_repl = mock.MagicMock()
pm.remove_repl = mock.MagicMock()
pm.toggle_repl(None)
pm.add_repl.assert_called_once_with()
pm.kernel_runner = True
pm.toggle_repl(None)
pm.remove_repl.assert_called_once_with()
def test_python_run_toggle_on():
"""
Check the handler for clicking run starts the new process and updates the
UI state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.button_bar.slots = {
"debug": mock.MagicMock(),
"modes": mock.MagicMock(),
"run": mock.MagicMock(),
}
pm = PythonMode(editor, view)
pm.runner = None
def runner(pm=pm):
pm.runner = True
pm.run_script = mock.MagicMock(side_effect=runner)
pm.run_toggle(None)
pm.run_script.assert_called_once_with()
slot = pm.view.button_bar.slots["run"]
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with("Stop")
slot.setToolTip.assert_called_once_with("Stop your Python script.")
pm.view.button_bar.slots["debug"].setEnabled.assert_called_once_with(False)
pm.view.button_bar.slots["modes"].setEnabled.assert_called_once_with(False)
def test_python_mode():
"""
Sanity check for setting up of the mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
assert pm.name == 'Python 3'
assert pm.description is not None
assert pm.icon == 'python'
assert pm.is_debugger is False
assert pm.editor == editor
assert pm.view == view
actions = pm.actions()
assert len(actions) == 3
assert actions[0]['name'] == 'run'
assert actions[0]['handler'] == pm.run_toggle
assert actions[1]['name'] == 'debug'
assert actions[1]['handler'] == pm.debug
assert actions[2]['name'] == 'repl'
assert actions[2]['handler'] == pm.toggle_repl
def test_python_mode_no_charts():
"""
If QCharts is not available, ensure the plotter feature is not available.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
assert pm.name == "Python 3"
assert pm.description is not None
assert pm.icon == "python"
assert pm.is_debugger is False
assert pm.editor == editor
assert pm.view == view
with mock.patch("mu.modes.python3.CHARTS", False):
actions = pm.actions()
assert len(actions) == 3
assert actions[0]["name"] == "run"
assert actions[0]["handler"] == pm.run_toggle
assert actions[1]["name"] == "debug"
assert actions[1]["handler"] == pm.debug
assert actions[2]["name"] == "repl"
assert actions[2]["handler"] == pm.toggle_repl
def test_python_remove_repl():
"""
Make sure the REPL is removed properly.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
pm.stop_kernel = mock.MagicMock()
pm.remove_repl()
pm.stop_kernel.emit.assert_called_once_with()
pm.set_buttons.assert_called_once_with(repl=False)
def test_python_run_toggle_on_cancelled():
"""
Ensure the button states are correct if running an unsaved script is
cancelled before the process is allowed to start. See issue #338.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = None
pm.run_script = mock.MagicMock()
pm.run_toggle(None)
pm.run_script.assert_called_once_with()
slot = pm.view.button_bar.slots["run"]
assert slot.setIcon.call_count == 0
def test_python_run_toggle_on():
"""
Check the handler for clicking run starts the new process and updates the
UI state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.runner = None
pm.run_script = mock.MagicMock()
pm.run_toggle(None)
pm.run_script.assert_called_once_with()
slot = pm.view.button_bar.slots['run']
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with('Stop')
slot.setToolTip.assert_called_once_with('Stop your Python script.')
pm.view.button_bar.slots['debug'].setEnabled.assert_called_once_with(False)
def test_python_run_toggle_off():
"""
Check the handler for clicking run stops the process and reverts the UI
state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
view.button_bar.slots = {
"debug": mock.MagicMock(),
"modes": mock.MagicMock(),
"run": mock.MagicMock(),
}
pm = PythonMode(editor, view)
pm.runner = True
pm.stop_script = mock.MagicMock()
pm.run_toggle(None)
pm.stop_script.assert_called_once_with()
slot = pm.view.button_bar.slots["run"]
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with("Run")
slot.setToolTip.assert_called_once_with("Run your Python script.")
pm.view.button_bar.slots["debug"].setEnabled.assert_called_once_with(True)
pm.view.button_bar.slots["modes"].setEnabled.assert_called_once_with(True)
def test_python_add_plotter():
"""
Ensure the plotter is added as expected.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
pm.add_plotter()
view.add_python3_plotter.assert_called_once_with(pm)
assert pm.plotter
pm.set_buttons.assert_called_once_with(debug=False)
# Check button states are updated depending on other aspects of the mode
# being enabled.
# REPL active.
pm.set_buttons.reset_mock()
pm.repl = True
pm.add_plotter()
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]['debug'] is False
assert pm.set_buttons.call_args_list[1][1]['run'] is False
# Runner active.
pm.set_buttons.reset_mock()
pm.repl = False
pm.runner = True
pm.add_plotter()
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]['debug'] is False
assert pm.set_buttons.call_args_list[1][1]['repl'] is False
def test_python_on_kernel_start():
"""
Ensure the handler for when the kernel has started updates the UI such that
the kernel manager and kernel client are used to add the Jupyter widget to
the UI, the REPL button is re-enabled and a status update is shown.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
mock_kernel_manager = mock.MagicMock()
mock_client = mock.MagicMock()
pm.on_kernel_start(mock_kernel_manager, mock_client)
view.add_jupyter_repl.assert_called_once_with(mock_kernel_manager,
mock_client)
pm.set_buttons.assert_called_once_with(repl=True)
editor.show_status_message.assert_called_once_with("REPL started.")
# Check button states are set according to what other aspects of the mode
# are currently enabled.
# With Runner enabled.
pm.set_buttons.reset_mock()
pm.runner = True
pm.on_kernel_start(mock_kernel_manager, mock_client)
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]["repl"] is True
assert pm.set_buttons.call_args_list[1][1]["plotter"] is False
# With Plotter enabled.
pm.set_buttons.reset_mock()
pm.runner = False
pm.plotter = True
pm.on_kernel_start(mock_kernel_manager, mock_client)
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]["repl"] is True
assert pm.set_buttons.call_args_list[1][1]["run"] is False
assert pm.set_buttons.call_args_list[1][1]["debug"] is False
def test_python_on_kernel_start():
"""
Ensure the handler for when the kernel has started updates the UI such that
the kernel manager and kernel client are used to add the Jupyter widget to
the UI, the REPL button is re-enabled and a status update is shown.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
mock_kernel_manager = mock.MagicMock()
mock_client = mock.MagicMock()
pm.on_kernel_start(mock_kernel_manager, mock_client)
view.add_jupyter_repl.assert_called_once_with(mock_kernel_manager,
mock_client)
pm.set_buttons.assert_called_once_with(repl=True)
editor.show_status_message.assert_called_once_with('REPL started.')
# Check button states are set according to what other aspects of the mode
# are currently enabled.
# With Runner enabled.
pm.set_buttons.reset_mock()
pm.runner = True
pm.on_kernel_start(mock_kernel_manager, mock_client)
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]['repl'] is True
assert pm.set_buttons.call_args_list[1][1]['plotter'] is False
# With Plotter enabled.
pm.set_buttons.reset_mock()
pm.runner = False
pm.plotter = True
pm.on_kernel_start(mock_kernel_manager, mock_client)
assert pm.set_buttons.call_count == 2
assert pm.set_buttons.call_args_list[0][1]['repl'] is True
assert pm.set_buttons.call_args_list[1][1]['run'] is False
assert pm.set_buttons.call_args_list[1][1]['debug'] is False
def test_python_run_script():
"""
Ensure that running the script launches the process as expected.
"""
editor = mock.MagicMock()
editor.envars = [['name', 'value']]
view = mock.MagicMock()
view.current_tab.path = '/foo/bar'
view.current_tab.isModified.return_value = True
view.current_tab.text = mock.MagicMock(return_value="abc")
mock_runner = mock.MagicMock()
view.add_python3_runner.return_value = mock_runner
pm = PythonMode(editor, view)
pm.run_script()
editor.save_tab_to_file.assert_called_once_with(view.current_tab)
view.add_python3_runner.assert_called_once_with('/foo/bar', '/foo',
interactive=True,
envars=editor.envars)
mock_runner.process.waitForStarted.assert_called_once_with()
# Check the buttons are set to the correct state when other aspects of the
# mode are also in play.
pm.set_buttons = mock.MagicMock()
pm.kernel_runner = True
pm.run_script()
pm.set_buttons.assert_called_once_with(plotter=False)
pm.set_buttons.reset_mock()
pm.kernel_runner = False
pm.plotter = True
pm.run_script()
pm.set_buttons.assert_called_once_with(repl=False)
def test_python_run_script():
"""
Ensure that running the script launches the process as expected.
"""
editor = mock.MagicMock()
editor.envars = [['name', 'value']]
view = mock.MagicMock()
view.current_tab.path = '/foo'
view.current_tab.isModified.return_value = True
mock_runner = mock.MagicMock()
view.add_python3_runner.return_value = mock_runner
pm = PythonMode(editor, view)
pm.workspace_dir = mock.MagicMock(return_value='/bar')
with mock.patch('builtins.open') as oa, \
mock.patch('mu.modes.python3.write_and_flush'):
pm.run_script()
oa.assert_called_once_with('/foo', 'w', newline='')
view.add_python3_runner.assert_called_once_with('/foo', '/bar',
interactive=True,
envars=editor.envars)
mock_runner.process.waitForStarted.assert_called_once_with()
# Check the buttons are set to the correct state when other aspects of the
# mode are also in play.
pm.set_buttons = mock.MagicMock()
pm.kernel_runner = True
with mock.patch('builtins.open') as oa, \
mock.patch('mu.modes.python3.write_and_flush'):
pm.run_script()
pm.set_buttons.assert_called_once_with(plotter=False)
pm.set_buttons.reset_mock()
pm.kernel_runner = False
pm.plotter = True
with mock.patch('builtins.open') as oa, \
mock.patch('mu.modes.python3.write_and_flush'):
pm.run_script()
pm.set_buttons.assert_called_once_with(repl=False)
def test_python_on_data_flood():
"""
Ensure that Python 3 mode cleans up correctly after a data flood of the
plotter is detected: reset the buttons, stop either the REPL or runner and
then call the base method.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PythonMode(editor, view)
pm.set_buttons = mock.MagicMock()
pm.run_toggle = mock.MagicMock()
pm.remove_repl = mock.MagicMock()
pm.runner = True
with mock.patch('builtins.super') as mock_super:
pm.on_data_flood()
pm.set_buttons.assert_called_once_with(run=True, repl=True, debug=True)
pm.run_toggle.assert_called_once_with(None)
mock_super().on_data_flood.assert_called_once_with()
pm.set_buttons.reset_mock()
pm.runner = False
pm.kernel_runner = True
with mock.patch('builtins.super') as mock_super:
pm.on_data_flood()
pm.set_buttons.assert_called_once_with(run=True, repl=True, debug=True)
pm.remove_repl.assert_called_once_with()
mock_super().on_data_flood.assert_called_once_with()
