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)