ai: apply changes for #915 (1 review thread)

Addresses:
  - #3802164062 at src/databricks/sql/session.py:71

Signed-off-by: peco-engineer-bot[bot] <peco-engineer-bot[bot]@users.noreply.github.com>

Author: peco-engineer-bot[bot]

src/databricks/sql/session.py

@@ -75,7 +75,7 @@ def _kernel_host_and_path(
             # unconditionally would otherwise yield a malformed double-``?``
             # path (e.g. ``/warehouses/abc?o=111?o=222``) that mis-parses
             # downstream and silently drops the org-id header.
-            path = "{}?{}".format(path.split("?", 1)[0], parts.query)
+            path = "{}?{}".format((path or "").split("?", 1)[0], parts.query)
         return host, path
 
     port = kwargs.get("_port")

tests/unit/test_session.py

@@ -792,6 +792,16 @@ def test_connection_uri_without_path_query_replaces_original_path_query(self):
         assert host == "https://h.example.com:8443"
         assert path == "/sql/1.0/warehouses/abc?o=222"
 
+    def test_connection_uri_without_path_and_none_http_path_applies_query(self):
+        # http_path is legitimately nullable on the connect path. A path-less,
+        # query-bearing URI combined with http_path=None must not raise; the
+        # retained path coerces to empty before the query is applied.
+        host, path = _kernel_host_and_path(
+            self.HOST, None, {"_connection_uri": "https://h.example.com:8443?o=222"}
+        )
+        assert host == "https://h.example.com:8443"
+        assert path == "?o=222"
+
     def test_connection_uri_wins_over_port(self):
         host, path = _kernel_host_and_path(
             self.HOST,