@@ -292,26 +292,6 @@ def test_command_execution_async(
292292 assert isinstance (mock_cursor .active_command_id , CommandId )
293293 assert mock_cursor .active_command_id .guid == "test-statement-456"
294294
295- # Test async with missing statement ID
296- mock_http_client .reset_mock ()
297- mock_http_client ._make_request .return_value = {"status" : {"state" : "PENDING" }}
298- with pytest .raises (ServerOperationError ) as excinfo :
299- sea_client .execute_command (
300- operation = "SELECT 1" ,
301- session_id = sea_session_id ,
302- max_rows = 100 ,
303- max_bytes = 1000 ,
304- lz4_compression = False ,
305- cursor = mock_cursor ,
306- use_cloud_fetch = False ,
307- parameters = [],
308- async_op = True ,
309- enforce_embedded_schema_correctness = False ,
310- )
311- assert "Failed to execute command: No statement ID returned" in str (
312- excinfo .value
313- )
314-
315295 def test_command_execution_advanced (
316296 self , sea_client , mock_http_client , mock_cursor , sea_session_id
317297 ):
@@ -411,26 +391,6 @@ def test_command_execution_advanced(
411391 )
412392 assert "Command failed" in str (excinfo .value )
413393
414- # Test missing statement ID
415- mock_http_client .reset_mock ()
416- mock_http_client ._make_request .return_value = {"status" : {"state" : "SUCCEEDED" }}
417- with pytest .raises (ServerOperationError ) as excinfo :
418- sea_client .execute_command (
419- operation = "SELECT 1" ,
420- session_id = sea_session_id ,
421- max_rows = 100 ,
422- max_bytes = 1000 ,
423- lz4_compression = False ,
424- cursor = mock_cursor ,
425- use_cloud_fetch = False ,
426- parameters = [],
427- async_op = False ,
428- enforce_embedded_schema_correctness = False ,
429- )
430- assert "Failed to execute command: No statement ID returned" in str (
431- excinfo .value
432- )
433-
434394 def test_command_management (
435395 self ,
436396 sea_client ,
0 commit comments