tests: allow running systests on emulator (#168)

Co-authored-by: Tres Seaver <tseaver@palladion.com>

Author: HemangChothani

packages/google-cloud-firestore/tests/system/test__helpers.py

@@ -1,10 +1,14 @@
 import os
 import re
+from google.cloud.firestore_v1.base_client import _FIRESTORE_EMULATOR_HOST
 from test_utils.system import unique_resource_id
+from test_utils.system import EmulatorCreds
 
 FIRESTORE_CREDS = os.environ.get("FIRESTORE_APPLICATION_CREDENTIALS")
 FIRESTORE_PROJECT = os.environ.get("GCLOUD_PROJECT")
 RANDOM_ID_REGEX = re.compile("^[a-zA-Z0-9]{20}$")
 MISSING_DOCUMENT = "No document to update: "
 DOCUMENT_EXISTS = "Document already exists: "
 UNIQUE_RESOURCE_ID = unique_resource_id("-")
+EMULATOR_CREDS = EmulatorCreds()
+FIRESTORE_EMULATOR = os.environ.get(_FIRESTORE_EMULATOR_HOST) is not None

packages/google-cloud-firestore/tests/system/test_system.py

@@ -35,12 +35,20 @@
     RANDOM_ID_REGEX,
     MISSING_DOCUMENT,
     UNIQUE_RESOURCE_ID,
+    EMULATOR_CREDS,
+    FIRESTORE_EMULATOR,
 )
 
 
 def _get_credentials_and_project():
-    credentials = service_account.Credentials.from_service_account_file(FIRESTORE_CREDS)
-    project = FIRESTORE_PROJECT or credentials.project_id
+    if FIRESTORE_EMULATOR:
+        credentials = EMULATOR_CREDS
+        project = FIRESTORE_PROJECT
+    else:
+        credentials = service_account.Credentials.from_service_account_file(
+            FIRESTORE_CREDS
+        )
+        project = FIRESTORE_PROJECT or credentials.project_id
     return credentials, project
 
 
@@ -139,6 +147,7 @@ def test_create_document_w_subcollection(client, cleanup):
     assert sorted(child.id for child in children) == sorted(child_ids)
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137866686")
 def test_cannot_use_foreign_key(client, cleanup):
     document_id = "cannot" + UNIQUE_RESOURCE_ID
     document = client.document("foreign-key", document_id)
@@ -291,6 +300,7 @@ def test_document_update_w_int_field(client, cleanup):
     assert snapshot1.to_dict() == expected
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137867104")
 def test_update_document(client, cleanup):
     document_id = "for-update" + UNIQUE_RESOURCE_ID
     document = client.document("made", document_id)
@@ -880,6 +890,7 @@ def test_collection_group_queries_filters(client, cleanup):
     assert found == set(["cg-doc2"])
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137865992")
 def test_get_all(client, cleanup):
     collection_name = "get-all" + UNIQUE_RESOURCE_ID
 

packages/google-cloud-firestore/tests/system/test_system_async.py

@@ -34,15 +34,23 @@
     RANDOM_ID_REGEX,
     MISSING_DOCUMENT,
     UNIQUE_RESOURCE_ID,
+    EMULATOR_CREDS,
+    FIRESTORE_EMULATOR,
 )
 
 _test_event_loop = asyncio.new_event_loop()
 pytestmark = pytest.mark.asyncio
 
 
 def _get_credentials_and_project():
-    credentials = service_account.Credentials.from_service_account_file(FIRESTORE_CREDS)
-    project = FIRESTORE_PROJECT or credentials.project_id
+    if FIRESTORE_EMULATOR:
+        credentials = EMULATOR_CREDS
+        project = FIRESTORE_PROJECT
+    else:
+        credentials = service_account.Credentials.from_service_account_file(
+            FIRESTORE_CREDS
+        )
+        project = FIRESTORE_PROJECT or credentials.project_id
     return credentials, project
 
 
@@ -148,6 +156,7 @@ async def test_create_document_w_subcollection(client, cleanup):
     assert sorted([child.id async for child in children]) == sorted(child_ids)
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137866686")
 async def test_cannot_use_foreign_key(client, cleanup):
     document_id = "cannot" + UNIQUE_RESOURCE_ID
     document = client.document("foreign-key", document_id)
@@ -300,6 +309,7 @@ async def test_document_update_w_int_field(client, cleanup):
     assert snapshot1.to_dict() == expected
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137867104")
 async def test_update_document(client, cleanup):
     document_id = "for-update" + UNIQUE_RESOURCE_ID
     document = client.document("made", document_id)
@@ -905,6 +915,7 @@ async def test_collection_group_queries_filters(client, cleanup):
     assert found == set(["cg-doc2"])
 
 
+@pytest.mark.skipif(FIRESTORE_EMULATOR, reason="Internal Issue b/137865992")
 async def test_get_all(client, cleanup):
     collection_name = "get-all" + UNIQUE_RESOURCE_ID