Apply (#2134) to async code

Author: sfc-gh-pczajka

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

@@ -18,7 +18,7 @@
 class AuthByKeyPair(AuthByPluginAsync, AuthByKeyPairSync):
     def __init__(
         self,
-        private_key: bytes | RSAPrivateKey,
+        private_key: bytes | str | RSAPrivateKey,
         lifetime_in_seconds: int = AuthByKeyPairSync.LIFETIME,
         **kwargs,
     ) -> None:

test/integ/aio/test_key_pair_authentication_async.py

@@ -6,6 +6,7 @@
 from __future__ import annotations
 
 import asyncio
+import base64
 import uuid
 
 import pytest
@@ -81,6 +82,11 @@ def fin():
             async with snowflake.connector.aio.SnowflakeConnection(**db_config) as _:
                 pass
 
+            # Ensure the base64-encoded version also works
+            db_config["private_key"] = base64.b64encode(private_key_der)
+            async with snowflake.connector.aio.SnowflakeConnection(**db_config) as _:
+                pass
+
 
 @pytest.mark.skipolddriver
 async def test_multiple_key_pair(is_public_test, request, conn_cnx, db_parameters):

test/unit/aio/test_auth_keypair_async.py

@@ -101,7 +101,7 @@ async def test_auth_keypair_bad_type():
     class Bad:
         pass
 
-    for bad_private_key in ("abcd", 1234, Bad()):
+    for bad_private_key in (1234, Bad()):
         auth_instance = AuthByKeyPair(private_key=bad_private_key)
         with raises(TypeError) as ex:
             await auth_instance.prepare(account=account, user=user)