Fix telemetry test mocks to set session.use_kernel

The connection-failure telemetry suppression reads session.use_kernel, but
these tests mock Session, so use_kernel was a truthy MagicMock — which
suppressed the wrapper failure log and broke
test_connection_failure_sends_correct_telemetry_payload. Set use_kernel
explicitly on the mock (False for the default Thrift case, True for the kernel
case) to reflect production.

Co-authored-by: Isaac <no-reply@databricks.com>
Signed-off-by: Rahul Singhal <rahul.singhal@databricks.com>

Author: rahuls-db

tests/unit/test_telemetry.py

@@ -482,6 +482,9 @@ def test_connection_failure_sends_correct_telemetry_payload(
         # Set up the mock to create a session instance first, then make open() fail
         mock_session_instance = MagicMock()
         mock_session_instance.is_open = False  # Ensure cleanup is safe
+        # Default (Thrift) session: the failure-telemetry suppression reads
+        # session.use_kernel, so the mock must expose a real bool, not a truthy Mock.
+        mock_session_instance.use_kernel = False
         mock_session_instance.open.side_effect = Exception(error_message)
         mock_session.return_value = mock_session_instance
 
@@ -511,6 +514,8 @@ def test_connection_failure_does_not_send_telemetry_for_kernel(
         error_message = "Could not connect to host"
         mock_session_instance = MagicMock()
         mock_session_instance.is_open = False
+        # Kernel session: use_kernel is read to suppress the wrapper-side failure log.
+        mock_session_instance.use_kernel = True
         mock_session_instance.open.side_effect = Exception(error_message)
         mock_session.return_value = mock_session_instance