Fix telemetry client lookup to use test-host in tests

Author: samikshya-db

tests/unit/test_telemetry.py

@@ -418,7 +418,7 @@ def test_telemetry_enabled_when_flag_is_true(self, mock_http_request, MockSessio
 
         assert conn.telemetry_enabled is True
         mock_http_request.assert_called_once()
-        client = TelemetryClientFactory.get_telemetry_client("test")
+        client = TelemetryClientFactory.get_telemetry_client("test-host")
         assert isinstance(client, TelemetryClient)
 
     @patch("databricks.sql.common.unified_http_client.UnifiedHttpClient.request")
@@ -449,7 +449,7 @@ def test_telemetry_disabled_when_flag_is_false(
 
         assert conn.telemetry_enabled is False
         mock_http_request.assert_called_once()
-        client = TelemetryClientFactory.get_telemetry_client("test")
+        client = TelemetryClientFactory.get_telemetry_client("test-host")
         assert isinstance(client, NoopTelemetryClient)
 
     @patch("databricks.sql.common.unified_http_client.UnifiedHttpClient.request")
@@ -480,7 +480,7 @@ def test_telemetry_disabled_when_flag_request_fails(
 
         assert conn.telemetry_enabled is False
         mock_http_request.assert_called_once()
-        client = TelemetryClientFactory.get_telemetry_client("test")
+        client = TelemetryClientFactory.get_telemetry_client("test-host")
         assert isinstance(client, NoopTelemetryClient)