tirkarthi commented on code in PR #23530:
URL: https://github.com/apache/airflow/pull/23530#discussion_r878645565
##########
tests/api_connexion/endpoints/test_task_instance_endpoint.py:
##########
@@ -1491,3 +1492,111 @@ def
test_should_raise_400_for_naive_and_bad_datetime(self, payload, expected, se
)
assert response.status_code == 400
assert response.json['detail'] == expected
+
+
+class _ForceHeartbeatCeleryExecutor(CeleryExecutor):
+ def heartbeat(self):
+ return True
+
+
[email protected](
+ "airflow.executors.executor_loader.ExecutorLoader.get_default_executor",
+ return_value=_ForceHeartbeatCeleryExecutor(),
+)
+class TestTaskInstanceRunEndpoint(TestTaskInstanceEndpoint):
Review Comment:
Thanks, this API uses same permissions as setting task instance state. I
have added similar tests.
--
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]
