Fix test. (#2147)

Author: Michael Lehenbauer

packages/firestore/test/unit/local/remote_document_cache.test.ts

@@ -360,7 +360,7 @@ function genericRemoteDocumentCacheTests(
       ],
       version(3)
     );
-    await cache.removeEntry(key('a/2'));
+    await cache.removeEntry(key('a/2'), version(4));
 
     const changedDocs = await cache.getNewDocumentChanges();
     assertMatches(

packages/firestore/test/unit/local/test_remote_document_cache.ts

@@ -76,14 +76,19 @@ export class TestRemoteDocumentCache {
     return this.addEntries([maybeDocument], maybeDocument.version);
   }
 
-  removeEntry(documentKey: DocumentKey): Promise<void> {
+  removeEntry(
+    documentKey: DocumentKey,
+    version?: SnapshotVersion
+  ): Promise<void> {
     return this.persistence.runTransaction(
       'removeEntry',
       'readwrite-primary',
       txn => {
-        const changeBuffer = this.newChangeBuffer();
+        const changeBuffer = this.newChangeBuffer(
+          version ? { trackRemovals: true } : undefined
+        );
         return changeBuffer.getEntry(txn, documentKey).next(() => {
-          changeBuffer.removeEntry(documentKey);
+          changeBuffer.removeEntry(documentKey, version);
           return changeBuffer.apply(txn);
         });
       }
@@ -128,7 +133,9 @@ export class TestRemoteDocumentCache {
     );
   }
 
-  newChangeBuffer(): RemoteDocumentChangeBuffer {
-    return this.cache.newChangeBuffer();
+  newChangeBuffer(options?: {
+    trackRemovals: boolean;
+  }): RemoteDocumentChangeBuffer {
+    return this.cache.newChangeBuffer(options);
   }
 }