Skip to content

Commit 93516a3

Browse files
SNOW-2203079: Allow sending through mock adapter for SSL tests
1 parent 6c66cff commit 93516a3

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

test/unit/test_retry_network.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,12 @@
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
4651
try:
@@ -382,7 +387,9 @@ def fake_request_exec(**kwargs):
382387

383388

384389
def 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(

0 commit comments

Comments
 (0)