1 files changed, 174 insertions, 0 deletions
diff --git a/Example/Database/Tests/Unit/FNodeTests.m b/Example/Database/Tests/Unit/FNodeTests.m
new file mode 100644
index 0000000..372b84f
--- /dev/null
+++ b/Example/Database/Tests/Unit/FNodeTests.m
@@ -0,0 +1,174 @@
+/*
+ * 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 <Foundation/Foundation.h>
+#import <XCTest/XCTest.h>
+
+#import "FSnapshotUtilities.h"
+#import "FEmptyNode.h"
+#import "FChildrenNode.h"
+#import "FLeafNode.h"
+
+@interface FNodeTests : XCTestCase
+
+@end
+
+@implementation FNodeTests
+
+- (void) testLeafNodeEqualsHashCode {
+    id<FNode> falseNode = [FSnapshotUtilities nodeFrom:@NO];
+    id<FNode> trueNode = [FSnapshotUtilities nodeFrom:@YES];
+    id<FNode> stringOneNode = [FSnapshotUtilities nodeFrom:@"one"];
+    id<FNode> stringTwoNode = [FSnapshotUtilities nodeFrom:@"two"];
+    id<FNode> zeroNode = [FSnapshotUtilities nodeFrom:@0];
+    id<FNode> oneNode = [FSnapshotUtilities nodeFrom:@1];
+    id<FNode> emptyNode1 = [FSnapshotUtilities nodeFrom:nil];
+    id<FNode> emptyNode2 = [FSnapshotUtilities nodeFrom:[NSNull null]];
+
+    XCTAssertEqualObjects(falseNode, [FSnapshotUtilities nodeFrom:@NO]);
+    XCTAssertEqual(falseNode.hash, [FSnapshotUtilities nodeFrom:@NO].hash);
+    XCTAssertEqualObjects(trueNode, [FSnapshotUtilities nodeFrom:@YES]);
+    XCTAssertEqual(trueNode.hash, [FSnapshotUtilities nodeFrom:@YES].hash);
+    XCTAssertFalse([falseNode isEqual:trueNode]);
+    XCTAssertFalse([falseNode isEqual:oneNode]);
+    XCTAssertFalse([falseNode isEqual:stringOneNode]);
+    XCTAssertFalse([falseNode isEqual:emptyNode1]);
+
+    XCTAssertEqualObjects(stringOneNode, [FSnapshotUtilities nodeFrom:@"one"]);
+    XCTAssertEqual(stringOneNode.hash, [FSnapshotUtilities nodeFrom:@"one"].hash);
+    XCTAssertFalse([stringOneNode isEqual:stringTwoNode]);
+    XCTAssertFalse([stringOneNode isEqual:emptyNode1]);
+    XCTAssertFalse([stringOneNode isEqual:oneNode]);
+    XCTAssertFalse([stringOneNode isEqual:trueNode]);
+
+    XCTAssertEqualObjects(zeroNode, [FSnapshotUtilities nodeFrom:@0]);
+    XCTAssertEqual(zeroNode.hash, [FSnapshotUtilities nodeFrom:@0].hash);
+    XCTAssertFalse([zeroNode isEqual:oneNode]);
+    XCTAssertFalse([zeroNode isEqual:emptyNode1]);
+    XCTAssertFalse([zeroNode isEqual:falseNode]);
+
+    XCTAssertEqualObjects(emptyNode1, emptyNode2);
+    XCTAssertEqual(emptyNode1.hash, emptyNode2.hash);
+}
+
+- (void) testLeafNodePrioritiesEqualsHashCode {
+    id<FNode> oneOne = [FSnapshotUtilities nodeFrom:@1 priority:@1];
+    id<FNode> stringOne = [FSnapshotUtilities nodeFrom:@"value" priority:@1];
+    id<FNode> oneString = [FSnapshotUtilities nodeFrom:@1 priority:@"value"];
+    id<FNode> stringString = [FSnapshotUtilities nodeFrom:@"value" priority:@"value"];
+
+    XCTAssertEqualObjects(oneOne, [FSnapshotUtilities nodeFrom:@1 priority:@1]);
+    XCTAssertEqual(oneOne.hash, [FSnapshotUtilities nodeFrom:@1 priority:@1].hash);
+    XCTAssertFalse([oneOne isEqual:stringOne]);
+    XCTAssertFalse([oneOne isEqual:oneString]);
+    XCTAssertFalse([oneOne isEqual:stringString]);
+
+    XCTAssertEqualObjects(stringOne, [FSnapshotUtilities nodeFrom:@"value" priority:@1]);
+    XCTAssertEqual(stringOne.hash, [FSnapshotUtilities nodeFrom:@"value" priority:@1].hash);
+    XCTAssertFalse([stringOne isEqual:oneOne]);
+    XCTAssertFalse([stringOne isEqual:oneString]);
+    XCTAssertFalse([stringOne isEqual:stringString]);
+
+    XCTAssertEqualObjects(oneString, [FSnapshotUtilities nodeFrom:@1 priority:@"value"]);
+    XCTAssertEqual(oneString.hash, [FSnapshotUtilities nodeFrom:@1 priority:@"value"].hash);
+    XCTAssertFalse([oneString isEqual:stringOne]);
+    XCTAssertFalse([oneString isEqual:oneOne]);
+    XCTAssertFalse([oneString isEqual:stringString]);
+
+    XCTAssertEqualObjects(stringString, [FSnapshotUtilities nodeFrom:@"value" priority:@"value"]);
+    XCTAssertEqual(stringString.hash, [FSnapshotUtilities nodeFrom:@"value" priority:@"value"].hash);
+    XCTAssertFalse([stringString isEqual:stringOne]);
+    XCTAssertFalse([stringString isEqual:oneString]);
+    XCTAssertFalse([stringString isEqual:oneOne]);
+}
+
+- (void)testChildrenNodeEqualsHashCode {
+    id<FNode> nodeOne = [FSnapshotUtilities nodeFrom:@{ @"one": @1, @"two": @2, @".priority": @"prio"}];
+    id<FNode> nodeTwo = [[FEmptyNode emptyNode] updateImmediateChild:@"one" withNewChild:[FSnapshotUtilities nodeFrom:@1]];
+    nodeTwo = [nodeTwo updateImmediateChild:@"two" withNewChild:[FSnapshotUtilities nodeFrom:@2]];
+    nodeTwo = [nodeTwo updatePriority:[FSnapshotUtilities nodeFrom:@"prio"]];
+
+    XCTAssertEqualObjects(nodeOne, nodeTwo);
+    XCTAssertEqual(nodeOne.hash, nodeTwo.hash);
+    XCTAssertFalse([[nodeOne updatePriority:[FEmptyNode emptyNode]] isEqual:nodeOne]);
+    XCTAssertFalse([[nodeOne updateImmediateChild:@"one" withNewChild:[FEmptyNode emptyNode]] isEqual:nodeOne]);
+    XCTAssertFalse([[nodeOne updateImmediateChild:@"one" withNewChild:[FSnapshotUtilities nodeFrom:@2]] isEqual:nodeOne]);
+}
+
+- (void)testLeadingZerosWorkCorrectly {
+    NSDictionary *data = @{ @"1": @1, @"01": @2, @"001": @3, @"0001": @4 };
+
+    id<FNode> node = [FSnapshotUtilities nodeFrom:data];
+    XCTAssertEqualObjects([node getImmediateChild:@"1"].val, @1);
+    XCTAssertEqualObjects([node getImmediateChild:@"01"].val, @2);
+    XCTAssertEqualObjects([node getImmediateChild:@"001"].val, @3);
+    XCTAssertEqualObjects([node getImmediateChild:@"0001"].val, @4);
+}
+
+- (void)testLeadindZerosArePreservedInValue {
+    NSDictionary *data = @{ @"1": @1, @"01": @2, @"001": @3, @"0001": @4 };
+
+    XCTAssertEqualObjects([FSnapshotUtilities nodeFrom:data].val, data);
+}
+
+- (void)testEmptyNodeEqualsEmptyChildrenNode {
+    XCTAssertEqualObjects([FEmptyNode emptyNode], [[FChildrenNode alloc] init]);
+    XCTAssertEqualObjects([[FChildrenNode alloc] init], [FEmptyNode emptyNode]);
+    XCTAssertEqual([[FChildrenNode alloc] init].hash, [FEmptyNode emptyNode].hash);
+}
+
+- (void)testUpdatingEmptyChildrenDoesntOverwriteLeafNode {
+    FLeafNode *node = [[FLeafNode alloc] initWithValue:@"value"];
+    XCTAssertEqualObjects(node, [node updateChild:[FPath pathWithString:@".priority"] withNewChild:[FEmptyNode emptyNode]]);
+    XCTAssertEqualObjects(node, [node updateChild:[FPath pathWithString:@"child"] withNewChild:[FEmptyNode emptyNode]]);
+    XCTAssertEqualObjects(node, [node updateChild:[FPath pathWithString:@"child/.priority"] withNewChild:[FEmptyNode emptyNode]]);
+    XCTAssertEqualObjects(node, [node updateImmediateChild:@"child" withNewChild:[FEmptyNode emptyNode]]);
+    XCTAssertEqualObjects(node, [node updateImmediateChild:@".priority" withNewChild:[FEmptyNode emptyNode]]);
+}
+
+- (void)testUpdatingPrioritiesOnEmptyNodesIsANoOp {
+    id<FNode> priority = [FSnapshotUtilities nodeFrom:@"prio"];
+    XCTAssertTrue([[[[FEmptyNode emptyNode] updatePriority:priority] getPriority] isEmpty]);
+    XCTAssertTrue([[[[FEmptyNode emptyNode] updateChild:[FPath pathWithString:@".priority"] withNewChild:priority] getPriority] isEmpty]);
+    XCTAssertTrue([[[[FEmptyNode emptyNode] updateImmediateChild:@".priority" withNewChild:priority] getPriority] isEmpty]);
+
+    id<FNode> valueNode = [FSnapshotUtilities nodeFrom:@"value"];
+    FPath *childPath = [FPath pathWithString:@"child"];
+    id<FNode> reemptiedChildren = [[[FEmptyNode emptyNode] updateChild:childPath withNewChild:valueNode] updateChild:childPath withNewChild:[FEmptyNode emptyNode]];
+    XCTAssertTrue([[[reemptiedChildren updatePriority:priority] getPriority] isEmpty]);
+    XCTAssertTrue([[[reemptiedChildren updateChild:[FPath pathWithString:@".priority"] withNewChild:priority] getPriority] isEmpty]);
+    XCTAssertTrue([[[reemptiedChildren updateImmediateChild:@".priority" withNewChild:priority] getPriority] isEmpty]);
+}
+
+- (void)testDeletingLastChildFromChildrenNodeRemovesPriority {
+    id<FNode> priority = [FSnapshotUtilities nodeFrom:@"prio"];
+    id<FNode> valueNode = [FSnapshotUtilities nodeFrom:@"value"];
+    FPath *childPath = [FPath pathWithString:@"child"];
+    id<FNode> withPriority = [[[FEmptyNode emptyNode] updateChild:childPath withNewChild:valueNode] updatePriority:priority];
+    XCTAssertEqualObjects(priority, [withPriority getPriority]);
+    id<FNode> deletedChild = [withPriority updateChild:childPath withNewChild:[FEmptyNode emptyNode]];
+    XCTAssertTrue([[deletedChild getPriority] isEmpty]);
+}
+
+- (void)testFromNodeReturnsEmptyNodesWithoutPriority {
+    id<FNode> empty1 = [FSnapshotUtilities nodeFrom:@{ @".priority": @"prio" }];
+    XCTAssertTrue([[empty1 getPriority] isEmpty]);
+
+    id<FNode> empty2 = [FSnapshotUtilities nodeFrom:@{ @"dummy": [NSNull null], @".priority": @"prio" }];
+    XCTAssertTrue([[empty2 getPriority] isEmpty]);
+}
+
+@end