test: Verify aggregate query error message when missing index (#1073)

Author: dconeybe

google-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITQueryAggregationsTest.java

@@ -28,6 +28,7 @@
 import static org.junit.Assume.assumeFalse;
 import static org.junit.Assume.assumeTrue;
 
+import com.google.api.core.ApiFuture;
 import com.google.cloud.firestore.*;
 import java.util.Map;
 import java.util.concurrent.ExecutionException;
@@ -1081,4 +1082,29 @@ public void aggregateInequalityFilterNoOrderByDocumentSnapshotReference2() throw
     AggregateQuerySnapshot snapshot = query.get().get();
     assertThat(snapshot.get(sum("num"))).isEqualTo(7);
   }
+
+  @Test
+  public void aggregateQueryShouldFailWithMessageWithConsoleLinkIfMissingIndex() {
+    assumeFalse(
+        "Skip this test when running against the Firestore emulator because the Firestore emulator "
+            + "does not use indexes and never fails with a 'missing index' error",
+        isRunningAgainstFirestoreEmulator(firestore));
+
+    CollectionReference collection = testCollection();
+    Query compositeIndexQuery = collection.whereEqualTo("field1", 42).whereLessThan("field2", 99);
+    AggregateQuery compositeIndexAggregateQuery =
+        compositeIndexQuery.aggregate(
+            AggregateField.count(), AggregateField.sum("pages"), AggregateField.average("pages"));
+    ApiFuture<AggregateQuerySnapshot> future = compositeIndexAggregateQuery.get();
+
+    ExecutionException executionException = assertThrows(ExecutionException.class, future::get);
+
+    Throwable throwable = executionException.getCause();
+    assertThat(throwable).hasMessageThat().ignoringCase().contains("index");
+    // TODO(b/316359394) Remove this check for the default databases once cl/582465034 is rolled out
+    //  to production.
+    if (collection.getFirestore().getOptions().getDatabaseId().equals("(default)")) {
+      assertThat(throwable).hasMessageThat().contains("https://console.firebase.google.com");
+    }
+  }
 }

google-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITQueryCountTest.java

@@ -22,6 +22,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.assumeFalse;
 import static org.junit.Assume.assumeTrue;
 
 import com.google.api.core.ApiFuture;
@@ -369,6 +370,29 @@ public void aggregateQueryInATransactionShouldRespectReadTime() throws Exception
     assertThat(transactionCount).isEqualTo(5);
   }
 
+  @Test
+  public void countQueryShouldFailWithMessageWithConsoleLinkIfMissingIndex() {
+    assumeFalse(
+        "Skip this test when running against the Firestore emulator because the Firestore emulator "
+            + "does not use indexes and never fails with a 'missing index' error",
+        isRunningAgainstFirestoreEmulator(firestore));
+
+    CollectionReference collection = createEmptyCollection();
+    Query compositeIndexQuery = collection.whereEqualTo("field1", 42).whereLessThan("field2", 99);
+    AggregateQuery compositeIndexCountQuery = compositeIndexQuery.count();
+    ApiFuture<AggregateQuerySnapshot> future = compositeIndexCountQuery.get();
+
+    ExecutionException executionException = assertThrows(ExecutionException.class, future::get);
+
+    Throwable throwable = executionException.getCause();
+    assertThat(throwable).hasMessageThat().ignoringCase().contains("index");
+    // TODO(b/316359394) Remove this check for the default databases once cl/582465034 is rolled out
+    //  to production.
+    if (collection.getFirestore().getOptions().getDatabaseId().equals("(default)")) {
+      assertThat(throwable).hasMessageThat().contains("https://console.firebase.google.com");
+    }
+  }
+
   private CollectionReference createEmptyCollection() {
     String collectionPath = "java-" + testName.getMethodName() + "-" + autoId();
     return firestore.collection(collectionPath);