| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| from __future__ import annotations |
| |
| import warnings |
| |
| import pytest |
| |
| from airflow.exceptions import RemovedInAirflow3Warning |
| from airflow.www import app |
| from tests.test_utils.config import conf_vars |
| from tests.test_utils.decorators import dont_initialize_flask_app_submodules |
| |
| |
| @pytest.fixture(scope="session") |
| def minimal_app_for_api(): |
| @dont_initialize_flask_app_submodules( |
| skip_all_except=[ |
| "init_appbuilder", |
| "init_api_experimental_auth", |
| "init_api_connexion", |
| "init_airflow_session_interface", |
| "init_appbuilder_views", |
| ] |
| ) |
| def factory(): |
| with conf_vars({("api", "auth_backends"): "tests.test_utils.remote_user_api_auth_backend"}): |
| return app.create_app(testing=True, config={"WTF_CSRF_ENABLED": False}) # type:ignore |
| |
| return factory() |
| |
| |
| @pytest.fixture |
| def session(): |
| from airflow.utils.session import create_session |
| |
| with create_session() as session: |
| yield session |
| |
| |
| @pytest.fixture(scope="module") |
| def dagbag(): |
| from airflow.models import DagBag |
| |
| with warnings.catch_warnings(): |
| # This explicitly shows off SubDagOperator, no point to warn about that. |
| warnings.filterwarnings( |
| "ignore", |
| category=RemovedInAirflow3Warning, |
| message=r".+Please use.+TaskGroup.+", |
| module=r".+example_subdag_operator$", |
| ) |
| DagBag(include_examples=True, read_dags_from_db=False).sync_to_db() |
| return DagBag(include_examples=True, read_dags_from_db=True) |