diff --git a/tests/test_runner.py b/tests/test_runner.py index 5f4e55e..5950db4 100644 --- a/tests/test_runner.py +++ b/tests/test_runner.py @@ -20,13 +20,17 @@ def runner(self, interpreter): r.stop() @pytest.fixture() - def mocked_runner(self, runner, mocker): - mocker.patch.object(runner, 'before_run') - mocker.patch.object(runner, 'before_execute') - mocker.patch.object(runner, 'after_execute') - mocker.patch.object(runner, 'after_run') - return runner - + def mocked_runner(self, interpreter, mocker): + class MockedRunner(AsyncRunner): + before_run = mocker.MagicMock() + before_execute = mocker.MagicMock() + after_execute = mocker.MagicMock() + after_run = mocker.MagicMock() + + r = MockedRunner(interpreter, interval=self.INTERVAL) + yield r + r.stop() + def test_not_yet_started(self, runner): assert runner.interpreter.configuration == []