Add LocalStoreTestCase

Author: schmidt-sebastian

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

@@ -21,6 +21,7 @@
 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.docMap;
+import static com.google.firebase.firestore.testutil.TestUtil.existenceFilterEvent;
 import static com.google.firebase.firestore.testutil.TestUtil.filter;
 import static com.google.firebase.firestore.testutil.TestUtil.key;
 import static com.google.firebase.firestore.testutil.TestUtil.keySet;
@@ -75,6 +76,7 @@
 import com.google.firebase.firestore.remote.WriteStream;
 import com.google.firebase.firestore.testutil.TestUtil;
 import com.google.firebase.firestore.util.AsyncQueue;
+import com.google.protobuf.ByteString;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
@@ -1123,6 +1125,38 @@ public void testUsesTargetMappingToExecuteQueries() {
     assertQueryReturned("foo/a", "foo/b");
   }
 
+  @Test
+  public void testIgnoresTargetMappingAfterExistenceFilterMismatch() {
+    assumeFalse(garbageCollectorIsEager());
+
+    Query query = query("foo").filter(filter("matches", "==", true));
+    int targetId = allocateQuery(query);
+
+    executeQuery(query);
+
+    // Persist a mapping with a single document
+    applyRemoteEvent(
+        addedRemoteEvent(
+            asList(doc("foo/a", 10, map("matches", true))), asList(targetId), emptyList()));
+    applyRemoteEvent(noChangeEvent(targetId, 10));
+    updateViews(targetId, /* fromCache= */ false);
+
+    TargetData cachedTargetData = localStore.getTargetData(query.toTarget());
+    Assert.assertEquals(version(10), cachedTargetData.getLastLimboFreeSnapshotVersion());
+
+    // Create an existence filter mismatch and verify that the last limbo free snapshot version
+    // is deleted
+    applyRemoteEvent(existenceFilterEvent(targetId, 2, 20));
+    cachedTargetData = localStore.getTargetData(query.toTarget());
+    Assert.assertEquals(version(0), cachedTargetData.getLastLimboFreeSnapshotVersion());
+    Assert.assertEquals(ByteString.EMPTY, cachedTargetData.getResumeToken());
+
+    // Re-run the query as a collection scan
+    executeQuery(query);
+    assertRemoteDocumentsRead(/* byKey= */ 0, /* byCollection= */ 1);
+    assertQueryReturned("foo/a");
+  }
+
   @Test
   public void testLastLimboFreeSnapshotIsAdvancedDuringViewProcessing() {
     // This test verifies that the `lastLimboFreeSnapshot` version for TargetData is advanced when

firebase-firestore/src/testUtil/java/com/google/firebase/firestore/testutil/TestUtil.java

@@ -71,6 +71,7 @@
 import com.google.firebase.firestore.model.mutation.Precondition;
 import com.google.firebase.firestore.model.mutation.SetMutation;
 import com.google.firebase.firestore.model.mutation.VerifyMutation;
+import com.google.firebase.firestore.remote.ExistenceFilter;
 import com.google.firebase.firestore.remote.RemoteEvent;
 import com.google.firebase.firestore.remote.TargetChange;
 import com.google.firebase.firestore.remote.WatchChange;
@@ -426,6 +427,20 @@ public static RemoteEvent addedRemoteEvent(
     return addedRemoteEvent(singletonList(doc), updatedInTargets, removedFromTargets);
   }
 
+  public static RemoteEvent existenceFilterEvent(int targetId, int count, int version) {
+    TargetData targetData = TestUtil.targetData(targetId, QueryPurpose.LISTEN, "foo");
+    TestTargetMetadataProvider testTargetMetadataProvider = new TestTargetMetadataProvider();
+    testTargetMetadataProvider.setSyncedKeys(targetData, DocumentKey.emptyKeySet());
+
+    ExistenceFilter existenceFilter = new ExistenceFilter(count);
+    WatchChangeAggregator aggregator = new WatchChangeAggregator(testTargetMetadataProvider);
+
+    WatchChange.ExistenceFilterWatchChange existenceFilterWatchChange =
+        new WatchChange.ExistenceFilterWatchChange(targetId, existenceFilter);
+    aggregator.handleExistenceFilter(existenceFilterWatchChange);
+    return aggregator.createRemoteEvent(version(version));
+  }
+
   public static RemoteEvent addedRemoteEvent(
       List<MutableDocument> docs,
       List<Integer> updatedInTargets,