Skip aggregateQueryInATransactionShouldLockTheCountedDocuments when running against prod (b/248152832)

dconeybe · dconeybe · commit 17543221150e · 2022-09-22T00:09:30.000-04:00
diff --git a/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java b/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java
@@ -20,6 +20,7 @@
 import static com.google.common.truth.Truth.assertThat;
 import static java.util.Collections.singletonMap;
 import static org.junit.Assert.assertThrows;
+import static org.junit.Assume.assumeTrue;
 
 import com.google.api.core.ApiFuture;
 import com.google.auto.value.AutoValue;
@@ -215,6 +216,11 @@ public void aggregateQueryShouldWorkInATransaction() throws Exception {
 
   @Test
   public void aggregateQueryInATransactionShouldLockTheCountedDocuments() throws Exception {
+    assumeTrue(
+        "Skip this test when running against production because "
+            + "it appears that production is failing to lock the counted documents b/248152832",
+        isRunningAgainstFirestoreEmulator());
+
     CollectionReference collection = createEmptyCollection();
     DocumentReference document = createDocumentInCollection(collection);
     CountDownLatch aggregateQueryExecutedSignal = new CountDownLatch(1);
@@ -388,6 +394,11 @@ private static void await(ApiFuture<?> future) throws InterruptedException {
     executor.shutdown();
   }
 
+  /** Returns whether the tests are running against the Firestore emulator. */
+  private boolean isRunningAgainstFirestoreEmulator() {
+    return firestore.getOptions().getHost().startsWith("localhost:");
+  }
+
   @AutoValue
   abstract static class CreatedCollectionInfo {
 
