def test_runner_stop_dont_raise_runtime_error(cancel_all_tasks_mock):
cancel_all_tasks_mock.side_effect = RuntimeError("faiô!")
runner = LoaferRunner()
with does_not_raise():
runner.loop.stop()
runner.stop()
def test_runner_start_and_stop():
runner = LoaferRunner(loop=mock.Mock())
runner.stop = mock.Mock()
runner.start(run_forever=False)
assert runner.loop.run_forever.called
assert runner.stop.called
assert runner.loop.close.called
def test_runner_start_and_stop():
runner = LoaferRunner(loop=mock.Mock())
runner.stop = mock.Mock()
runner.start(run_forever=False)
assert runner.loop.run_forever.called
assert runner.stop.called
assert runner.loop.close.called
def test_runner_prepare_stop(loop_mock):
loop_mock.return_value.is_running.return_value = True
runner = LoaferRunner()
runner.prepare_stop()
loop_mock.return_value.stop.assert_called_once_with()
def test_runner_stop_with_callback(loop_mock):
callback = mock.Mock()
runner = LoaferRunner(on_stop_callback=callback)
runner.stop()
assert callback.called
def test_runner_start_and_stop(loop_mock):
runner = LoaferRunner()
runner.stop = mock.Mock()
runner.start()
assert runner.stop.called
assert loop_mock.return_value.run_forever.called
assert loop_mock.return_value.close.called
def test_runner_prepare_stop_already_stopped(get_loop_mock):
loop = mock.Mock(is_running=mock.Mock(return_value=False))
get_loop_mock.return_value = loop
runner = LoaferRunner()
runner.prepare_stop()
loop.is_running.assert_called_once_with()
assert loop.stop.called is False
def test_runner_stop_dont_raise_cancelled_error():
async def some_coro():
raise asyncio.CancelledError()
runner = LoaferRunner()
loop = runner.loop
task = loop.create_task(some_coro())
assert task.done() is False
assert task.cancelled() is False
runner.stop()
assert task.done() is True
assert task.cancelled() is True
with pytest.raises(asyncio.CancelledError):
task.exception()
def test_runner_start_run_until_complete():
runner = LoaferRunner(loop=mock.Mock())
runner.stop = mock.Mock()
runner.start(run_forever=False)
assert runner.loop.run_until_complete.called
assert runner.stop.called
def test_runner_prepare_stop():
runner = LoaferRunner(loop=mock.Mock())
runner.prepare_stop()
runner.loop.stop.assert_called_once_with()
def test_runner_start_with_debug():
loop = mock.Mock()
runner = LoaferRunner(loop=loop)
runner.start(debug=True)
assert loop.set_debug.called_once_with(enabled=True)
def test_runner_start():
runner = LoaferRunner(loop=mock.Mock())
runner.start()
assert runner.loop.run_forever.called
def test_runner_start_with_debug(loop_mock):
runner = LoaferRunner()
runner.start(debug=True)
loop_mock.return_value.set_debug.assert_called_once_with(enabled=True)
def test_runner_with_cancelled_error():
runner = LoaferRunner(loop=mock.Mock())
runner.loop.run_forever.side_effect = CancelledError
runner.start()
assert runner.loop.run_forever.called
assert runner.loop.close.called
def test_runner_stop_with_callback():
callback = mock.Mock()
runner = LoaferRunner(loop=mock.Mock(), on_stop_callback=callback)
runner.stop()
assert runner.loop.stop.called
assert callback.called
def test_runner_start_with_debug():
loop = mock.Mock()
runner = LoaferRunner(loop=loop)
runner.start(debug=True)
loop.set_debug.assert_called_once_with(enabled=True)
def test_runner_prepare_stop():
runner = LoaferRunner(loop=mock.Mock())
runner.prepare_stop()
runner.loop.stop.assert_called_once_with()
def test_runner_prepare_stop_already_stopped():
loop = mock.Mock(is_running=mock.Mock(return_value=False))
runner = LoaferRunner(loop=loop)
runner.prepare_stop()
loop.is_running.assert_called_once_with()
assert loop.stop.called is False
def test_runner_start(loop_mock):
runner = LoaferRunner()
runner.start()
assert loop_mock.return_value.run_forever.called
def test_runner_stop():
runner = LoaferRunner(loop=mock.Mock())
runner.stop()
assert runner.loop.stop.called