1 files changed, 66 insertions, 0 deletions
diff --git a/Firestore/Example/Tests/Remote/FSTWatchChangeTests.mm b/Firestore/Example/Tests/Remote/FSTWatchChangeTests.mm
new file mode 100644
index 0000000..df2496b
--- /dev/null
+++ b/Firestore/Example/Tests/Remote/FSTWatchChangeTests.mm
@@ -0,0 +1,66 @@
+/*
+ * 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/Source/Remote/FSTWatchChange.h"
+
+#import <XCTest/XCTest.h>
+
+#import "Firestore/Source/Model/FSTDocument.h"
+#import "Firestore/Source/Remote/FSTExistenceFilter.h"
+
+#import "Firestore/Example/Tests/Remote/FSTWatchChange+Testing.h"
+#import "Firestore/Example/Tests/Util/FSTHelpers.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface FSTWatchChangeTests : XCTestCase
+@end
+
+@implementation FSTWatchChangeTests
+
+- (void)testDocumentChange {
+  FSTMaybeDocument *doc = FSTTestDoc(@"a/b", 1, @{}, NO);
+  FSTDocumentWatchChange *change =
+      [[FSTDocumentWatchChange alloc] initWithUpdatedTargetIDs:@[ @1, @2, @3 ]
+                                              removedTargetIDs:@[ @4, @5 ]
+                                                   documentKey:doc.key
+                                                      document:doc];
+  XCTAssertEqual(change.updatedTargetIDs.count, 3);
+  XCTAssertEqual(change.removedTargetIDs.count, 2);
+  // Testing object identity here is fine.
+  XCTAssertEqual(change.document, doc);
+}
+
+- (void)testExistenceFilterChange {
+  FSTExistenceFilter *filter = [FSTExistenceFilter filterWithCount:7];
+  FSTExistenceFilterWatchChange *change =
+      [FSTExistenceFilterWatchChange changeWithFilter:filter targetID:5];
+  XCTAssertEqual(change.filter.count, 7);
+  XCTAssertEqual(change.targetID, 5);
+}
+
+- (void)testWatchTargetChange {
+  FSTWatchTargetChange *change =
+      [FSTWatchTargetChange changeWithState:FSTWatchTargetChangeStateReset
+                                  targetIDs:@[ @1, @2 ]
+                                      cause:nil];
+  XCTAssertEqual(change.state, FSTWatchTargetChangeStateReset);
+  XCTAssertEqual(change.targetIDs.count, 2);
+}
+
+@end
+
+NS_ASSUME_NONNULL_END