File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -824,7 +824,7 @@ def test_close_connection_closes_cursors(self):
824824 status_request = ttypes .TGetOperationStatusReq (
825825 operationHandle = ars .command_id , getProgressUpdate = False
826826 )
827- op_status_at_server = ars .thrift_backend ._client .GetOperationStatus (
827+ op_status_at_server = ars .backend ._client .GetOperationStatus (
828828 status_request
829829 )
830830 assert (
@@ -836,7 +836,7 @@ def test_close_connection_closes_cursors(self):
836836
837837 # When connection closes, any cursor operations should no longer exist at the server
838838 with pytest .raises (SessionAlreadyClosedError ) as cm :
839- op_status_at_server = ars .thrift_backend ._client .GetOperationStatus (
839+ op_status_at_server = ars .backend ._client .GetOperationStatus (
840840 status_request
841841 )
842842
@@ -916,7 +916,7 @@ def test_cursor_error_handling(self):
916916 assert op_handle is not None
917917
918918 # Manually close the operation to simulate server-side closure
919- conn .thrift_backend .close_command (op_handle )
919+ conn .session . backend .close_command (op_handle )
920920
921921 cursor .close ()
922922
@@ -936,7 +936,7 @@ def test_result_set_close(self):
936936
937937 result_set .close ()
938938
939- assert result_set .op_state == result_set .thrift_backend .CLOSED_OP_STATE
939+ assert result_set .op_state == result_set .backend .CLOSED_OP_STATE
940940 assert result_set .op_state != initial_op_state
941941
942942 # Closing the result set again should be a no-op and not raise exceptions
You can’t perform that action at this time.
0 commit comments