def test_prepend_task_empty_relations(self):
task_1 = Task(task_id=TEST_MAPPER_NAME + "_1", template_name="pig.tpl")
task_2 = Task(task_id=TEST_MAPPER_NAME + "_2", template_name="pig.tpl")
tasks, relations = ActionMapper.prepend_task(task_to_prepend=task_1,
tasks=[task_2],
relations=[])
self.assertEqual([task_1, task_2], tasks)
self.assertEqual([
Relation(from_task_id="mapper_name_1", to_task_id="mapper_name_2")
], relations)
def test_prepend_task_no_tasks(self):
task_1 = Task(task_id=TEST_MAPPER_NAME + "_1", template_name="pig.tpl")
with self.assertRaises(IndexError):
ActionMapper.prepend_task(task_to_prepend=task_1,
tasks=[],
relations=[])