def multiprocessing_helper(executor):
client = MagicMock()
monkeypatch.setattr(
"prefect.engine.cloud.task_runner.Client",
MagicMock(return_value=client),
)
runner = CloudTaskRunner(task=sleeper)
runner._heartbeat = update
with set_temporary_config({"cloud.heartbeat_interval": 0.025}):
return runner.run(executor=executor)
def multiprocessing_helper(executor):
client = MagicMock()
monkeypatch.setattr(
"prefect.engine.cloud.task_runner.Client",
MagicMock(return_value=client),
)
runner = CloudTaskRunner(task=Task())
runner.cache_result = lambda *args, **kwargs: time.sleep(0.2)
runner._heartbeat = update
with set_temporary_config({"cloud.heartbeat_interval": 0.05}):
return runner.run(executor=executor)
def multiprocessing_helper(executor):
set_task_run_state = MagicMock(
side_effect=lambda task_run_id, version, state, cache_for:
state)
client = MagicMock(set_task_run_state=set_task_run_state)
monkeypatch.setattr(
"prefect.engine.cloud.task_runner.Client",
MagicMock(return_value=client),
)
runner = CloudTaskRunner(task=sleeper)
runner._heartbeat = update
with set_temporary_config({"cloud.heartbeat_interval": 0.025}):
return runner.run(executor=executor)
