diff --git a/examples/experimental/dagster-airlift/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/__init__.py b/examples/experimental/dagster-airlift/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/__init__.py new file mode 100644 index 0000000000000..e69de29bb2d1d diff --git a/examples/experimental/dagster-airlift/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/dags.py b/examples/experimental/dagster-airlift/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/dags.py new file mode 100644 index 0000000000000..e31b68f0eb121 --- /dev/null +++ b/examples/experimental/dagster-airlift/dagster_airlift_tests/integration_tests/scaffold_test_airflow_home/airflow_dags/dags.py @@ -0,0 +1,33 @@ +from datetime import datetime + +from airflow import DAG +from airflow.operators.python import PythonOperator + + +def print_hello() -> None: + print("Hello") # noqa: T201 + + +default_args = { + "owner": "airflow", + "depends_on_past": False, + "start_date": datetime(2023, 1, 1), + "retries": 0, +} + + +def make_print_dag(dag_id: str) -> DAG: + with DAG( + dag_id, + default_args=default_args, + schedule_interval=None, + is_paused_upon_creation=False, + ) as dag: + PythonOperator(task_id="print_task", python_callable=print_hello) >> PythonOperator( + task_id="downstream_print_task", python_callable=print_hello + ) # type: ignore + return dag + + +print_dag = make_print_dag("print_dag") +other_print_dag = make_print_dag("other_print_dag")