Skip to content

Commit 5b527d0

Browse files
fix unit tests
1 parent 027c68f commit 5b527d0

File tree

4 files changed

+9
-0
lines changed

4 files changed

+9
-0
lines changed

src/snowflake/connector/aio/_connection.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,9 @@ def __init__(
159159
self._stream_downloader = StreamDownloader(self)
160160
self._snowflake_version: str | None = None
161161

162+
# CRL is disabled for async connection
163+
self._crl_config = None # default value = disabled
164+
162165
@property
163166
async def snowflake_version(self) -> str:
164167
# The result from SELECT CURRENT_VERSION() is `<version> <internal hash>`,

src/snowflake/connector/aio/auth/_okta.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ async def _step1(
121121
conn._internal_application_name,
122122
conn._internal_application_version,
123123
conn._ocsp_mode(),
124+
conn.cert_revocation_check_mode,
124125
conn.login_timeout,
125126
conn.network_timeout,
126127
conn.socket_timeout,

src/snowflake/connector/aio/auth/_webbrowser.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,7 @@ async def _get_sso_url(
377377
conn._internal_application_name,
378378
conn._internal_application_version,
379379
conn._ocsp_mode(),
380+
conn.cert_revocation_check_mode,
380381
conn.login_timeout,
381382
conn.network_timeout,
382383
conn.socket_timeout,

test/unit/aio/mock_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ def mock_connection(
5252
backoff_policy=DEFAULT_BACKOFF_POLICY,
5353
disable_saml_url_check=False,
5454
session_manager=None,
55+
cert_revocation_check_mode="DISABLED",
56+
platform_detection_timeout_seconds=0.0,
5557
):
5658
return AsyncMock(
5759
_login_timeout=login_timeout,
@@ -66,4 +68,6 @@ def mock_connection(
6668
_disable_saml_url_check=disable_saml_url_check,
6769
_session_manager=session_manager or get_mock_session_manager(),
6870
_update_parameters=AsyncMock(return_value=None),
71+
cert_revocation_check_mode=cert_revocation_check_mode,
72+
platform_detection_timeout_seconds=platform_detection_timeout_seconds,
6973
)

0 commit comments

Comments
 (0)