fix gcp connection type

astronomer · Jun 7, 2023 · 374f8a1 · 374f8a1
1 parent d884f58
commit 374f8a1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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"})