File tree Expand file tree Collapse file tree 2 files changed +1
-16
lines changed Expand file tree Collapse file tree 2 files changed +1
-16
lines changed Original file line number Diff line number Diff line change @@ -1078,7 +1078,7 @@ def cancel(self) -> None:
1078
1078
def close (self ) -> None :
1079
1079
"""Close cursor"""
1080
1080
self .open = False
1081
- self .active_op_handle = None
1081
+ self .active_command_id = None
1082
1082
if self .active_result_set :
1083
1083
self ._close_and_clear_active_result_set ()
1084
1084
Original file line number Diff line number Diff line change @@ -271,21 +271,6 @@ def test_context_manager_closes_cursor(self):
271
271
cursor .close = mock_close
272
272
mock_close .assert_called_once_with ()
273
273
274
- @patch ("%s.client.ThriftBackend" % PACKAGE_NAME )
275
- def test_context_manager_closes_connection (self , mock_client_class ):
276
- instance = mock_client_class .return_value
277
-
278
- mock_open_session_resp = MagicMock (spec = TOpenSessionResp )()
279
- mock_open_session_resp .sessionHandle .sessionId = b"\x22 "
280
- instance .open_session .return_value = mock_open_session_resp
281
-
282
- with databricks .sql .connect (** self .DUMMY_CONNECTION_ARGS ) as connection :
283
- pass
284
-
285
- # Check the close session request has an id of x22
286
- close_session_id = instance .close_session .call_args [0 ][0 ].sessionId
287
- self .assertEqual (close_session_id , b"\x22 " )
288
-
289
274
def dict_product (self , dicts ):
290
275
"""
291
276
Generate cartesion product of values in input dictionary, outputting a dictionary
You can’t perform that action at this time.
0 commit comments