fix: Add configurable to run tests against different firestore backen…

…ds (#1659)
googleapis · Apr 26, 2024 · a60d2a3 · a60d2a3
1 parent ac3a60b
commit a60d2a3
Showing 1 changed file with 29 additions and 1 deletion.
diff --git a/google-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITBaseTest.java b/google-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITBaseTest.java
@@ -19,6 +19,7 @@
 import static com.google.cloud.firestore.LocalFirestoreHelper.autoId;
 import static com.google.cloud.firestore.it.ITQueryTest.map;
 
+import com.google.api.gax.rpc.TransportChannelProvider;
 import com.google.cloud.firestore.DocumentReference;
 import com.google.cloud.firestore.Firestore;
 import com.google.cloud.firestore.FirestoreOptions;
@@ -54,15 +55,42 @@ public abstract class ITBaseTest {
   public void before() throws Exception {
     FirestoreOptions.Builder optionsBuilder = FirestoreOptions.newBuilder();
 
-    String namedDb = System.getProperty("FIRESTORE_NAMED_DATABASE");
+    String dbPropertyName = "FIRESTORE_NAMED_DATABASE";
+    String namedDb = System.getProperty(dbPropertyName);
+    if (namedDb == null) {
+      namedDb = System.getenv(dbPropertyName);
+    }
     if (namedDb != null) {
       logger.log(Level.INFO, "Integration test using named database " + namedDb);
       optionsBuilder = optionsBuilder.setDatabaseId(namedDb);
     } else {
       logger.log(Level.INFO, "Integration test using default database.");
     }
 
+    String targetPropertyName = "FIRESTORE_TARGET_BACKEND";
+    String targetBackend = System.getProperty(targetPropertyName);
+    if (targetBackend == null) {
+      targetBackend = System.getenv(targetPropertyName);
+    }
+    TransportChannelProvider defaultProvider = optionsBuilder.build().getTransportChannelProvider();
+    if (targetBackend != null) {
+      if (targetBackend.equals("PROD")) {
+        // do nothing to use the default
+      } else if (targetBackend.equals("QA")) {
+        optionsBuilder.setChannelProvider(
+            defaultProvider.withEndpoint("staging-firestore.sandbox.googleapis.com:443"));
+      } else if (targetBackend.equals("NIGHTLY")) {
+        optionsBuilder.setChannelProvider(
+            defaultProvider.withEndpoint("test-firestore.sandbox.googleapis.com:443"));
+      } else {
+        throw new IllegalArgumentException("Illegal target backend: " + targetBackend);
+      }
+    }
+
     firestoreOptions = optionsBuilder.build();
+    logger.log(
+        Level.INFO,
+        "Integration test against " + firestoreOptions.getTransportChannelProvider().getEndpoint());
     firestore = firestoreOptions.getService();
     primeBackend();
   }