From b1eb5e2ece869cd938a06a846389b314d3d5ef07 Mon Sep 17 00:00:00 2001
From: Ben Cassell <98852248+benc-db@users.noreply.github.com>
Date: Thu, 1 Aug 2024 09:56:25 -0700
Subject: [PATCH] Allow configuring port for tests (#755)

---
 tests/profiles.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/profiles.py b/tests/profiles.py
index 940d2e2a..2c80af44 100644
--- 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