This is an automated email from the ASF dual-hosted git repository.
shahar1 pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 775b7e4cfdb Serialize poll_interval and impersonation_chain on
DataFusionStartPipelineTrigger (#66968)
775b7e4cfdb is described below
commit 775b7e4cfdb24647c19279574cdf00a30fe5c58d
Author: Nishita Matlani <[email protected]>
AuthorDate: Fri May 15 12:26:14 2026 -0400
Serialize poll_interval and impersonation_chain on
DataFusionStartPipelineTrigger (#66968)
---
.../providers/google/cloud/triggers/datafusion.py | 2 ++
.../unit/google/cloud/triggers/test_datafusion.py | 18 ++++++++++++++++++
2 files changed, 20 insertions(+)
diff --git
a/providers/google/src/airflow/providers/google/cloud/triggers/datafusion.py
b/providers/google/src/airflow/providers/google/cloud/triggers/datafusion.py
index 0c900a811d8..c7d1acc852a 100644
--- a/providers/google/src/airflow/providers/google/cloud/triggers/datafusion.py
+++ b/providers/google/src/airflow/providers/google/cloud/triggers/datafusion.py
@@ -82,6 +82,8 @@ class DataFusionStartPipelineTrigger(BaseTrigger):
"pipeline_name": self.pipeline_name,
"pipeline_id": self.pipeline_id,
"pipeline_type": self.pipeline_type,
+ "poll_interval": self.poll_interval,
+ "impersonation_chain": self.impersonation_chain,
"success_states": self.success_states,
},
)
diff --git
a/providers/google/tests/unit/google/cloud/triggers/test_datafusion.py
b/providers/google/tests/unit/google/cloud/triggers/test_datafusion.py
index 6cc2397bafd..487875b36d0 100644
--- a/providers/google/tests/unit/google/cloud/triggers/test_datafusion.py
+++ b/providers/google/tests/unit/google/cloud/triggers/test_datafusion.py
@@ -71,10 +71,28 @@ class TestDataFusionStartPipelineTrigger:
"pipeline_name": PIPELINE_NAME,
"pipeline_id": PIPELINE_ID,
"pipeline_type": PIPELINE_TYPE,
+ "poll_interval": TEST_POLL_INTERVAL,
"gcp_conn_id": TEST_GCP_PROJECT_ID,
+ "impersonation_chain": None,
"success_states": None,
}
+ def
test_start_pipeline_trigger_serialize_round_trip_poll_and_impersonation(self):
+ trigger = DataFusionStartPipelineTrigger(
+ instance_url=INSTANCE_URL,
+ namespace=NAMESPACE,
+ pipeline_name=PIPELINE_NAME,
+ pipeline_id=PIPELINE_ID,
+ pipeline_type=PIPELINE_TYPE,
+ poll_interval=7.5,
+ gcp_conn_id=TEST_GCP_PROJECT_ID,
+ impersonation_chain="[email protected]",
+ )
+ _, kwargs = trigger.serialize()
+ restored = DataFusionStartPipelineTrigger(**kwargs)
+ assert restored.poll_interval == 7.5
+ assert restored.impersonation_chain ==
"[email protected]"
+
@pytest.mark.asyncio
@mock.patch(HOOK_STATUS_STR)
async def
test_start_pipeline_trigger_on_success_should_execute_successfully(
