Improve testIgnoresTargetMappingAfterExistenceFilterMismatch (#3306)

Author: schmidt-sebastian

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

@@ -1155,7 +1155,7 @@ public void testIgnoresTargetMappingAfterExistenceFilterMismatch() {
 
     // Create an existence filter mismatch and verify that the last limbo free snapshot version
     // is deleted
-    applyRemoteEvent(existenceFilterEvent(targetId, 2, 20));
+    applyRemoteEvent(existenceFilterEvent(targetId, keySet(key("foo/a")), 2, 20));
     cachedTargetData = localStore.getTargetData(query.toTarget());
     Assert.assertEquals(version(0), cachedTargetData.getLastLimboFreeSnapshotVersion());
     Assert.assertEquals(ByteString.EMPTY, cachedTargetData.getResumeToken());

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

@@ -427,12 +427,13 @@ public static RemoteEvent addedRemoteEvent(
     return addedRemoteEvent(singletonList(doc), updatedInTargets, removedFromTargets, null);
   }
 
-  public static RemoteEvent existenceFilterEvent(int targetId, int count, int version) {
+  public static RemoteEvent existenceFilterEvent(
+      int targetId, ImmutableSortedSet<DocumentKey> syncedKeys, int remoteCount, int version) {
     TargetData targetData = TestUtil.targetData(targetId, QueryPurpose.LISTEN, "foo");
     TestTargetMetadataProvider testTargetMetadataProvider = new TestTargetMetadataProvider();
-    testTargetMetadataProvider.setSyncedKeys(targetData, DocumentKey.emptyKeySet());
+    testTargetMetadataProvider.setSyncedKeys(targetData, syncedKeys);
 
-    ExistenceFilter existenceFilter = new ExistenceFilter(count);
+    ExistenceFilter existenceFilter = new ExistenceFilter(remoteCount);
     WatchChangeAggregator aggregator = new WatchChangeAggregator(testTargetMetadataProvider);
 
     WatchChange.ExistenceFilterWatchChange existenceFilterWatchChange =