def test_adapter_results_with_pretask(tmpdir):
temp_path = tmpdir.mkdir("test")
pretask = SimplePreTask("Starting")
builder = TaskBuilder(SimpleTaskBuilder(), temp_path)
builder.set_pretask(subtask=pretask)
builder.add_subtask(subtask=SimpleSubtask("First"))
builder.add_subtask(subtask=SimpleSubtask("Second"))
new_task = builder.build_task()
with worker.ToolJobManager() as manager:
for subtask in new_task.subtasks:
adapted_tool = speedwagon.worker.SubtaskJobAdapter(subtask)
manager.add_job(adapted_tool, adapted_tool.settings)
manager.start()
results = list()
for r in manager.get_results():
results.append(r.data)
assert len(results) == 3
assert "Starting" == results[0]
assert "First" == results[1]
assert "Second" == results[2]
shutil.rmtree(tmpdir)
shortcut = os.path.join(tmpdir.dirname,
"test_adapter_results_with_pretcurrent")
os.unlink(shortcut)
def test_pretask_builder(tmpdir):
temp_path = tmpdir.mkdir("test")
pretask = SimplePreTask("Starting")
builder = TaskBuilder(SimpleTaskBuilder(), temp_path)
builder.set_pretask(subtask=pretask)
builder.add_subtask(subtask=SimpleSubtask("First"))
builder.add_subtask(subtask=SimpleSubtask("Second"))
task = builder.build_task()
assert task.pretask == pretask
shutil.rmtree(tmpdir)
shortcut = os.path.join(tmpdir.dirname, "test_pretask_buildercurrent")
os.unlink(shortcut)