diff --git a/airflow_dbt_python/hooks/target.py b/airflow_dbt_python/hooks/target.py index 4ef0512..f314d46 100644 --- a/airflow_dbt_python/hooks/target.py +++ b/airflow_dbt_python/hooks/target.py @@ -515,6 +515,7 @@ class DbtTrinoHook(DbtConnectionHook): DbtConnectionParam("method", default="none"), DbtConnectionParam("http_scheme", default="https"), DbtConnectionParam("verify", default=True), + DbtConnectionParam("cert", default=None), DbtConnectionParam("database", "catalog", "hive"), "catalog", "host", diff --git a/tests/hooks/test_trino.py b/tests/hooks/test_trino.py index 2bdb3d0..db834cd 100644 --- a/tests/hooks/test_trino.py +++ b/tests/hooks/test_trino.py @@ -21,7 +21,8 @@ def test_trino_registration(): "http_scheme": "https", "database": "example", "schema": "example", - "verify": false + "verify": false, + "cert": false } """, ) @@ -38,6 +39,7 @@ def test_trino_registration(): assert details["schema"] == "example" assert details["catalog"] == "example" assert details["verify"] is False + assert details["cert"] is False def test_trino_catalog_overrides_database():