diff --git a/dbt/adapters/hive/connections.py b/dbt/adapters/hive/connections.py index 5383575..751e722 100644 --- a/dbt/adapters/hive/connections.py +++ b/dbt/adapters/hive/connections.py @@ -65,6 +65,7 @@ class HiveCredentials(Credentials): use_http_transport: Optional[bool] = True http_path: Optional[str] = None kerberos_service_name: Optional[str] = None + krb_host: Optional[str] = None usage_tracking: Optional[bool] = True # usage tracking is enabled by default _ALIASES = {"pass": "password", "user": "username"} @@ -226,6 +227,7 @@ def open(cls, connection): port=credentials.port, auth_mechanism="GSSAPI", kerberos_service_name=credentials.kerberos_service_name, + krb_host=credentials.krb_host, use_http_transport=credentials.use_http_transport, use_ssl=credentials.use_ssl, ca_cert=credentials.ca_cert, diff --git a/tests/conftest.py b/tests/conftest.py index 4a5b26f..46612c3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -63,6 +63,7 @@ def cdh_kerberos_target(): "port": int(os.getenv("DBT_HIVE_PORT")), "schema": os.getenv("DBT_HIVE_SCHEMA") or "dbt_adapter_test", "kerberos_service_name": os.getenv("DBT_HIVE_KERBEROS_SERVICE_NAME") or "hive", + "krb_host": os.getenv("DBT_HIVE_KRB_HOST") or None, "use_http_transport": bool(strtobool(os.getenv("DBT_HIVE_USE_HTTP_TRANSPORT") or "f")), "use_ssl": bool(strtobool(os.getenv("DBT_HIVE_USE_SSL") or "f")), "ca_cert": os.getenv("DBT_CA_CERT") or None,