diff --git a/cosmos/profiles/clickhouse/user_pass.py b/cosmos/profiles/clickhouse/user_pass.py index 7d168895a..e6f0ac9c2 100644 --- a/cosmos/profiles/clickhouse/user_pass.py +++ b/cosmos/profiles/clickhouse/user_pass.py @@ -1,4 +1,4 @@ -"""Maps Airflow Postgres connections using user + password authentication to dbt profiles.""" +"""Maps Airflow Clickhouse connections using user + password authentication to dbt profiles.""" from __future__ import annotations @@ -20,7 +20,7 @@ class ClickhouseUserPasswordProfileMapping(BaseProfileMapping): required_fields = [ "host", - "login", + "user", "schema", "clickhouse", ] @@ -29,7 +29,7 @@ class ClickhouseUserPasswordProfileMapping(BaseProfileMapping): ] airflow_param_mapping = { "host": "host", - "login": "login", + "user": "login", "password": "password", "port": "port", "schema": "schema", diff --git a/tests/profiles/clickhouse/test_clickhouse_userpass.py b/tests/profiles/clickhouse/test_clickhouse_userpass.py index 1f623c803..a140427b5 100644 --- a/tests/profiles/clickhouse/test_clickhouse_userpass.py +++ b/tests/profiles/clickhouse/test_clickhouse_userpass.py @@ -83,7 +83,7 @@ def test_profile_args(mock_clickhouse_conn: Connection) -> None: assert profile_mapping.profile == { "type": "clickhouse", "schema": mock_clickhouse_conn.schema, - "login": mock_clickhouse_conn.login, + "user": mock_clickhouse_conn.login, "password": "{{ env_var('COSMOS_CONN_GENERIC_PASSWORD') }}", "driver": "native", "port": 9000, @@ -102,7 +102,7 @@ def test_mock_profile() -> None: assert profile_mapping.mock_profile == { "type": "clickhouse", "schema": "mock_value", - "login": "mock_value", + "user": "mock_value", "driver": "native", "port": 9000, "host": "mock_value",