def test_get_execution_or_self_with_no_execution(self, sample_task):
task = Task(sample_task, None)
execution = task.get_execution_or_self(999)
assert execution == task
def test_get_execution_or_self(self, sample_task):
task = Task(sample_task, None)
execution = task.get_execution_or_self(0)
assert execution.display_name == "sharding_auth_gen"
