@@ -40,36 +40,8 @@ def execute_response(self):
4040 ("test_value" , "INT" , None , None , None , None , None )
4141 ]
4242 mock_response .is_staging_operation = False
43- mock_response .sea_response = {
44- "statement_id" : "test-statement-123" ,
45- "status" : {"state" : "SUCCEEDED" },
46- "result" : {"data_array" : [["1" ]]},
47- }
4843 return mock_response
4944
50- def test_init_with_sea_response (
51- self , mock_connection , mock_sea_client , sea_response
52- ):
53- """Test initializing SeaResultSet with a SEA response."""
54- result_set = SeaResultSet (
55- connection = mock_connection ,
56- sea_client = mock_sea_client ,
57- sea_response = sea_response ,
58- buffer_size_bytes = 1000 ,
59- arraysize = 100 ,
60- )
61-
62- # Verify basic properties
63- assert result_set .statement_id == "test-statement-123"
64- assert result_set .status == CommandState .SUCCEEDED
65- assert result_set .command_id .guid == "test-statement-123"
66- assert result_set .command_id .backend_type == BackendType .SEA
67- assert result_set .connection == mock_connection
68- assert result_set .backend == mock_sea_client
69- assert result_set .buffer_size_bytes == 1000
70- assert result_set .arraysize == 100
71- assert result_set ._response == sea_response
72-
7345 def test_init_with_execute_response (
7446 self , mock_connection , mock_sea_client , execute_response
7547 ):
@@ -225,4 +197,4 @@ def test_fill_results_buffer_not_implemented(
225197 with pytest .raises (
226198 NotImplementedError , match = "fetchone is not implemented for SEA backend"
227199 ):
228- result_set ._fill_results_buffer ()
200+ result_set ._fill_results_buffer ()
0 commit comments