File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change 4040 SnowflakeRestful ,
4141)
4242
43- from .mock_utils import mock_connection , mock_request_with_action , zero_backoff
43+ from .mock_utils import (
44+ get_mock_session_manager ,
45+ mock_connection ,
46+ mock_request_with_action ,
47+ zero_backoff ,
48+ )
4449
4550# We need these for our OldDriver tests. We run most up to date tests with the oldest supported driver version
4651try :
@@ -382,7 +387,9 @@ def fake_request_exec(**kwargs):
382387
383388
384389def test_retry_connection_reset_error (caplog ):
385- connection = mock_connection ()
390+ connection = mock_connection (
391+ session_manager = get_mock_session_manager (allow_send = True )
392+ )
386393 connection .errorhandler = Mock (return_value = None )
387394
388395 rest = SnowflakeRestful (
You can’t perform that action at this time.
0 commit comments