Fix integration test

Author: schmidt-sebastian

Firestore/Example/Tests/Integration/API/FIRDatabaseTests.mm

@@ -65,9 +65,14 @@ - (void)testCanUpdateAnUnknownDocument {
         [self readDocumentForRef:writerRef source:FIRFirestoreSourceCache];
     XCTAssertTrue(writerSnap.exists);
 
-    XCTAssertThrows(^() {
-      [self readDocumentForRef:readerRef source:FIRFirestoreSourceCache];
-    });
+    XCTestExpectation *expectation =
+        [self expectationWithDescription:@"testCanUpdateAnUnknownDocument"];
+    [readerRef getDocumentWithSource:FIRFirestoreSourceCache
+                          completion:^(FIRDocumentSnapshot *doc, NSError *_Nullable error) {
+                            XCTAssertNotNil(error);
+                            [expectation fulfill];
+                          }];
+    [self awaitExpectations];
 
     writerSnap = [self readDocumentForRef:writerRef];
     XCTAssertEqualObjects(writerSnap.data, (@{@"a" : @"a", @"b" : @"b"}));

Firestore/Source/Core/FSTFirestoreClient.mm

@@ -287,14 +287,20 @@ - (void)getDocumentFromLocalCache:(FIRDocumentReference *)doc
     FSTMaybeDocument *maybeDoc = [self.localStore readDocument:doc.key];
     FIRDocumentSnapshot *_Nullable result = nil;
     NSError *_Nullable error = nil;
-    if (maybeDoc) {
-      FSTDocument *_Nullable document =
-          ([maybeDoc isKindOfClass:[FSTDocument class]]) ? (FSTDocument *)maybeDoc : nil;
+
+    if ([maybeDoc isKindOfClass:[FSTDocument class]]) {
+      FSTDocument *document = (FSTDocument *)maybeDoc;
       result = [FIRDocumentSnapshot snapshotWithFirestore:doc.firestore
                                               documentKey:doc.key
                                                  document:document
                                                 fromCache:YES
                                          hasPendingWrites:document.hasLocalMutations];
+    } else if ([maybeDoc isKindOfClass:[FSTDeletedDocument class]]) {
+      result = [FIRDocumentSnapshot snapshotWithFirestore:doc.firestore
+                                              documentKey:doc.key
+                                                 document:nil
+                                                fromCache:YES
+                                         hasPendingWrites:NO];
     } else {
       error = [NSError errorWithDomain:FIRFirestoreErrorDomain
                                   code:FIRFirestoreErrorCodeUnavailable

Firestore/Source/Local/FSTLocalStore.mm

@@ -407,7 +407,7 @@ - (void)releaseQuery:(FSTQuery *)query {
       queryData = cachedQueryData;
       [self.queryCache updateQueryData:queryData];
     }
-    
+
     // References for documents sent via Watch are automatically removed when we delete a
     // query's target data from the reference delegate. Since this does not remove references
     // for locally mutated documents, we have to remove the target associations for these