@@ -40,36 +40,8 @@ def execute_response(self):
40
40
("test_value" , "INT" , None , None , None , None , None )
41
41
]
42
42
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
- }
48
43
return mock_response
49
44
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
-
73
45
def test_init_with_execute_response (
74
46
self , mock_connection , mock_sea_client , execute_response
75
47
):
@@ -225,4 +197,4 @@ def test_fill_results_buffer_not_implemented(
225
197
with pytest .raises (
226
198
NotImplementedError , match = "fetchone is not implemented for SEA backend"
227
199
):
228
- result_set ._fill_results_buffer ()
200
+ result_set ._fill_results_buffer ()
0 commit comments