Make get_access_token sync compatible

CHANGELOG.md

@@ -9,10 +9,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
-- `VertexAICustomTrainingJob` infrastructure block - [#75](https://github.com/PrefectHQ/prefect-gcp/pull/75)
-
 ### Changed
 
+- Made `GcpCredentials.get_access_token` sync compatible - [#80](https://github.com/PrefectHQ/prefect-gcp/pull/80)
+
 ### Deprecated
 
 ### Removed

prefect_gcp/credentials.py

@@ -30,7 +30,7 @@
     pass
 
 from prefect.blocks.core import Block
-from prefect.utilities.asyncutils import run_sync_in_worker_thread
+from prefect.utilities.asyncutils import run_sync_in_worker_thread, sync_compatible
 
 
 def _raise_help_msg(key: str):
@@ -153,6 +153,7 @@ def get_credentials_from_service_account(self) -> Credentials:
             credentials, _ = google.auth.default()
         return credentials
 
+    @sync_compatible
     async def get_access_token(self):
         """
         See: https://stackoverflow.com/a/69107745

tests/conftest.py

@@ -167,10 +167,19 @@ def destroy_secret_version(self, name, **kwds):
 @pytest.fixture
 def mock_credentials(monkeypatch):
     mock_credentials = MagicMock(name="MockGoogleCredentials")
+    mock_authenticated_credentials = MagicMock(token="my-token")
+    mock_credentials.from_service_account_info = mock_authenticated_credentials
+    mock_credentials.from_service_account_file = mock_authenticated_credentials
     monkeypatch.setattr(
-        "prefect_gcp.cloud_run.GcpCredentials.get_credentials_from_service_account",  # noqa
+        "prefect_gcp.credentials.Credentials",  # noqa
         mock_credentials,
     )
+    mock_auth = MagicMock()
+    mock_auth.default.return_value = (mock_authenticated_credentials, "project")
+    monkeypatch.setattr(
+        "prefect_gcp.credentials.google.auth",  # noqa
+        mock_auth,
+    )
     return mock_credentials
 
 

tests/test_credentials.py

@@ -196,3 +196,14 @@ def test_flow():
         }
     }
     assert test_flow() == expected
+
+
+async def test_get_access_token_async(gcp_credentials):
+    print(gcp_credentials.get_credentials_from_service_account().token)
+    token = await gcp_credentials.get_access_token()
+    assert token == "my-token"
+
+
+def test_get_access_token_sync_compatible(gcp_credentials):
+    token = gcp_credentials.get_access_token()
+    assert token == "my-token"