fix async tests

Author: sfc-gh-pczajka

test/integ/aio/test_connection_async.py

@@ -1450,20 +1450,11 @@ async def test_server_session_keep_alive(conn_cnx):
 
 
 @pytest.mark.skipolddriver
-async def test_ocsp_mode_disable_ocsp_checks(conn_cnx, is_public_test, caplog):
+async def test_ocsp_mode_disable_ocsp_checks(conn_cnx, caplog):
     caplog.set_level(logging.DEBUG, "snowflake.connector.aio._ocsp_snowflake")
     async with conn_cnx(disable_ocsp_checks=True) as conn, conn.cursor() as cur:
         assert await (await cur.execute("select 1")).fetchall() == [(1,)]
-        assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
-        caplog.clear()
-
-    async with conn_cnx() as conn, conn.cursor() as cur:
-        assert await (await cur.execute("select 1")).fetchall() == [(1,)]
-        if is_public_test:
-            assert "snowflake.connector.aio._ocsp_snowflake" in caplog.text
-            assert "This connection does not perform OCSP checks." not in caplog.text
-        else:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+        assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
 
 
 @pytest.mark.skipolddriver
@@ -1472,10 +1463,10 @@ async def test_ocsp_mode_insecure_mode(conn_cnx, is_public_test, caplog):
     async with conn_cnx(insecure_mode=True) as conn, conn.cursor() as cur:
         assert await (await cur.execute("select 1")).fetchall() == [(1,)]
         if is_public_test:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
             assert "This connection does not perform OCSP checks." in caplog.text
         else:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
 
 
 @pytest.mark.skipolddriver
@@ -1488,14 +1479,14 @@ async def test_ocsp_mode_insecure_mode_and_disable_ocsp_checks_match(
     ) as conn, conn.cursor() as cur:
         assert await (await cur.execute("select 1")).fetchall() == [(1,)]
         if is_public_test:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
             assert (
                 "The values for 'disable_ocsp_checks' and 'insecure_mode' differ. "
                 "Using the value of 'disable_ocsp_checks."
             ) not in caplog.text
             assert "This connection does not perform OCSP checks." in caplog.text
         else:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
 
 
 @pytest.mark.skipolddriver
@@ -1508,34 +1499,30 @@ async def test_ocsp_mode_insecure_mode_and_disable_ocsp_checks_mismatch_ocsp_dis
     ) as conn, conn.cursor() as cur:
         assert await (await cur.execute("select 1")).fetchall() == [(1,)]
         if is_public_test:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
             assert (
                 "The values for 'disable_ocsp_checks' and 'insecure_mode' differ. "
                 "Using the value of 'disable_ocsp_checks."
             ) in caplog.text
             assert "This connection does not perform OCSP checks." in caplog.text
         else:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+            assert "snowflake.connector.aio._ocsp_snowflake:" not in caplog.text
 
 
 @pytest.mark.skipolddriver
 async def test_ocsp_mode_insecure_mode_and_disable_ocsp_checks_mismatch_ocsp_enabled(
-    conn_cnx, is_public_test, caplog
+    conn_cnx, caplog
 ):
     caplog.set_level(logging.DEBUG, "snowflake.connector.aio._ocsp_snowflake")
     async with conn_cnx(
         insecure_mode=True, disable_ocsp_checks=False
     ) as conn, conn.cursor() as cur:
         assert await (await cur.execute("select 1")).fetchall() == [(1,)]
-        if is_public_test:
-            assert "snowflake.connector.aio._ocsp_snowflake" in caplog.text
-            assert (
-                "The values for 'disable_ocsp_checks' and 'insecure_mode' differ. "
-                "Using the value of 'disable_ocsp_checks."
-            ) in caplog.text
-            assert "This connection does not perform OCSP checks." not in caplog.text
-        else:
-            assert "snowflake.connector.aio._ocsp_snowflake" not in caplog.text
+        assert (
+            "The values for 'disable_ocsp_checks' and 'insecure_mode' differ. "
+            "Using the value of 'disable_ocsp_checks."
+        ) in caplog.text
+        assert "snowflake.connector.aio._ocsp_snowflake:" in caplog.text
 
 
 @pytest.mark.skipolddriver