@@ -88,45 +88,6 @@ def test_init_with_execute_response(
8888 assert isinstance (result_set .results , JsonQueue )
8989 assert result_set .results .data_array == []
9090
91- def test_init_with_result_data (
92- self ,
93- mock_connection ,
94- mock_sea_client ,
95- execute_response ,
96- mock_result_data ,
97- mock_manifest ,
98- ):
99- """Test initializing SeaResultSet with result data."""
100- with patch (
101- "databricks.sql.result_set.SeaResultSetQueueFactory"
102- ) as mock_factory :
103- mock_queue = Mock (spec = JsonQueue )
104- mock_factory .build_queue .return_value = mock_queue
105-
106- result_set = SeaResultSet (
107- connection = mock_connection ,
108- execute_response = execute_response ,
109- sea_client = mock_sea_client ,
110- buffer_size_bytes = 1000 ,
111- arraysize = 100 ,
112- result_data = mock_result_data ,
113- manifest = mock_manifest ,
114- )
115-
116- # Verify that the factory was called with the correct arguments
117- mock_factory .build_queue .assert_called_once_with (
118- mock_result_data ,
119- mock_manifest ,
120- str (execute_response .command_id .to_sea_statement_id ()),
121- description = execute_response .description ,
122- max_download_threads = mock_sea_client .max_download_threads ,
123- sea_client = mock_sea_client ,
124- lz4_compressed = execute_response .lz4_compressed ,
125- )
126-
127- # Verify that the queue was set correctly
128- assert result_set .results == mock_queue
129-
13091 def test_close (self , mock_connection , mock_sea_client , execute_response ):
13192 """Test closing a result set."""
13293 result_set = SeaResultSet (
0 commit comments