Add Test

Author: schmidt-sebastian

firebase-firestore/src/test/java/com/google/firebase/firestore/local/IndexingEnabledSQLiteLocalStoreTest.java

@@ -16,12 +16,15 @@
 
 import static com.google.common.truth.Truth.assertThat;
 import static com.google.firebase.firestore.testutil.TestUtil.addedRemoteEvent;
+import static com.google.firebase.firestore.testutil.TestUtil.deletedDoc;
 import static com.google.firebase.firestore.testutil.TestUtil.doc;
 import static com.google.firebase.firestore.testutil.TestUtil.fieldIndex;
 import static com.google.firebase.firestore.testutil.TestUtil.filter;
 import static com.google.firebase.firestore.testutil.TestUtil.map;
 import static com.google.firebase.firestore.testutil.TestUtil.noChangeEvent;
 import static com.google.firebase.firestore.testutil.TestUtil.query;
+import static com.google.firebase.firestore.testutil.TestUtil.updateRemoteEvent;
+import static java.util.Collections.emptyList;
 import static java.util.Collections.singletonList;
 
 import com.google.firebase.firestore.core.Query;
@@ -105,4 +108,30 @@ public void testUsesPartialIndexesWhenAvailable() {
     assertRemoteDocumentsRead(/* byKey= */ 1, /* byQuery= */ 1);
     assertQueryReturned("coll/a", "coll/b");
   }
+
+  @Test
+  public void testDeletedDocumentRemovesIndex() {
+    FieldIndex index = fieldIndex("coll", 0, FieldIndex.INITIAL_STATE, "matches", Kind.ASCENDING);
+    configureFieldIndexes(singletonList(index));
+
+    Query query = query("coll").filter(filter("matches", "==", true));
+    int targetId = allocateQuery(query);
+
+    applyRemoteEvent(addedRemoteEvent(doc("coll/a", 10, map("matches", true)), targetId));
+
+    // Add the document to the index
+    backfillIndexes();
+
+    executeQuery(query);
+    assertRemoteDocumentsRead(/* byKey= */ 1, /* byQuery= */ 0);
+    assertQueryReturned("coll/a");
+
+    applyRemoteEvent(
+        updateRemoteEvent(deletedDoc("coll/a", 0), singletonList(targetId), emptyList()));
+
+    // No backfill needed for deleted document.
+    executeQuery(query);
+    assertRemoteDocumentsRead(/* byKey= */ 0, /* byQuery= */ 0);
+    assertQueryReturned();
+  }
 }