Skip to content

Commit c0e6780

Browse files
committed
feat(api_connexion): add from_rest_api to extra of dataset event created through api
1 parent f172131 commit c0e6780

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

airflow/api_connexion/endpoints/dataset_endpoint.py

+1
Original file line numberDiff line numberDiff line change
@@ -347,6 +347,7 @@ def create_dataset_event(session: Session = NEW_SESSION) -> APIResponse:
347347
raise NotFound(title="Dataset not found", detail=f"Dataset with uri: '{uri}' not found")
348348
timestamp = timezone.utcnow()
349349
extra = json_body.get("extra", {})
350+
extra["from_rest_api"] = True
350351
dataset_event = dataset_manager.register_dataset_change(
351352
dataset=Dataset(uri),
352353
timestamp=timestamp,

tests/api_connexion/endpoints/test_dataset_endpoint.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,7 @@ def test_should_respond_200(self, session):
604604
"created_dagruns": [],
605605
"dataset_uri": event_payload["dataset_uri"],
606606
"dataset_id": ANY,
607-
"extra": {"foo": "bar"},
607+
"extra": {"foo": "bar", "from_rest_api": True},
608608
"source_dag_id": None,
609609
"source_task_id": None,
610610
"source_run_id": None,

0 commit comments

Comments
 (0)