1 files changed, 115 insertions, 0 deletions
diff --git a/Firestore/Example/Tests/API/FSTAPIHelpers.mm b/Firestore/Example/Tests/API/FSTAPIHelpers.mm
new file mode 100644
index 0000000..da899b7
--- /dev/null
+++ b/Firestore/Example/Tests/API/FSTAPIHelpers.mm
@@ -0,0 +1,115 @@
+/*
+ * Copyright 2017 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#import "Firestore/Example/Tests/API/FSTAPIHelpers.h"
+
+#import <FirebaseFirestore/FIRDocumentChange.h>
+#import <FirebaseFirestore/FIRDocumentReference.h>
+#import <FirebaseFirestore/FIRSnapshotMetadata.h>
+
+#import "Firestore/Source/API/FIRCollectionReference+Internal.h"
+#import "Firestore/Source/API/FIRDocumentReference+Internal.h"
+#import "Firestore/Source/API/FIRDocumentSnapshot+Internal.h"
+#import "Firestore/Source/API/FIRFirestore+Internal.h"
+#import "Firestore/Source/API/FIRQuerySnapshot+Internal.h"
+#import "Firestore/Source/API/FIRSnapshotMetadata+Internal.h"
+#import "Firestore/Source/Core/FSTQuery.h"
+#import "Firestore/Source/Core/FSTViewSnapshot.h"
+#import "Firestore/Source/Model/FSTDocument.h"
+#import "Firestore/Source/Model/FSTDocumentKey.h"
+#import "Firestore/Source/Model/FSTDocumentSet.h"
+#import "Firestore/Source/Model/FSTPath.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+FIRFirestore *FSTTestFirestore() {
+  static FIRFirestore *sharedInstance = nil;
+  static dispatch_once_t onceToken;
+
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Wnonnull"
+  dispatch_once(&onceToken, ^{
+    sharedInstance = [[FIRFirestore alloc] initWithProjectID:@"abc"
+                                                    database:@"abc"
+                                              persistenceKey:@"db123"
+                                         credentialsProvider:nil
+                                         workerDispatchQueue:nil
+                                                 firebaseApp:nil];
+  });
+#pragma clang diagnostic pop
+  return sharedInstance;
+}
+
+FIRDocumentSnapshot *FSTTestDocSnapshot(NSString *path,
+                                        FSTTestSnapshotVersion version,
+                                        NSDictionary<NSString *, id> *_Nullable data,
+                                        BOOL hasMutations,
+                                        BOOL fromCache) {
+  FSTDocument *doc = data ? FSTTestDoc(path, version, data, hasMutations) : nil;
+  return [FIRDocumentSnapshot snapshotWithFirestore:FSTTestFirestore()
+                                        documentKey:FSTTestDocKey(path)
+                                           document:doc
+                                          fromCache:fromCache];
+}
+
+FIRCollectionReference *FSTTestCollectionRef(NSString *path) {
+  return [FIRCollectionReference referenceWithPath:FSTTestPath(path) firestore:FSTTestFirestore()];
+}
+
+FIRDocumentReference *FSTTestDocRef(NSString *path) {
+  return [FIRDocumentReference referenceWithPath:FSTTestPath(path) firestore:FSTTestFirestore()];
+}
+
+/** A convenience method for creating a query snapshots for tests. */
+FIRQuerySnapshot *FSTTestQuerySnapshot(
+    NSString *path,
+    NSDictionary<NSString *, NSDictionary<NSString *, id> *> *oldDocs,
+    NSDictionary<NSString *, NSDictionary<NSString *, id> *> *docsToAdd,
+    BOOL hasPendingWrites,
+    BOOL fromCache) {
+  FIRSnapshotMetadata *metadata =
+      [FIRSnapshotMetadata snapshotMetadataWithPendingWrites:hasPendingWrites fromCache:fromCache];
+  FSTDocumentSet *oldDocuments = FSTTestDocSet(FSTDocumentComparatorByKey, @[]);
+  for (NSString *key in oldDocs) {
+    oldDocuments = [oldDocuments
+        documentSetByAddingDocument:FSTTestDoc([NSString stringWithFormat:@"%@/%@", path, key], 1,
+                                               oldDocs[key], hasPendingWrites)];
+  }
+  FSTDocumentSet *newDocuments = oldDocuments;
+  NSArray<FSTDocumentViewChange *> *documentChanges = [NSArray array];
+  for (NSString *key in docsToAdd) {
+    FSTDocument *docToAdd = FSTTestDoc([NSString stringWithFormat:@"%@/%@", path, key], 1,
+                                       docsToAdd[key], hasPendingWrites);
+    newDocuments = [newDocuments documentSetByAddingDocument:docToAdd];
+    documentChanges = [documentChanges
+        arrayByAddingObject:[FSTDocumentViewChange
+                                changeWithDocument:docToAdd
+                                              type:FSTDocumentViewChangeTypeAdded]];
+  }
+  FSTViewSnapshot *viewSnapshot = [[FSTViewSnapshot alloc] initWithQuery:FSTTestQuery(path)
+                                                               documents:newDocuments
+                                                            oldDocuments:oldDocuments
+                                                         documentChanges:documentChanges
+                                                               fromCache:fromCache
+                                                        hasPendingWrites:hasPendingWrites
+                                                        syncStateChanged:YES];
+  return [FIRQuerySnapshot snapshotWithFirestore:FSTTestFirestore()
+                                   originalQuery:FSTTestQuery(path)
+                                        snapshot:viewSnapshot
+                                        metadata:metadata];
+}
+
+NS_ASSUME_NONNULL_END