Apply #2515 to async code

Author: sfc-gh-pczajka

test/integ/aio/test_connection_async.py

@@ -1105,11 +1105,11 @@ async def test_disable_query_context_cache(conn_cnx) -> None:
 
 
 @pytest.mark.skipolddriver
-@pytest.mark.parametrize(
-    "mode",
-    ("file", "env"),
-)
-async def test_connection_name_loading(monkeypatch, db_parameters, tmp_path, mode):
+@pytest.mark.parametrize("mode", ("file", "env"))
+@pytest.mark.parametrize("connection_name", ["default", "custom_connection_for_test"])
+async def test_connection_name_loading(
+    monkeypatch, db_parameters, tmp_path, mode, connection_name
+):
     import tomlkit
 
     doc = tomlkit.document()
@@ -1119,16 +1119,16 @@ async def test_connection_name_loading(monkeypatch, db_parameters, tmp_path, mod
         # If anything unexpected fails here, don't want to expose password
         for k, v in db_parameters.items():
             default_con[k] = v
-        doc["default"] = default_con
+        doc[connection_name] = default_con
         with monkeypatch.context() as m:
             if mode == "env":
-                m.setenv("SF_CONNECTIONS", tomlkit.dumps(doc))
+                m.setenv("SNOWFLAKE_CONNECTIONS", tomlkit.dumps(doc))
             else:
                 tmp_connections_file = tmp_path / "connections.toml"
                 tmp_connections_file.write_text(tomlkit.dumps(doc))
                 tmp_connections_file.chmod(stat.S_IRUSR | stat.S_IWUSR)
             async with snowflake.connector.aio.SnowflakeConnection(
-                connection_name="default",
+                connection_name=connection_name,
                 connections_file_path=tmp_connections_file,
             ) as conn:
                 async with conn.cursor() as cur:
@@ -1143,7 +1143,10 @@ async def test_connection_name_loading(monkeypatch, db_parameters, tmp_path, mod
 
 
 @pytest.mark.skipolddriver
-async def test_default_connection_name_loading(monkeypatch, db_parameters):
+@pytest.mark.parametrize("connection_name", ["default", "custom_connection_for_test"])
+async def test_default_connection_name_loading(
+    monkeypatch, db_parameters, connection_name
+):
     import tomlkit
 
     doc = tomlkit.document()
@@ -1152,10 +1155,10 @@ async def test_default_connection_name_loading(monkeypatch, db_parameters):
         # If anything unexpected fails here, don't want to expose password
         for k, v in db_parameters.items():
             default_con[k] = v
-        doc["default"] = default_con
+        doc[connection_name] = default_con
         with monkeypatch.context() as m:
             m.setenv("SNOWFLAKE_CONNECTIONS", tomlkit.dumps(doc))
-            m.setenv("SNOWFLAKE_DEFAULT_CONNECTION_NAME", "default")
+            m.setenv("SNOWFLAKE_DEFAULT_CONNECTION_NAME", connection_name)
             async with snowflake.connector.aio.SnowflakeConnection() as conn:
                 async with conn.cursor() as cur:
                     assert await (await cur.execute("select 1;")).fetchall() == [