def test_get_post_tasks(self, workflow):
caller = Mock()
task_generator = runner_strategies.TaskGenerator(
workflow=workflow,
options={},
working_directory=os.path.join("some", "real", "directory"),
caller=caller)
list(task_generator.get_post_tasks("dummy", []))
assert workflow.completion_task.called is True
def test_tasks_runs_completion_task(self, workflow):
task_generator = runner_strategies.TaskGenerator(
workflow=workflow,
options={},
working_directory=os.path.join("some", "real", "directory"))
for subtask in task_generator.tasks():
assert isinstance(subtask, speedwagon.tasks.Subtask)
assert workflow.completion_task.called is True
def test_tasks_request_more_info(self, workflow):
caller = Mock()
task_generator = runner_strategies.TaskGenerator(
workflow=workflow,
options={},
working_directory=os.path.join("some", "real", "directory"),
caller=caller)
for subtask in task_generator.tasks():
assert isinstance(subtask, speedwagon.tasks.Subtask)
assert caller.request_more_info.called is True