(airflow) branch main updated: Enable plugin tests in test_databricks_workflow.py‎ for Airflow 3.0+ (#66442)

[shahar]

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 c7566378058 Enable plugin tests in test_databricks_workflow.py‎ for 
Airflow 3.0+ (#66442)
c7566378058 is described below

commit c756637805860a799dcc1367c7a26204e79d60f3
Author: Roman Karichev <104169193+zin...@users.noreply.github.com>
AuthorDate: Wed May 6 05:59:23 2026 +0200

    Enable plugin tests in test_databricks_workflow.py‎ for Airflow 3.0+ 
(#66442)
---
 .../unit/databricks/plugins/test_databricks_workflow.py | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git 
a/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py
 
b/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py
index 5f3a7ac885b..e021724c2bd 100644
--- 
a/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py
+++ 
b/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py
@@ -24,14 +24,6 @@ import pytest
 
 pytest.importorskip("airflow.providers.fab")
 
-from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS
-
-if AIRFLOW_V_3_0_PLUS:
-    pytest.skip(
-        "``airflow/providers/databricks/plugins/databricks_workflow.py`` is 
only compatible with Airflow 2.X.",
-        allow_module_level=True,
-    )
-
 from flask import url_for
 
 from airflow.models.dagrun import DagRun
@@ -39,17 +31,22 @@ from airflow.models.taskinstance import TaskInstanceKey
 from airflow.providers.common.compat.sdk import AirflowException, AirflowPlugin
 from airflow.providers.databricks.plugins.databricks_workflow import (
     DatabricksWorkflowPlugin,
-    RepairDatabricksTasks,
     WorkflowJobRepairSingleTaskLink,
     WorkflowJobRunLink,
     _get_launch_task_key,
-    _repair_task,
     get_databricks_task_ids,
     get_launch_task_id,
     store_databricks_job_run_link,
 )
 
 from tests_common import RUNNING_TESTS_AGAINST_AIRFLOW_PACKAGES
+from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS
+
+if not AIRFLOW_V_3_0_PLUS:
+    from airflow.providers.databricks.plugins.databricks_workflow import (
+        RepairDatabricksTasks,
+        _repair_task,
+    )
 
 DAG_ID = "test_dag"
 TASK_ID = "test_task"