Apply #2169 changes to async code

Author: sfc-gh-pczajka

src/snowflake/connector/aio/_connection.py

@@ -170,7 +170,7 @@ async def __open_connection(self):
                 os.environ["SF_OCSP_RESPONSE_CACHE_SERVER_URL"],
             )
 
-        if ".privatelink.snowflakecomputing." in self.host:
+        if ".privatelink.snowflakecomputing." in self.host.lower():
             await SnowflakeConnection.setup_ocsp_privatelink(
                 self.application, self.host
             )
@@ -969,6 +969,7 @@ async def get_query_status_throw_if_error(self, sf_qid: str) -> QueryStatus:
 
     @staticmethod
     async def setup_ocsp_privatelink(app, hostname) -> None:
+        hostname = hostname.lower()
         async with SnowflakeConnection.OCSP_ENV_LOCK:
             ocsp_cache_server = f"http://ocsp.{hostname}/ocsp_response_cache.json"
             os.environ["SF_OCSP_RESPONSE_CACHE_SERVER_URL"] = ocsp_cache_server

test/integ/aio/test_connection_async.py

@@ -783,6 +783,19 @@ async def test_privatelink_ocsp_url_concurrent_snowsql():
             raise AssertionError()
 
 
+@pytest.mark.skipolddriver
+async def test_uppercase_privatelink_ocsp_url_creation():
+    account = "TESTACCOUNT.US-EAST-1.PRIVATELINK"
+    hostname = account + ".snowflakecomputing.com"
+
+    await SnowflakeConnection.setup_ocsp_privatelink(CLIENT_NAME, hostname)
+    ocsp_cache_server = os.getenv("SF_OCSP_RESPONSE_CACHE_SERVER_URL", None)
+    assert (
+        ocsp_cache_server
+        == "http://ocsp.testaccount.us-east-1.privatelink.snowflakecomputing.com/ocsp_response_cache.json"
+    )
+
+
 class ExecPrivatelinkAsyncTask:
     def __init__(self, bucket, hostname, expectation, client_name):
         self.bucket = bucket