remove redundant tests

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

tests/unit/test_sea_backend.py

@@ -890,76 +890,3 @@ def test_get_columns(self, sea_client, sea_session_id, mock_cursor):
                     cursor=mock_cursor,
                 )
             assert "Catalog name is required for get_columns" in str(excinfo.value)
-
-    def test_get_chunk_link(self, sea_client, mock_http_client, sea_command_id):
-        """Test get_chunk_link method."""
-        # Setup mock response
-        mock_response = {
-            "external_links": [
-                {
-                    "external_link": "https://example.com/data/chunk0",
-                    "expiration": "2025-07-03T05:51:18.118009",
-                    "row_count": 100,
-                    "byte_count": 1024,
-                    "row_offset": 0,
-                    "chunk_index": 0,
-                    "next_chunk_index": 1,
-                    "http_headers": {"Authorization": "Bearer token123"},
-                }
-            ]
-        }
-        mock_http_client._make_request.return_value = mock_response
-
-        # Call the method
-        result = sea_client.get_chunk_link("test-statement-123", 0)
-
-        # Verify the HTTP client was called correctly
-        mock_http_client._make_request.assert_called_once_with(
-            method="GET",
-            path=sea_client.CHUNK_PATH_WITH_ID_AND_INDEX.format(
-                "test-statement-123", 0
-            ),
-        )
-
-        # Verify the result
-        assert result.external_link == "https://example.com/data/chunk0"
-        assert result.expiration == "2025-07-03T05:51:18.118009"
-        assert result.row_count == 100
-        assert result.byte_count == 1024
-        assert result.row_offset == 0
-        assert result.chunk_index == 0
-        assert result.next_chunk_index == 1
-        assert result.http_headers == {"Authorization": "Bearer token123"}
-
-    def test_get_chunk_link_not_found(self, sea_client, mock_http_client):
-        """Test get_chunk_link when the requested chunk is not found."""
-        # Setup mock response with no matching chunk
-        mock_response = {
-            "external_links": [
-                {
-                    "external_link": "https://example.com/data/chunk1",
-                    "expiration": "2025-07-03T05:51:18.118009",
-                    "row_count": 100,
-                    "byte_count": 1024,
-                    "row_offset": 100,
-                    "chunk_index": 1,  # Different chunk index
-                    "next_chunk_index": 2,
-                    "http_headers": {"Authorization": "Bearer token123"},
-                }
-            ]
-        }
-        mock_http_client._make_request.return_value = mock_response
-
-        # Call the method and expect an exception
-        with pytest.raises(
-            ServerOperationError, match="No link found for chunk index 0"
-        ):
-            sea_client.get_chunk_link("test-statement-123", 0)
-
-        # Verify the HTTP client was called correctly
-        mock_http_client._make_request.assert_called_once_with(
-            method="GET",
-            path=sea_client.CHUNK_PATH_WITH_ID_AND_INDEX.format(
-                "test-statement-123", 0
-            ),
-        )

tests/unit/test_sea_queue.py

@@ -490,86 +490,6 @@ def test_progress_chunk_link_no_next_chunk(self, mock_logger):
         assert result is None
         assert queue._current_chunk_link is None
 
-    @patch("databricks.sql.backend.sea.queue.logger")
-    def test_progress_chunk_link_success(self, mock_logger, mock_sea_client):
-        """Test _progress_chunk_link with successful progression."""
-        # Create a queue instance without initializing
-        queue = Mock(spec=SeaCloudFetchQueue)
-        queue._current_chunk_link = ExternalLink(
-            external_link="https://example.com/data/chunk0",
-            expiration="2025-07-03T05:51:18.118009",
-            row_count=100,
-            byte_count=1024,
-            row_offset=0,
-            chunk_index=0,
-            next_chunk_index=1,
-            http_headers={"Authorization": "Bearer token123"},
-        )
-        queue._sea_client = mock_sea_client
-        queue._statement_id = "test-statement-123"
-
-        # Setup the mock client to return a new link
-        next_link = ExternalLink(
-            external_link="https://example.com/data/chunk1",
-            expiration="2025-07-03T05:51:18.235843",
-            row_count=50,
-            byte_count=512,
-            row_offset=100,
-            chunk_index=1,
-            next_chunk_index=None,
-            http_headers={"Authorization": "Bearer token123"},
-        )
-        mock_sea_client.get_chunk_link.return_value = next_link
-
-        # Call the method directly
-        SeaCloudFetchQueue._progress_chunk_link(queue)
-
-        # Verify the client was called
-        mock_sea_client.get_chunk_link.assert_called_once_with("test-statement-123", 1)
-
-        # Verify debug message was logged
-        mock_logger.debug.assert_called_with(
-            f"SeaCloudFetchQueue: Progressed to link for chunk 1: {next_link}"
-        )
-
-    @patch("databricks.sql.backend.sea.queue.logger")
-    def test_progress_chunk_link_error(self, mock_logger, mock_sea_client):
-        """Test _progress_chunk_link with error during chunk fetch."""
-        # Create a queue instance without initializing
-        queue = Mock(spec=SeaCloudFetchQueue)
-        queue._current_chunk_link = ExternalLink(
-            external_link="https://example.com/data/chunk0",
-            expiration="2025-07-03T05:51:18.118009",
-            row_count=100,
-            byte_count=1024,
-            row_offset=0,
-            chunk_index=0,
-            next_chunk_index=1,
-            http_headers={"Authorization": "Bearer token123"},
-        )
-        queue._sea_client = mock_sea_client
-        queue._statement_id = "test-statement-123"
-
-        # Setup the mock client to raise an error
-        error_message = "Network error"
-        mock_sea_client.get_chunk_link.side_effect = Exception(error_message)
-
-        # Call the method directly
-        result = SeaCloudFetchQueue._progress_chunk_link(queue)
-
-        # Verify the client was called
-        mock_sea_client.get_chunk_link.assert_called_once_with("test-statement-123", 1)
-
-        # Verify error message was logged
-        mock_logger.error.assert_called_with(
-            "SeaCloudFetchQueue: Error fetching link for chunk {}: {}".format(
-                1, error_message
-            )
-        )
-
-        # Verify the result is None
-        assert result is None
-
     @patch("databricks.sql.backend.sea.queue.logger")
     def test_create_next_table_no_current_link(self, mock_logger):
         """Test _create_next_table with no current link."""