try modified privatelink

Author: sfc-gh-pcyrek

test/integ/test_connection.py

@@ -455,7 +455,7 @@ def test_us_west_connection(tmpdir, conn_cnx):
 
 
 @pytest.mark.timeout(60)
-def test_privatelink(db_parameters):
+def test_privatelink(conn_cnx):
     """Ensure the OCSP cache server URL is overridden if privatelink connection is used."""
     try:
         os.environ["SF_OCSP_FAIL_OPEN"] = "false"
@@ -477,20 +477,13 @@ def test_privatelink(db_parameters):
             "ocsp_response_cache.json"
         )
 
-    cnx = snowflake.connector.connect(
-        user=db_parameters["user"],
-        password=db_parameters["password"],
-        host=db_parameters["host"],
-        port=db_parameters["port"],
-        account=db_parameters["account"],
-        database=db_parameters["database"],
-        protocol=db_parameters["protocol"],
-        timezone="UTC",
-    )
-    assert cnx, "invalid cnx"
+    # Test that normal connections don't set the privatelink OCSP URL
+    with conn_cnx(timezone="UTC") as cnx:
+        assert cnx, "invalid cnx"
 
-    ocsp_url = os.getenv("SF_OCSP_RESPONSE_CACHE_SERVER_URL")
-    assert ocsp_url is None, f"OCSP URL should be None: {ocsp_url}"
+        ocsp_url = os.getenv("SF_OCSP_RESPONSE_CACHE_SERVER_URL")
+        assert ocsp_url is None, f"OCSP URL should be None: {ocsp_url}"
+    
     del os.environ["SF_OCSP_DO_RETRY"]
     del os.environ["SF_OCSP_FAIL_OPEN"]