FrankYang0529 commented on code in PR #68902:
URL: https://github.com/apache/airflow/pull/68902#discussion_r3525909508
##########
airflow-core/tests/unit/api_fastapi/execution_api/versions/head/test_task_instances.py:
##########
@@ -2794,7 +2794,36 @@ def test_ti_put_rtif_missing_ti(self, client, session,
create_task_instance):
random_id = uuid6.uuid7()
response = client.put(f"/execution/task-instances/{random_id}/rtif",
json=payload)
assert response.status_code == 404
- assert response.json()["detail"] == "Not Found"
+ assert response.json()["detail"] == {
+ "reason": "not_found",
+ "message": "Task Instance not found",
+ }
+
+ def test_ti_put_rtif_archived_ti_returns_410(self, client, session,
create_task_instance):
+ ti = create_task_instance(
+ task_id="test_ti_put_rtif_archived",
+ state=State.RUNNING,
+ session=session,
+ )
+ session.commit()
+ old_ti_id = ti.id
+
+ # Archive the current try to TIH and assign a new UUID, mirroring
prepare_db_for_next_try().
+ ti.prepare_db_for_next_try(session)
+ session.commit()
+
+ assert session.get(TaskInstance, old_ti_id) is None
+
Review Comment:
Thanks for the suggestion. Updated it.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
