|
| 1 | +// |
| 2 | +// BNCUserAgentCollectorTests.m |
| 3 | +// Branch-SDK-Tests |
| 4 | +// |
| 5 | +// Created by Ernest Cho on 8/29/19. |
| 6 | +// Copyright © 2019 Branch, Inc. All rights reserved. |
| 7 | +// |
| 8 | + |
| 9 | +#import <XCTest/XCTest.h> |
| 10 | +#import "BNCPreferenceHelper.h" |
| 11 | +#import "BNCUserAgentCollector.h" |
| 12 | + |
| 13 | +// expose private methods for unit testing |
| 14 | +@interface BNCUserAgentCollector() |
| 15 | + |
| 16 | +- (NSString *)loadUserAgentForSystemBuildVersion:(NSString *)systemBuildVersion; |
| 17 | +- (void)saveUserAgent:(NSString *)userAgent forSystemBuildVersion:(NSString *)systemBuildVersion; |
| 18 | +- (void)collectUserAgentWithCompletion:(void (^)(NSString * _Nullable userAgent))completion; |
| 19 | + |
| 20 | +@end |
| 21 | + |
| 22 | +@interface BNCUserAgentCollectorTests : XCTestCase |
| 23 | + |
| 24 | +@end |
| 25 | + |
| 26 | +@implementation BNCUserAgentCollectorTests |
| 27 | + |
| 28 | ++ (void)setUp { |
| 29 | + [BNCUserAgentCollectorTests resetPersistentData]; |
| 30 | +} |
| 31 | + |
| 32 | +- (void)setUp { |
| 33 | + |
| 34 | +} |
| 35 | + |
| 36 | +- (void)tearDown { |
| 37 | + [BNCUserAgentCollectorTests resetPersistentData]; |
| 38 | +} |
| 39 | + |
| 40 | ++ (void)resetPersistentData { |
| 41 | + BNCPreferenceHelper *preferences = [BNCPreferenceHelper preferenceHelper]; |
| 42 | + preferences.browserUserAgentString = nil; |
| 43 | + preferences.lastSystemBuildVersion = nil; |
| 44 | +} |
| 45 | + |
| 46 | +- (void)testResetPersistentData { |
| 47 | + BNCPreferenceHelper *preferences = [BNCPreferenceHelper preferenceHelper]; |
| 48 | + XCTAssertNil(preferences.browserUserAgentString); |
| 49 | + XCTAssertNil(preferences.lastSystemBuildVersion); |
| 50 | +} |
| 51 | + |
| 52 | +- (void)testSaveAndLoadUserAgent { |
| 53 | + NSString *systemBuildVersion = @"test"; |
| 54 | + NSString *userAgent = @"UserAgent"; |
| 55 | + |
| 56 | + BNCUserAgentCollector *collector = [BNCUserAgentCollector new]; |
| 57 | + [collector saveUserAgent:userAgent forSystemBuildVersion:systemBuildVersion]; |
| 58 | + NSString *expected = [collector loadUserAgentForSystemBuildVersion:systemBuildVersion]; |
| 59 | + XCTAssertTrue([userAgent isEqualToString:expected]); |
| 60 | +} |
| 61 | + |
| 62 | +- (void)testCollectUserAgent { |
| 63 | + XCTestExpectation *expectation = [self expectationWithDescription:@"expectation"]; |
| 64 | + |
| 65 | + BNCUserAgentCollector *collector = [BNCUserAgentCollector new]; |
| 66 | + [collector collectUserAgentWithCompletion:^(NSString * _Nullable userAgent) { |
| 67 | + XCTAssertNotNil(userAgent); |
| 68 | + XCTAssertTrue([userAgent containsString:@"AppleWebKit"]); |
| 69 | + [expectation fulfill]; |
| 70 | + }]; |
| 71 | + |
| 72 | + [self waitForExpectationsWithTimeout:2.0 handler:^(NSError * _Nullable error) { |
| 73 | + |
| 74 | + }]; |
| 75 | +} |
| 76 | + |
| 77 | +- (void)testLoadUserAgent_EmptyDataStore { |
| 78 | + XCTestExpectation *expectation = [self expectationWithDescription:@"expectation"]; |
| 79 | + NSString *systemBuildVersion = @"test"; |
| 80 | + |
| 81 | + BNCUserAgentCollector *collector = [BNCUserAgentCollector new]; |
| 82 | + [collector loadUserAgentForSystemBuildVersion:systemBuildVersion withCompletion:^(NSString * _Nullable userAgent) { |
| 83 | + XCTAssertNotNil(userAgent); |
| 84 | + XCTAssertTrue([userAgent containsString:@"AppleWebKit"]); |
| 85 | + [expectation fulfill]; |
| 86 | + }]; |
| 87 | + |
| 88 | + [self waitForExpectationsWithTimeout:2.0 handler:^(NSError * _Nullable error) { |
| 89 | + |
| 90 | + }]; |
| 91 | +} |
| 92 | + |
| 93 | +- (void)testLoadUserAgent_FilledDataStore { |
| 94 | + XCTestExpectation *expectation = [self expectationWithDescription:@"expectation"]; |
| 95 | + NSString *systemBuildVersion = @"test"; |
| 96 | + NSString *savedUserAgent = @"UserAgent"; |
| 97 | + |
| 98 | + BNCUserAgentCollector *collector = [BNCUserAgentCollector new]; |
| 99 | + [collector saveUserAgent:savedUserAgent forSystemBuildVersion:systemBuildVersion]; |
| 100 | + [collector loadUserAgentForSystemBuildVersion:systemBuildVersion withCompletion:^(NSString * _Nullable userAgent) { |
| 101 | + XCTAssertNotNil(userAgent); |
| 102 | + XCTAssertTrue([userAgent isEqualToString:savedUserAgent]); |
| 103 | + XCTAssertFalse([userAgent containsString:@"AppleWebKit"]); |
| 104 | + [expectation fulfill]; |
| 105 | + }]; |
| 106 | + |
| 107 | + [self waitForExpectationsWithTimeout:2.0 handler:^(NSError * _Nullable error) { |
| 108 | + |
| 109 | + }]; |
| 110 | +} |
| 111 | + |
| 112 | +@end |
0 commit comments