NO-SNOW: Fix tests

Author: sfc-gh-turbaszek

setup.cfg

@@ -96,6 +96,7 @@ development =
     pytest-xdist
     pytzdata
     responses
+    pytest-asyncio
 pandas =
     pandas>=2.1.2,<3.0.0
     pyarrow

src/snowflake/connector/aio/_connection.py

@@ -1019,6 +1019,7 @@ async def connect(self, **kwargs) -> None:
         else:
             self.__config(**self._conn_parameters)
 
+        self._crl_config: CRLConfig = CRLConfig.from_connection(self)
         self._http_config: AioHttpConfig = AioHttpConfig(
             connector_factory=SnowflakeSSLConnectorFactory(),
             use_pooling=not self.disable_request_pooling,

src/snowflake/connector/aio/_session_manager.py

@@ -86,7 +86,7 @@ async def connect(
         )
         if (
             FEATURE_CRL_CONFIG.cert_revocation_check_mode
-            != CertRevocationCheckMode.ENABLED
+            != CertRevocationCheckMode.DISABLED
         ):
             self.validate_crl(protocol, req)
             logger.debug(

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

@@ -99,6 +99,7 @@ async def authenticate(
             self._rest._connection._internal_application_name,
             self._rest._connection._internal_application_version,
             self._rest._connection._ocsp_mode(),
+            self._rest._connection.cert_revocation_check_mode,
             self._rest._connection._login_timeout,
             self._rest._connection._network_timeout,
             self._rest._connection._socket_timeout,

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

@@ -121,6 +121,7 @@ async def _step1(
             conn._internal_application_name,
             conn._internal_application_version,
             conn._ocsp_mode(),
+            conn.cert_revocation_check_mode,
             conn.login_timeout,
             conn.network_timeout,
             conn.socket_timeout,

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

@@ -377,6 +377,7 @@ async def _get_sso_url(
             conn._internal_application_name,
             conn._internal_application_version,
             conn._ocsp_mode(),
+            conn.cert_revocation_check_mode,
             conn.login_timeout,
             conn.network_timeout,
             conn.socket_timeout,

test/unit/aio/test_auth_async.py

@@ -24,6 +24,7 @@ def _init_rest(application, post_requset):
     connection = mock_connection()
     connection.errorhandler = Mock(return_value=None)
     connection._ocsp_mode = Mock(return_value=OCSPMode.FAIL_OPEN)
+    connection.cert_revocation_check_mode = "TEST_CRL_MODE"
     type(connection).application = PropertyMock(return_value=application)
     type(connection)._internal_application_name = PropertyMock(return_value=CLIENT_NAME)
     type(connection)._internal_application_version = PropertyMock(

test/unit/aio/test_auth_keypair_async.py

@@ -146,6 +146,7 @@ def _init_rest(application, post_requset):
     connection = mock_connection()
     connection.errorhandler = Mock(return_value=None)
     connection._ocsp_mode = Mock(return_value=OCSPMode.FAIL_OPEN)
+    connection.cert_revocation_check_mode = "TEST_CRL_MODE"
     type(connection).application = PropertyMock(return_value=application)
     type(connection)._internal_application_name = PropertyMock(return_value=CLIENT_NAME)
     type(connection)._internal_application_version = PropertyMock(

test/unit/aio/test_auth_okta_async.py

@@ -342,6 +342,7 @@ async def post_request(url, headers, body, **kwargs):
     connection = mock_connection(disable_saml_url_check=disable_saml_url_check)
     connection.errorhandler = Mock(return_value=None)
     connection._ocsp_mode = Mock(return_value=OCSPMode.FAIL_OPEN)
+    connection.cert_revocation_check_mode = "TEST_CRL_MODE"
     type(connection).application = PropertyMock(return_value=CLIENT_NAME)
     type(connection)._internal_application_name = PropertyMock(return_value=CLIENT_NAME)
     type(connection)._internal_application_version = PropertyMock(

test/unit/aio/test_auth_webbrowser_async.py

@@ -413,6 +413,7 @@ async def post_request(url, headers, body, **kwargs):
     connection = mock_connection(socket_timeout=socket_timeout)
     connection.errorhandler = Mock(return_value=None)
     connection._ocsp_mode = Mock(return_value=OCSPMode.FAIL_OPEN)
+    connection.cert_revocation_check_mode = "TEST_CRL_MODE"
     connection._disable_console_login = disable_console_login
     type(connection).application = PropertyMock(return_value=CLIENT_NAME)
     type(connection)._internal_application_name = PropertyMock(return_value=CLIENT_NAME)