import bashoperator using test utils compat in views/utils

apache · Sep 19, 2024 · c5f4c04 · c5f4c04
1 parent 4148020
commit c5f4c04
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 8 deletions.
diff --git a/dev/breeze/tests/test_selective_checks.py b/dev/breeze/tests/test_selective_checks.py
@@ -706,7 +706,7 @@ def assert_outputs_are_printed(expected_outputs: dict[str, str], stderr: str):
         pytest.param(
             ("airflow/providers/standard/core/operators/bash.py",),
             {
-                "affected-providers-list-as-string": "edge standard",
+                "affected-providers-list-as-string": "celery edge standard",
                 "all-python-versions": "['3.8']",
                 "all-python-versions-list-as-string": "3.8",
                 "python-versions": "['3.8']",
@@ -1723,7 +1723,7 @@ def test_upgrade_to_newer_dependencies(
         ),
         pytest.param(
             ("airflow/providers/celery/file.py",),
-            {"docs-list-as-string": "apache-airflow celery cncf.kubernetes"},
+            {"docs-list-as-string": "apache-airflow celery cncf.kubernetes standard"},
             id="Celery python files changed",
         ),
         pytest.param(

diff --git a/tests/utils/test_dot_renderer.py b/tests/utils/test_dot_renderer.py
@@ -25,11 +25,11 @@
 from airflow.models.dag import DAG
 from airflow.operators.empty import EmptyOperator
 from airflow.operators.python import PythonOperator
-from airflow.providers.standard.core.operators.bash import BashOperator
 from airflow.serialization.dag_dependency import DagDependency
 from airflow.utils import dot_renderer, timezone
 from airflow.utils.state import State
 from airflow.utils.task_group import TaskGroup
+from tests.test_utils.compat import BashOperator
 from tests.test_utils.db import clear_db_dags
 
 START_DATE = timezone.utcnow()

diff --git a/tests/utils/test_task_group.py b/tests/utils/test_task_group.py
@@ -36,10 +36,10 @@
 from airflow.models.xcom_arg import XComArg
 from airflow.operators.empty import EmptyOperator
 from airflow.operators.python import PythonOperator
-from airflow.providers.standard.core.operators.bash import BashOperator
 from airflow.utils.dag_edges import dag_edges
 from airflow.utils.task_group import TASKGROUP_ARGS_EXPECTED_TYPES, TaskGroup, task_group_to_dict
 from tests.models import DEFAULT_DATE
+from tests.test_utils.compat import BashOperator
 
 
 def make_task(name, type_="classic"):

diff --git a/tests/www/views/test_views_rendered.py b/tests/www/views/test_views_rendered.py
@@ -29,14 +29,13 @@
 from airflow.models.renderedtifields import RenderedTaskInstanceFields
 from airflow.models.variable import Variable
 from airflow.operators.python import PythonOperator
-from airflow.providers.standard.core.operators.bash import BashOperator
 from airflow.serialization.serialized_objects import SerializedDAG
 from airflow.utils import timezone
 from airflow.utils.session import create_session
 from airflow.utils.state import DagRunState, TaskInstanceState
 from airflow.utils.types import DagRunType
 from tests.conftest import initial_db_init
-from tests.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests.test_utils.compat import AIRFLOW_V_3_0_PLUS, BashOperator
 from tests.test_utils.db import clear_db_dags, clear_db_runs, clear_rendered_ti_fields
 from tests.test_utils.www import check_content_in_response, check_content_not_in_response
 

diff --git a/tests/www/views/test_views_tasks.py b/tests/www/views/test_views_tasks.py
@@ -36,7 +36,6 @@
 from airflow.models.xcom import XCom
 from airflow.operators.empty import EmptyOperator
 from airflow.providers.celery.executors.celery_executor import CeleryExecutor
-from airflow.providers.standard.core.operators.bash import BashOperator
 from airflow.security import permissions
 from airflow.utils import timezone
 from airflow.utils.log.logging_mixin import ExternalLoggingMixin
@@ -45,7 +44,7 @@
 from airflow.utils.types import DagRunType
 from airflow.www.views import TaskInstanceModelView, _safe_parse_datetime
 from tests.test_utils.api_connexion_utils import create_user, delete_roles, delete_user
-from tests.test_utils.compat import AIRFLOW_V_3_0_PLUS
+from tests.test_utils.compat import AIRFLOW_V_3_0_PLUS, BashOperator
 from tests.test_utils.config import conf_vars
 from tests.test_utils.db import clear_db_runs, clear_db_xcom
 from tests.test_utils.www import check_content_in_response, check_content_not_in_response, client_with_login