1 files changed, 116 insertions, 0 deletions
diff --git a/Example/Database/Tests/Unit/FUtilitiesTest.m b/Example/Database/Tests/Unit/FUtilitiesTest.m
new file mode 100644
index 0000000..a012250
--- /dev/null
+++ b/Example/Database/Tests/Unit/FUtilitiesTest.m
@@ -0,0 +1,116 @@
+/*
+ * 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 <UIKit/UIKit.h>
+#import <XCTest/XCTest.h>
+#import "FUtilities.h"
+#import "FIRDatabase_Private.h"
+#import "FIRDatabaseReference_Private.h"
+#import "FClock.h"
+#import "FIRDatabaseConfig_Private.h"
+#import "FWebSocketConnection.h"
+#import "FConstants.h"
+
+@interface FWebSocketConnection (Tests)
+- (NSString*)userAgent;
+@end
+
+@interface FUtilitiesTest : XCTestCase
+
+@end
+
+@implementation FUtilitiesTest
+
+- (void)testUrlWithSchema {
+    FParsedUrl *parsedUrl = [FUtilities parseUrl:@"https://repo.firebaseio.com"];
+    XCTAssertEqualObjects(parsedUrl.repoInfo.host, @"repo.firebaseio.com");
+    XCTAssertEqualObjects(parsedUrl.repoInfo.namespace, @"repo");
+    XCTAssertTrue(parsedUrl.repoInfo.secure);
+    XCTAssertEqualObjects(parsedUrl.path, [FPath empty]);
+}
+
+- (void)testUrlParsedWithoutSchema {
+    FParsedUrl *parsedUrl = [FUtilities parseUrl:@"repo.firebaseio.com"];
+    XCTAssertEqualObjects(parsedUrl.repoInfo.host, @"repo.firebaseio.com");
+    XCTAssertEqualObjects(parsedUrl.repoInfo.namespace, @"repo");
+    XCTAssertTrue(parsedUrl.repoInfo.secure);
+    XCTAssertEqualObjects(parsedUrl.path, [FPath empty]);
+}
+
+- (void)testDefaultCacheSizeIs10MB {
+    XCTAssertEqual([FIRDatabaseReference defaultConfig].persistenceCacheSizeBytes, (NSUInteger)10*1024*1024);
+    XCTAssertEqual([FIRDatabaseConfig configForName:@"test-config"].persistenceCacheSizeBytes, (NSUInteger)10*1024*1024);
+}
+
+- (void)testSettingCacheSizeToHighOrToLowThrows {
+    FIRDatabaseConfig *config = [FIRDatabaseConfig configForName:@"config-tests-config"];
+    config.persistenceCacheSizeBytes = 5*1024*1024; // Works fine
+    XCTAssertThrows(config.persistenceCacheSizeBytes = (1024*1024-1));
+    XCTAssertThrows(config.persistenceCacheSizeBytes = 100*1024*1024+1);
+}
+
+- (void)testSystemClockMatchesCurrentTime {
+    NSTimeInterval currentTime = [[NSDate date] timeIntervalSince1970];
+    // Accuracy within 10ms
+    XCTAssertEqualWithAccuracy(currentTime, [[FSystemClock clock] currentTime], 0.010);
+}
+
+// This test is here for a lack of a better place to put it
+- (void)testUserAgentString {
+    FWebSocketConnection *conn = [[FWebSocketConnection alloc] init];
+
+    NSString *agent = [conn performSelector:@selector(userAgent) withObject:nil];
+
+    NSArray *parts = [agent componentsSeparatedByString:@"/"];
+    XCTAssertEqual(parts.count, (NSUInteger)5);
+    XCTAssertEqualObjects(parts[0], @"Firebase");
+    XCTAssertEqualObjects(parts[1], kWebsocketProtocolVersion); // Wire protocol version
+    XCTAssertEqualObjects(parts[2], [FIRDatabase buildVersion]); // Build version
+    XCTAssertEqualObjects(parts[3], [[UIDevice currentDevice] systemVersion]); // iOS Version
+#if TARGET_OS_IPHONE
+    NSString *deviceName = [UIDevice currentDevice].model;
+    XCTAssertEqualObjects([parts[4] componentsSeparatedByString:@"_"][0], deviceName);
+#endif
+
+}
+
+- (void)testKeyComparison {
+    NSArray *order = @[
+      @"-2147483648", @"0", @"1", @"2", @"10", @"2147483647", // Treated as integers
+      @"-2147483649", @"-2147483650", @"-a", @"2147483648", @"21474836480", @"2147483649", @"a" // treated as strings
+    ];
+    for (NSInteger i = 0; i < order.count; i++) {
+        for (NSInteger j = i + 1; j < order.count; j++) {
+            NSString *first = order[i];
+            NSString *second = order[j];
+            XCTAssertEqual([FUtilities compareKey:first toKey:second], NSOrderedAscending,
+                           @"Expected %@ < %@", first, second);
+            XCTAssertEqual([FUtilities compareKey:first toKey:first], NSOrderedSame,
+                           @"Expected %@ == %@", first, first);
+            XCTAssertEqual([FUtilities compareKey:second toKey:first], NSOrderedDescending,
+                           @"Expected %@ > %@", second, first);
+        }
+    }
+}
+
+// Enforce a > b, b < a, a != b, because this is apparently something that happens semi-regularly
+- (void)testUnicodeKeyComparison {
+  XCTAssertEqual([FUtilities compareKey:@"유주연" toKey:@"윤규완오빠"], NSOrderedAscending);
+  XCTAssertEqual([FUtilities compareKey:@"윤규완오빠" toKey:@"유주연"], NSOrderedDescending);
+  XCTAssertNotEqual([FUtilities compareKey:@"윤규완오빠" toKey:@"유주연"], NSOrderedSame);
+}
+
+@end