Simplify IndexFree tests (#790)

Author: schmidt-sebastian

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

@@ -1073,49 +1073,31 @@ public void testUsesTargetMappingToExecuteQueries() {
 
   @Test
   public void testLastLimboFreeSnapshotIsAdvancedDuringViewProcessing() {
-    assumeFalse(garbageCollectorIsEager());
-    assumeTrue(queryEngine instanceof IndexFreeQueryEngine);
-
     // This test verifies that the `lastLimboFreeSnapshot` version for QueryData is advanced when
     // we compute a limbo-free free view and that the mapping is persisted when we release a query.
 
-    writeMutation(setMutation("foo/ignored", map("matches", false)));
-    acknowledgeMutation(10);
-
-    Query query =
-        Query.atPath(ResourcePath.fromString("foo")).filter(filter("matches", "==", true));
+    Query query = Query.atPath(ResourcePath.fromString("foo"));
     int targetId = allocateQuery(query);
 
-    // Mark the query as current.
-    applyRemoteEvent(
-        addedRemoteEvent(
-            asList(doc("foo/a", 10, map("matches", true)), doc("foo/b", 10, map("matches", true))),
-            asList(targetId),
-            emptyList()));
+    // Advance the query snapshot.
     applyRemoteEvent(noChangeEvent(targetId, 10));
 
     // At this point, we have not yet confirmed that the query is limbo free.
-    Assert.assertEquals(
-        SnapshotVersion.NONE, localStore.getQueryData(query).getLastLimboFreeSnapshotVersion());
+    QueryData cachedQueryData = localStore.getQueryData(query);
+    Assert.assertEquals(SnapshotVersion.NONE, cachedQueryData.getLastLimboFreeSnapshotVersion());
 
-    // Update the view, but don't mark the view synced.
-    Assert.assertEquals(
-        SnapshotVersion.NONE, localStore.getQueryData(query).getLastLimboFreeSnapshotVersion());
+    // Mark the view synced, which updates the last limbo free snapshot version.
     udpateViews(targetId, /* fromCache=*/ false);
-
-    // The query is marked limbo-free only when we mark the view synced.
-    udpateViews(targetId, /* fromCache=*/ false);
-    Assert.assertNotEquals(
-        SnapshotVersion.NONE, localStore.getQueryData(query).getLastLimboFreeSnapshotVersion());
+    cachedQueryData = localStore.getQueryData(query);
+    Assert.assertEquals(version(10), cachedQueryData.getLastLimboFreeSnapshotVersion());
 
     // The last limbo free snapshot version is persisted even if we release the query.
     releaseQuery(query);
-    allocateQuery(query);
 
-    // Verify that we only read the two documents that match the query.
-    executeQuery(query);
-    assertRemoteDocumentsRead(2);
-    assertQueryReturned("foo/a", "foo/b");
+    if (!garbageCollectorIsEager()) {
+      cachedQueryData = localStore.getQueryData(query);
+      Assert.assertEquals(version(10), cachedQueryData.getLastLimboFreeSnapshotVersion());
+    }
   }
 
   @Test
@@ -1130,28 +1112,26 @@ public void testQueriesIncludeLocallyModifiedDocuments() {
 
     applyRemoteEvent(
         addedRemoteEvent(
-            asList(doc("foo/a", 10, map("matches", true)), doc("foo/b", 10, map("matches", true))),
-            asList(targetId),
-            emptyList()));
+            asList(doc("foo/a", 10, map("matches", true))), asList(targetId), emptyList()));
     applyRemoteEvent(noChangeEvent(targetId, 10));
     udpateViews(targetId, /* fromCache= */ false);
 
     // Execute the query based on the RemoteEvent.
     executeQuery(query);
-    assertQueryReturned("foo/a", "foo/b");
+    assertQueryReturned("foo/a");
 
     // Write a document.
-    writeMutation(setMutation("foo/c", map("matches", true)));
+    writeMutation(setMutation("foo/b", map("matches", true)));
 
     // Execute the query and make sure that the pending mutation is included in the result.
     executeQuery(query);
-    assertQueryReturned("foo/a", "foo/b", "foo/c");
+    assertQueryReturned("foo/a", "foo/b");
 
     acknowledgeMutation(11);
 
     // Execute the query and make sure that the acknowledged mutation is included in the result.
     executeQuery(query);
-    assertQueryReturned("foo/a", "foo/b", "foo/c");
+    assertQueryReturned("foo/a", "foo/b");
   }
 
   @Test
@@ -1167,23 +1147,17 @@ public void testQueriesIncludeDocumentsFromOtherQueries() {
 
     applyRemoteEvent(
         addedRemoteEvent(
-            asList(doc("foo/a", 10, map("matches", true)), doc("foo/b", 10, map("matches", true))),
-            asList(targetId),
-            emptyList()));
+            asList(doc("foo/a", 10, map("matches", true))), asList(targetId), emptyList()));
     applyRemoteEvent(noChangeEvent(targetId, 10));
     udpateViews(targetId, /* fromCache=*/ false);
     releaseQuery(filteredQuery);
 
     // Start another query and add more matching documents to the collection.
-    Query fullQuery =
-        Query.atPath(ResourcePath.fromString("foo")).filter(filter("matches", "==", true));
-    targetId = allocateQuery(filteredQuery);
+    Query fullQuery = Query.atPath(ResourcePath.fromString("foo"));
+    targetId = allocateQuery(fullQuery);
     applyRemoteEvent(
         addedRemoteEvent(
-            asList(
-                doc("foo/a", 10, map("matches", true)),
-                doc("foo/b", 10, map("matches", true)),
-                doc("foo/c", 20, map("matches", true))),
+            asList(doc("foo/a", 10, map("matches", true)), doc("foo/b", 20, map("matches", true))),
             asList(targetId),
             emptyList()));
     releaseQuery(fullQuery);
@@ -1192,7 +1166,7 @@ public void testQueriesIncludeDocumentsFromOtherQueries() {
     // matches are included in the result set.
     allocateQuery(filteredQuery);
     executeQuery(filteredQuery);
-    assertQueryReturned("foo/a", "foo/b", "foo/c");
+    assertQueryReturned("foo/a", "foo/b");
   }
 
   @Test
@@ -1217,9 +1191,8 @@ public void testQueriesFilterDocumentsThatNoLongerMatch() {
     releaseQuery(filteredQuery);
 
     // Modify one of the documents to no longer match while the filtered query is inactive.
-    Query fullQuery =
-        Query.atPath(ResourcePath.fromString("foo")).filter(filter("matches", "==", true));
-    targetId = allocateQuery(filteredQuery);
+    Query fullQuery = Query.atPath(ResourcePath.fromString("foo"));
+    targetId = allocateQuery(fullQuery);
     applyRemoteEvent(
         addedRemoteEvent(
             asList(doc("foo/a", 10, map("matches", true)), doc("foo/b", 20, map("matches", false))),

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

@@ -418,13 +418,15 @@ public QueryData getQueryDataForTarget(int targetId) {
               }
             });
 
+    SnapshotVersion version = SnapshotVersion.NONE;
+
     for (MaybeDocument doc : docs) {
       DocumentChange change =
           new DocumentChange(updatedInTargets, removedFromTargets, doc.getKey(), doc);
       aggregator.handleDocumentChange(change);
+      version = doc.getVersion().compareTo(version) > 0 ? doc.getVersion() : version;
     }
 
-    SnapshotVersion version = docs.get(0).getVersion();
     return aggregator.createRemoteEvent(version);
   }