def test_pgzero_run_game_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 = PyGameZeroMode(editor, view)
pm.stop_game = mock.MagicMock()
pm.run_game()
editor.save.assert_called_once_with()
def test_pgzero_run_game_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 = PyGameZeroMode(editor, view)
pm.stop_game = mock.MagicMock()
pm.run_game()
assert pm.runner is None
pm.stop_game.assert_called_once_with()
def test_pgzero_play_toggle_on_cancelled():
"""
Ensure the button states are correct if playing an unsaved script is
cancelled before the process is allowed to start.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PyGameZeroMode(editor, view)
pm.runner = None
pm.run_game = mock.MagicMock()
pm.play_toggle(None)
pm.run_game.assert_called_once_with()
slot = pm.view.button_bar.slots["play"]
assert slot.setIcon.call_count == 0
def test_pgzero_stop_game():
"""
Check that the child process is killed, the runner cleaned up and UI
is reset.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PyGameZeroMode(editor, view)
mock_runner = mock.MagicMock()
pm.runner = mock_runner
pm.stop_game()
mock_runner.process.kill.assert_called_once_with()
mock_runner.process.waitForFinished.assert_called_once_with()
assert pm.runner is None
view.remove_python_runner.assert_called_once_with()
def test_pgzero_play_toggle_off():
"""
Check the handler for clicking play stops the process and reverts the UI
state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PyGameZeroMode(editor, view)
pm.runner = True
pm.stop_game = mock.MagicMock()
pm.play_toggle(None)
pm.stop_game.assert_called_once_with()
slot = pm.view.button_bar.slots['play']
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with('Play')
slot.setToolTip.assert_called_once_with('Play your Pygame Zero game.')
def test_pgzero_run_game():
"""
Ensure that running the game 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
mock_runner = mock.MagicMock()
view.add_python3_runner.return_value = mock_runner
pm = PyGameZeroMode(editor, view)
pm.run_game()
editor.save_tab_to_file.called_once_with(view.current_tab)
py_args = ['-m', 'pgzero']
view.add_python3_runner.assert_called_once_with('/foo/bar', '/foo',
interactive=False,
envars=editor.envars,
python_args=py_args)
mock_runner.process.waitForStarted.assert_called_once_with()
def test_pgzero_play_toggle_on():
"""
Check the handler for clicking play starts the new process and updates the
UI state.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PyGameZeroMode(editor, view)
pm.runner = None
def runner(pm=pm):
pm.runner = True
pm.run_game = mock.MagicMock(side_effect=runner)
pm.play_toggle(None)
pm.run_game.assert_called_once_with()
slot = pm.view.button_bar.slots['play']
assert slot.setIcon.call_count == 1
slot.setText.assert_called_once_with('Stop')
slot.setToolTip.assert_called_once_with('Stop your Pygame Zero game.')
def test_pgzero_run_game():
"""
Ensure that running the game 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 = PyGameZeroMode(editor, view)
pm.workspace_dir = mock.MagicMock(return_value='/bar')
with mock.patch('builtins.open') as oa, \
mock.patch('mu.modes.pygamezero.write_and_flush'):
pm.run_game()
oa.assert_called_once_with('/foo', 'w', newline='')
view.add_python3_runner.assert_called_once_with('/foo',
'/bar',
interactive=False,
runner='pgzrun')
mock_runner.process.waitForStarted.assert_called_once_with()
def test_pgzero_mode():
"""
Sanity check for setting up of the mode.
"""
editor = mock.MagicMock()
view = mock.MagicMock()
pm = PyGameZeroMode(editor, view)
assert pm.name == 'Pygame Zero'
assert pm.description is not None
assert pm.icon == 'pygamezero'
assert pm.is_debugger is False
assert pm.editor == editor
assert pm.view == view
assert pm.builtins
actions = pm.actions()
assert len(actions) == 3
assert actions[0]['name'] == 'play'
assert actions[0]['handler'] == pm.play_toggle
assert actions[1]['name'] == 'images'
assert actions[1]['handler'] == pm.show_images
assert actions[2]['name'] == 'sounds'
assert actions[2]['handler'] == pm.show_sounds