adding await for generate_key()

I have added the await where ever the key generation is happening, but after running the unit tests, the code is failing. I am pushing the changes although, will keep on testing until the tests are passed.

google/cloud/alloydb/connector/instance.py

@@ -98,7 +98,7 @@ async def _perform_refresh(self) -> RefreshResult:
 
         try:
             await self._refresh_rate_limiter.acquire()
-            priv_key, pub_key = self._keys
+            priv_key, pub_key = await self._keys
             # fetch metadata
             metadata_task = asyncio.create_task(
                 self._client._get_metadata(

tests/unit/test_client.py

@@ -77,7 +77,7 @@ async def test__get_client_certificate(
     Test _get_client_certificate returns successfully.
     """
     test_client = AlloyDBClient("", "", credentials, client)
-    keys = generate_keys()
+    keys = await generate_keys()
     certs = await test_client._get_client_certificate(
         "test-project", "test-region", "test-cluster", keys[1]
     )

tests/unit/test_instance.py

@@ -31,7 +31,7 @@ async def test_Instance_init() -> None:
     Test to check whether the __init__ method of Instance
     can tell if the instance URI that's passed in is formatted correctly.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     async with aiohttp.ClientSession() as client:
         instance = Instance(
             "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",
@@ -52,7 +52,7 @@ async def test_Instance_init_invalid_instant_uri() -> None:
     Test to check whether the __init__ method of Instance
     will throw error for invalid instance URI.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     async with aiohttp.ClientSession() as client:
         with pytest.raises(ValueError):
             Instance("invalid/instance/uri/", client, keys)
@@ -64,7 +64,7 @@ async def test_Instance_close() -> None:
     Test that Instance's close method
     cancels tasks gracefully.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     instance = Instance(
         "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",
@@ -84,7 +84,7 @@ async def test_Instance_close() -> None:
 @pytest.mark.asyncio
 async def test_perform_refresh() -> None:
     """Test that _perform refresh returns valid RefreshResult"""
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     instance = Instance(
         "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",
@@ -104,7 +104,7 @@ async def test_schedule_refresh_replaces_result() -> None:
     Test to check whether _schedule_refresh replaces a valid refresh result
     with another refresh result.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     instance = Instance(
         "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",
@@ -131,7 +131,7 @@ async def test_schedule_refresh_wont_replace_valid_result_with_invalid() -> None
     Test to check whether _schedule_refresh won't replace a valid
     refresh result with an invalid one.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     instance = Instance(
         "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",
@@ -160,7 +160,7 @@ async def test_schedule_refresh_expired_cert() -> None:
     Test to check whether _schedule_refresh will throw RefreshError on
     expired certificate.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     # set certificate to be expired
     client.instance.cert_before = datetime.now() - timedelta(minutes=20)
@@ -182,7 +182,7 @@ async def test_force_refresh_cancels_pending_refresh() -> None:
     """
     Test that force_refresh cancels pending task if refresh_in_progress event is not set.
     """
-    keys = generate_keys()
+    keys = await generate_keys()
     client = FakeAlloyDBClient()
     instance = Instance(
         "projects/test-project/locations/test-region/clusters/test-cluster/instances/test-instance",