Added impacted files using util method

Author: nikhilsuri-db

src/databricks/sql/common/feature_flag.py

@@ -6,6 +6,7 @@
 from typing import Dict, Optional, List, Any, TYPE_CHECKING
 
 from databricks.sql.common.http import HttpMethod
+from databricks.sql.common.url_utils import normalize_host_with_protocol
 
 if TYPE_CHECKING:
     from databricks.sql.client import Connection
@@ -67,7 +68,7 @@ def __init__(
 
         endpoint_suffix = FEATURE_FLAGS_ENDPOINT_SUFFIX_FORMAT.format(__version__)
         self._feature_flag_endpoint = (
-            f"https://{self._connection.session.host}{endpoint_suffix}"
+            normalize_host_with_protocol(self._connection.session.host) + endpoint_suffix
         )
 
         # Use the provided HTTP client

src/databricks/sql/telemetry/telemetry_client.py

@@ -47,6 +47,7 @@
     TelemetryPushClient,
     CircuitBreakerTelemetryPushClient,
 )
+from databricks.sql.common.url_utils import normalize_host_with_protocol
 
 if TYPE_CHECKING:
     from databricks.sql.client import Connection
@@ -278,7 +279,7 @@ def _send_telemetry(self, events):
             if self._auth_provider
             else self.TELEMETRY_UNAUTHENTICATED_PATH
         )
-        url = f"https://{self._host_url}{path}"
+        url = normalize_host_with_protocol(self._host_url) + path
 
         headers = {"Accept": "application/json", "Content-Type": "application/json"}
 

tests/unit/test_url_utils.py

@@ -28,12 +28,7 @@ class TestNormalizeHostWithProtocol:
         ("myserver.com/", "https://myserver.com"),
         ("https://myserver.com/", "https://myserver.com"),
         ("http://localhost/", "http://localhost"),
-        
-        # Real Databricks patterns
-        ("adb-123456.azuredatabricks.net", "https://adb-123456.azuredatabricks.net"),
-        ("https://adb-123456.azuredatabricks.net", "https://adb-123456.azuredatabricks.net"),
-        ("https://adb-123456.azuredatabricks.net/", "https://adb-123456.azuredatabricks.net"),
-        ("workspace.databricks.com/", "https://workspace.databricks.com"),
+    
     ])
     def test_normalize_host_with_protocol(self, input_host, expected_output):
         """Test host normalization with various input formats."""