unit test fix

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

tests/unit/test_fetches.py

@@ -1,12 +1,25 @@
 import unittest
 import pytest
-from unittest.mock import Mock
+from unittest.mock import Mock, patch
 
 try:
     import pyarrow as pa
 except ImportError:
     pa = None
 
+def noop_log_latency_decorator(*args, **kwargs):
+    """
+    This is a no-op decorator. It is used to patch the log_latency decorator
+    during tests, so that the tests for client logic are not affected by the
+    telemetry logging logic. It accepts any arguments and returns a decorator
+    that returns the original function unmodified.
+    """
+    def decorator(func):
+        return func
+    return decorator
+
+patch('databricks.sql.telemetry.latency_logger.log_latency', new=noop_log_latency_decorator).start()
+
 import databricks.sql.client as client
 from databricks.sql.utils import ExecuteResponse, ArrowQueue
 
@@ -31,15 +44,6 @@ def make_arrow_queue(batch):
         _, table = FetchTests.make_arrow_table(batch)
         queue = ArrowQueue(table, len(batch))
         return queue
-    
-    @classmethod
-    def mock_thrift_backend_with_retry_policy(cls): # Required for log_latency() decorator
-        """Create a simple thrift_backend mock with retry_policy for basic tests."""
-        mock_thrift_backend = Mock()
-        mock_retry_policy = Mock()
-        mock_retry_policy.history = []
-        mock_thrift_backend.retry_policy = mock_retry_policy
-        return mock_thrift_backend
 
     @staticmethod
     def make_dummy_result_set_from_initial_results(initial_results):
@@ -48,7 +52,7 @@ def make_dummy_result_set_from_initial_results(initial_results):
         arrow_queue = ArrowQueue(arrow_table, len(initial_results), 0)
         rs = client.ResultSet(
             connection=Mock(),
-            thrift_backend=FetchTests.mock_thrift_backend_with_retry_policy(),
+            thrift_backend=None,
             execute_response=ExecuteResponse(
                 status=None,
                 has_been_closed_server_side=True,
@@ -88,7 +92,7 @@ def fetch_results(
 
             return results, batch_index < len(batch_list)
 
-        mock_thrift_backend = FetchTests.mock_thrift_backend_with_retry_policy()
+        mock_thrift_backend = Mock()
         mock_thrift_backend.fetch_results = fetch_results
         num_cols = len(batch_list[0][0]) if batch_list and batch_list[0] else 0