Skip to content

Commit 7cece5e

Browse files
remove unnecessary init with sea_response tests
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 4d0aeca commit 7cece5e

File tree

1 file changed

+1
-29
lines changed

1 file changed

+1
-29
lines changed

tests/unit/test_sea_result_set.py

Lines changed: 1 addition & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)