def test_should_extend_task_with_values_from_other_task(self): def callable_one(): pass def callable_two(param): pass task = Task("task", callable_one, "dependency", "description") replacement = Task("replacement", callable_two, "another_dependency", "replacement description") task.extend(replacement) self.assertEquals("task", task.name) self.assertEquals(["dependency", "another_dependency"], task.dependencies) self.assertEquals(["description", "replacement description"], task.description)
def test_should_execute_both_callables_when_extending_task(self): def callable_one(): callable_one.called = True callable_one.called = False def callable_two(param): callable_two.called = True callable_two.called = False task_one = Task("task", callable_one) task_two = Task("task", callable_two) task_one.extend(task_two) task_one.execute(mock(), {"param": "spam"}) self.assertTrue(callable_one.called) self.assertTrue(callable_two.called)