Use service account for CI tests

aazuspan · Jan 29, 2025 · 4df17ff · 4df17ff
1 parent 3215f13
commit 4df17ff
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,4 +1,5 @@
 import json
+import os
 import warnings
 from collections import UserDict
 from pathlib import Path
@@ -65,7 +66,13 @@ def object_cache():
 
 
 def pytest_sessionstart(session):
-    ee.Initialize()
+    if os.environ.get("GITHUB_ACTIONS"):
+        service_account = os.environ.get("EE_SERVICE_ACCOUNT")
+        credentials = ee.ServiceAccountCredentials(None, key_data=service_account)
+    else:
+        credentials = "persistent"
+
+    ee.Initialize(credentials=credentials)
 
 
 @pytest.fixture(autouse=True)