Allow configuring port for tests (#755)

databricks · Aug 1, 2024 · b1eb5e2 · b1eb5e2
1 parent 1b89b97
commit b1eb5e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/profiles.py b/tests/profiles.py
@@ -39,6 +39,8 @@ def _build_databricks_cluster_target(
         profile["schema"] = schema
     if session_properties is not None:
         profile["session_properties"] = session_properties
+    if os.getenv("DBT_DATABRICKS_PORT"):
+        profile["connection_parameters"] = {"_port": os.getenv("DBT_DATABRICKS_PORT")}
     return profile