removed function in test_fetches cuz it is only used once

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

Author: saishreeeee

src/databricks/sql/telemetry/latency_logger.py

@@ -145,7 +145,7 @@ def get_extractor(obj):
         TelemetryExtractor: A specialized extractor instance:
             - CursorExtractor for Cursor objects
             - ResultSetExtractor for ResultSet objects
-            - TelemetryExtractor (base) for all other objects
+            - Throws an NotImplementedError for all other objects
     """
     if obj.__class__.__name__ == "Cursor":
         return CursorExtractor(obj)

tests/unit/test_fetches.py

@@ -59,14 +59,6 @@ def make_dummy_result_set_from_initial_results(initial_results):
         ]
         return rs
 
-    @staticmethod
-    def mock_thrift_backend_with_retry_policy():
-        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_batch_list(batch_list):
         batch_index = 0
@@ -87,7 +79,10 @@ 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_retry_policy = Mock()
+        mock_retry_policy.history = []
+        mock_thrift_backend.retry_policy = mock_retry_policy
         mock_thrift_backend.fetch_results = fetch_results
         num_cols = len(batch_list[0][0]) if batch_list and batch_list[0] else 0