def test_completed_time_one_completed_attempt():
task_created_time = datetime.datetime(2018, 1, 15, 12, 35, 0)
task = Task("1234", "some cmd", task_created_time, max_attempts=2)
attempt_time = datetime.datetime(2018, 1, 15, 12, 35, 30)
attempt = task.attempt_task("runner 1", attempt_time)
attempt_completed_time = datetime.datetime(2018, 1, 15, 12, 35, 45)
attempt.mark_completed(attempt_completed_time)
assert task.completed_time() == attempt_completed_time
def test_completed_time_one_failed_attempt():
task = Task("1234",
"some cmd",
datetime.datetime(2018, 1, 15, 12, 35, 0),
max_attempts=2)
attempt_time = datetime.datetime(2018, 1, 15, 12, 35, 30)
attempt = task.attempt_task("runner 1", attempt_time)
attempt.mark_failed(datetime.datetime(2018, 1, 15, 12, 35, 45))
assert task.completed_time() is None
def test_completed_time_new_task():
task_created_time = datetime.datetime(2018, 1, 15, 12, 35, 0)
task = Task("1234", "some cmd", task_created_time, max_attempts=2)
assert task.completed_time() is None
