1 files changed, 15 insertions, 0 deletions
diff --git a/Firestore/Source/Core/FSTSyncEngine.m b/Firestore/Source/Core/FSTSyncEngine.m
index 98658e4..aa1ccae 100644
--- a/Firestore/Source/Core/FSTSyncEngine.m
+++ b/Firestore/Source/Core/FSTSyncEngine.m
@@ -318,6 +318,21 @@ NS_ASSUME_NONNULL_BEGIN
   [self emitNewSnapshotsWithChanges:changes remoteEvent:remoteEvent];
 }
 
+- (void)applyOnlineStateChange:(FSTOnlineState)onlineState {
+  NSMutableArray<FSTViewSnapshot *> *newViewSnapshots = [NSMutableArray array];
+  [self.queryViewsByQuery
+      enumerateKeysAndObjectsUsingBlock:^(FSTQuery *query, FSTQueryView *queryView, BOOL *stop) {
+        FSTViewChange *viewChange = [queryView.view applyOnlineStateChange:onlineState];
+        FSTAssert(viewChange.limboChanges.count == 0,
+                  @"OnlineState should not affect limbo documents.");
+        if (viewChange.snapshot) {
+          [newViewSnapshots addObject:viewChange.snapshot];
+        }
+      }];
+
+  [self.delegate handleViewSnapshots:newViewSnapshots];
+}
+
 - (void)rejectListenWithTargetID:(FSTBoxedTargetID *)targetID error:(NSError *)error {
   [self assertDelegateExistsForSelector:_cmd];