From 374f8a1cdfc6f956e6934c4d24e9da0af2eb5e4e Mon Sep 17 00:00:00 2001 From: Julian LaNeve Date: Wed, 7 Jun 2023 12:43:28 -0400 Subject: [PATCH] fix gcp connection type --- cosmos/__init__.py | 2 +- .../dbt/core/profiles/bigquery/service_account_file.py | 2 +- .../bigquery/tests/test_bq_service_account_file.py | 8 ++++---- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/cosmos/__init__.py b/cosmos/__init__.py index e566e1563..7bff9b13c 100644 --- a/cosmos/__init__.py +++ b/cosmos/__init__.py @@ -2,4 +2,4 @@ Astronomer Cosmos is a library for rendering 3rd party workflows in Airflow. """ -__version__ = "0.7.0" +__version__ = "0.7.1" diff --git a/cosmos/providers/dbt/core/profiles/bigquery/service_account_file.py b/cosmos/providers/dbt/core/profiles/bigquery/service_account_file.py index 10e61251b..7befffe90 100644 --- a/cosmos/providers/dbt/core/profiles/bigquery/service_account_file.py +++ b/cosmos/providers/dbt/core/profiles/bigquery/service_account_file.py @@ -14,7 +14,7 @@ class GoogleCloudServiceAccountFileProfileMapping(BaseProfileMapping): https://airflow.apache.org/docs/apache-airflow-providers-google/stable/connections/gcp.html """ - airflow_connection_type: str = "google-cloud-platform" + airflow_connection_type: str = "google_cloud_platform" required_fields = [ "project", diff --git a/cosmos/providers/dbt/core/profiles/bigquery/tests/test_bq_service_account_file.py b/cosmos/providers/dbt/core/profiles/bigquery/tests/test_bq_service_account_file.py index 82d247066..fc1bef96d 100644 --- a/cosmos/providers/dbt/core/profiles/bigquery/tests/test_bq_service_account_file.py +++ b/cosmos/providers/dbt/core/profiles/bigquery/tests/test_bq_service_account_file.py @@ -23,7 +23,7 @@ def mock_bigquery_conn(): # type: ignore } conn = Connection( conn_id="my_bigquery_connection", - conn_type="google-cloud-platform", + conn_type="google_cloud_platform", extra=json.dumps(extra), ) @@ -52,7 +52,7 @@ def test_connection_claiming() -> None: del values[key] conn = Connection( conn_id="my_bigquery_connection", - conn_type="google-cloud-platform", + conn_type="google_cloud_platform", extra=json.dumps(values), ) @@ -64,7 +64,7 @@ def test_connection_claiming() -> None: # also test when there's no schema conn = Connection( conn_id="my_bigquery_connection", - conn_type="google-cloud-platform", + conn_type="google_cloud_platform", extra=json.dumps(extra), ) profile_mapping = GoogleCloudServiceAccountFileProfileMapping(conn, {}) @@ -73,7 +73,7 @@ def test_connection_claiming() -> None: # if we have them all, it should claim conn = Connection( conn_id="my_bigquery_connection", - conn_type="google-cloud-platform", + conn_type="google_cloud_platform", extra=json.dumps(extra), ) profile_mapping = GoogleCloudServiceAccountFileProfileMapping(conn, {"dataset": "my_dataset"})