diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index dcac01496..379b96b65 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -140,7 +140,7 @@ jobs: hatch run tests.py${{ matrix.python-version }}-${{ matrix.airflow-version }}:test-integration env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres DATABRICKS_HOST: mock DATABRICKS_WAREHOUSE_ID: mock DATABRICKS_TOKEN: mock @@ -210,7 +210,7 @@ jobs: DATABRICKS_UNIQUE_ID="${{github.run_id}}" hatch run tests.py${{ matrix.python-version }}-${{ matrix.airflow-version }}:test-integration-expensive env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH AIRFLOW_CONN_DATABRICKS_DEFAULT: ${{ secrets.AIRFLOW_CONN_DATABRICKS_DEFAULT }} DATABRICKS_CLUSTER_ID: ${{ secrets.DATABRICKS_CLUSTER_ID }} @@ -231,7 +231,7 @@ jobs: env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH AIRFLOW_CONN_DATABRICKS_DEFAULT: ${{ secrets.AIRFLOW_CONN_DATABRICKS_DEFAULT }} DATABRICKS_CLUSTER_ID: ${{ secrets.DATABRICKS_CLUSTER_ID }} @@ -272,7 +272,7 @@ jobs: hatch run tests.py${{ matrix.python-version }}-${{ matrix.airflow-version }}:test-integration-sqlite env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres AIRFLOW__CORE__DAGBAG_IMPORT_TIMEOUT: 90.0 PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH COSMOS_CONN_POSTGRES_PASSWORD: ${{ secrets.COSMOS_CONN_POSTGRES_PASSWORD }} @@ -295,7 +295,7 @@ jobs: env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH Run-Integration-Tests-DBT-1-5-4: @@ -345,7 +345,7 @@ jobs: hatch run tests.py${{ matrix.python-version }}-${{ matrix.airflow-version }}:test-integration-dbt-1-5-4 env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres AIRFLOW__CORE__DAGBAG_IMPORT_TIMEOUT: 90.0 PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH COSMOS_CONN_POSTGRES_PASSWORD: ${{ secrets.COSMOS_CONN_POSTGRES_PASSWORD }} @@ -368,7 +368,7 @@ jobs: env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH Run-Performance-Tests: @@ -424,7 +424,7 @@ jobs: cat /tmp/performance_results.txt > $GITHUB_STEP_SUMMARY env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres AIRFLOW__CORE__DAGBAG_IMPORT_TIMEOUT: 90.0 PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH COSMOS_CONN_POSTGRES_PASSWORD: ${{ secrets.COSMOS_CONN_POSTGRES_PASSWORD }} @@ -437,7 +437,7 @@ jobs: MODEL_COUNT: ${{ matrix.num-models }} env: AIRFLOW_HOME: /home/runner/work/astronomer-cosmos/astronomer-cosmos/ - AIRFLOW_CONN_AIRFLOW_DB: postgres://postgres:postgres@0.0.0.0:5432/postgres + AIRFLOW_CONN_EXAMPLE_CONN: postgres://postgres:postgres@0.0.0.0:5432/postgres PYTHONPATH: /home/runner/work/astronomer-cosmos/astronomer-cosmos/:$PYTHONPATH Code-Coverage: diff --git a/dev/dags/basic_cosmos_dag.py b/dev/dags/basic_cosmos_dag.py index b4f5fed77..f71f351f0 100644 --- a/dev/dags/basic_cosmos_dag.py +++ b/dev/dags/basic_cosmos_dag.py @@ -16,7 +16,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, disable_event_tracking=True, ), diff --git a/dev/dags/basic_cosmos_task_group.py b/dev/dags/basic_cosmos_task_group.py index 209c1a258..4221e3019 100644 --- a/dev/dags/basic_cosmos_task_group.py +++ b/dev/dags/basic_cosmos_task_group.py @@ -20,7 +20,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/dev/dags/cosmos_manifest_example.py b/dev/dags/cosmos_manifest_example.py index 11df6ac94..225b38ddc 100644 --- a/dev/dags/cosmos_manifest_example.py +++ b/dev/dags/cosmos_manifest_example.py @@ -16,7 +16,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, dbt_config_vars=DbtProfileConfigVars(send_anonymous_usage_stats=True), ), diff --git a/dev/dags/cosmos_seed_dag.py b/dev/dags/cosmos_seed_dag.py index b682ecc36..d0b4f4103 100644 --- a/dev/dags/cosmos_seed_dag.py +++ b/dev/dags/cosmos_seed_dag.py @@ -29,7 +29,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/dev/dags/example_model_version.py b/dev/dags/example_model_version.py index 5ff66fe31..df9205f3e 100644 --- a/dev/dags/example_model_version.py +++ b/dev/dags/example_model_version.py @@ -16,7 +16,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/dev/dags/example_virtualenv.py b/dev/dags/example_virtualenv.py index 36f1b3ac3..cd38cba9e 100644 --- a/dev/dags/example_virtualenv.py +++ b/dev/dags/example_virtualenv.py @@ -17,7 +17,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/dev/dags/performance_dag.py b/dev/dags/performance_dag.py index 0cd6e6b09..3ed23c94c 100644 --- a/dev/dags/performance_dag.py +++ b/dev/dags/performance_dag.py @@ -17,7 +17,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/tests/dbt/test_graph.py b/tests/dbt/test_graph.py index c16571966..a08b4e5c0 100644 --- a/tests/dbt/test_graph.py +++ b/tests/dbt/test_graph.py @@ -51,7 +51,7 @@ def postgres_profile_config() -> ProfileConfig: profile_name="default", target_name="default", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), ) diff --git a/tests/operators/test_local.py b/tests/operators/test_local.py index f90237082..ed46caf88 100644 --- a/tests/operators/test_local.py +++ b/tests/operators/test_local.py @@ -56,7 +56,7 @@ profile_name="default", target_name="dev", profile_mapping=PostgresUserPasswordProfileMapping( - conn_id="airflow_db", + conn_id="example_conn", profile_args={"schema": "public"}, ), )