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)