/*

* Copyright 2019 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 <XCTest/XCTest.h>

#import "OCMStubRecorder.h"

#import "OCMock.h"

#import "FirebaseRemoteConfig/Sources/Private/FIRRemoteConfig_Private.h"

#import "FirebaseRemoteConfig/Sources/Private/RCNConfigFetch.h"

#import "FirebaseRemoteConfig/Sources/Public/FirebaseRemoteConfig/FIRRemoteConfig.h"

#import "FirebaseRemoteConfig/Sources/RCNConfigConstants.h"

#import "FirebaseRemoteConfig/Sources/RCNConfigDBManager.h"

#import "FirebaseRemoteConfig/Sources/RCNUserDefaultsManager.h"

#import "FirebaseRemoteConfig/Tests/Unit/RCNTestUtilities.h"

#import <GoogleUtilities/GULNSData+zlib.h>

#import "FirebaseCore/Sources/Private/FirebaseCoreInternal.h"

@interface RCNConfigFetch (ForTest)

- (instancetype)initWithContent:(RCNConfigContent *)content

DBManager:(RCNConfigDBManager *)DBManager

settings:(RCNConfigSettings *)settings

experiment:(RCNConfigExperiment *)experiment

queue:(dispatch_queue_t)queue

namespace:firebaseNamespace

app:firebaseApp;

/// Skip fetching user properties from analytics because we cannot mock the action here. Instead

/// overriding the method to skip.

- (void)fetchWithUserPropertiesCompletionHandler:(FIRAInteropUserPropertiesCallback)block;

- (NSURLSessionDataTask *)URLSessionDataTaskWithContent:(NSData *)content

completionHandler:

(RCNConfigFetcherCompletion)fetcherCompletion;

- (void)fetchWithUserProperties:(NSDictionary *)userProperties

completionHandler:(FIRRemoteConfigFetchCompletion)completionHandler;

- (NSString *)constructServerURL;

- (NSURLSession *)currentNetworkSession;

@end

@interface FIRRemoteConfig (ForTest)

- (void)updateWithNewInstancesForConfigFetch:(RCNConfigFetch *)configFetch

configContent:(RCNConfigContent *)configContent

configSettings:(RCNConfigSettings *)configSettings

configExperiment:(RCNConfigExperiment *)configExperiment;

@end

@implementation FIRRemoteConfig (ForTest)

- (void)updateWithNewInstancesForConfigFetch:(RCNConfigFetch *)configFetch

configContent:(RCNConfigContent *)configContent

configSettings:(RCNConfigSettings *)configSettings

configExperiment:(RCNConfigExperiment *)configExperiment {

[self setValue:configFetch forKey:@"_configFetch"];

[self setValue:configContent forKey:@"_configContent"];

[self setValue:configSettings forKey:@"_settings"];

[self setValue:configExperiment forKey:@"_configExperiment"];

}

@end

@interface RCNConfigDBManager (Test)

- (void)removeDatabaseOnDatabaseQueueAtPath:(NSString *)path;

@end

@interface RCNUserDefaultsManager (Test)

+ (NSUserDefaults *)sharedUserDefaultsForBundleIdentifier:(NSString *)bundleIdentifier;

@end

typedef NS_ENUM(NSInteger, RCNTestRCInstance) {

RCNTestRCInstanceDefault,

RCNTestRCNumTotalInstances, // TODO(mandard): Remove once OCMock issue is resolved (#4877).

RCNTestRCInstanceSecondNamespace,

RCNTestRCInstanceSecondApp,

};

@interface RCNRemoteConfigTest : XCTestCase {

NSTimeInterval _expectationTimeout;

NSTimeInterval _checkCompletionTimeout;

NSMutableArray<FIRRemoteConfig *> *_configInstances;

NSMutableArray<NSDictionary<NSString *, NSString *> *> *_entries;

NSMutableArray<NSDictionary<NSString *, id> *> *_response;

NSMutableArray<NSData *> *_responseData;

NSMutableArray<NSURLResponse *> *_URLResponse;

NSMutableArray<RCNConfigFetch *> *_configFetch;

RCNConfigDBManager *_DBManager;

NSUserDefaults *_userDefaults;

NSString *_userDefaultsSuiteName;

NSString *_DBPath;

id _DBManagerMock;

id _userDefaultsMock;

}

@end

@implementation RCNRemoteConfigTest

- (void)setUp {

[super setUp];

FIRSetLoggerLevel(FIRLoggerLevelMax);

_expectationTimeout = 5;

_checkCompletionTimeout = 1.0;

// Always remove the database at the start of testing.

_DBPath = [RCNTestUtilities remoteConfigPathForTestDatabase];

_DBManagerMock = OCMClassMock([RCNConfigDBManager class]);

OCMStub([_DBManagerMock remoteConfigPathForDatabase]).andReturn(_DBPath);

_DBManager = [[RCNConfigDBManager alloc] init];

_userDefaultsSuiteName = [RCNTestUtilities userDefaultsSuiteNameForTestSuite];

_userDefaults = [[NSUserDefaults alloc] initWithSuiteName:_userDefaultsSuiteName];

_userDefaultsMock = OCMClassMock([RCNUserDefaultsManager class]);

OCMStub([_userDefaultsMock sharedUserDefaultsForBundleIdentifier:[OCMArg any]])

.andReturn(_userDefaults);

RCNConfigContent *configContent = [[RCNConfigContent alloc] initWithDBManager:_DBManager];

_configInstances = [[NSMutableArray alloc] initWithCapacity:3];

_entries = [[NSMutableArray alloc] initWithCapacity:3];

_response = [[NSMutableArray alloc] initWithCapacity:3];

_responseData = [[NSMutableArray alloc] initWithCapacity:3];

_URLResponse = [[NSMutableArray alloc] initWithCapacity:3];

_configFetch = [[NSMutableArray alloc] initWithCapacity:3];

// Populate the default, second app, second namespace instances.

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

// Fake a response for default instance.

NSMutableDictionary<NSString *, NSString *> *valuesDict = [[NSMutableDictionary alloc] init];

for (int count = 1; count <= 100; count++) {

NSString *key = [NSString stringWithFormat:@"key%d-%d", count, i];

NSString *value = [NSString stringWithFormat:@"value%d-%d", count, i];

valuesDict[key] = value;

}

_entries[i] = valuesDict;

NSString *currentAppName = nil;

FIROptions *currentOptions = nil;

NSString *currentNamespace = nil;

switch (i) {

case RCNTestRCInstanceSecondNamespace:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsPerfNamespace;

break;

case RCNTestRCInstanceSecondApp:

currentAppName = RCNTestsSecondFIRAppName;

currentOptions = [self secondAppOptions];

currentNamespace = FIRNamespaceGoogleMobilePlatform;

break;

case RCNTestRCInstanceDefault:

default:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsFIRNamespace;

break;

}

NSString *fullyQualifiedNamespace =

[NSString stringWithFormat:@"%@:%@", currentNamespace, currentAppName];

FIRRemoteConfig *config =

OCMPartialMock([[FIRRemoteConfig alloc] initWithAppName:currentAppName

FIROptions:currentOptions

namespace:currentNamespace

DBManager:_DBManager

configContent:configContent

analytics:nil]);

_configInstances[i] = config;

RCNConfigSettings *settings =

[[RCNConfigSettings alloc] initWithDatabaseManager:_DBManager

namespace:fullyQualifiedNamespace

firebaseAppName:currentAppName

googleAppID:currentOptions.googleAppID];

dispatch_queue_t queue = dispatch_queue_create(

[[NSString stringWithFormat:@"testqueue: %d", i] cStringUsingEncoding:NSUTF8StringEncoding],

DISPATCH_QUEUE_SERIAL);

_configFetch[i] = OCMPartialMock([[RCNConfigFetch alloc] initWithContent:configContent

DBManager:_DBManager

settings:settings

analytics:nil

experiment:nil

queue:queue

namespace:fullyQualifiedNamespace

options:currentOptions]);

OCMStubRecorder *mock = OCMStub([_configFetch[i] fetchConfigWithExpirationDuration:0

completionHandler:OCMOCK_ANY]);

mock = [mock ignoringNonObjectArgs];

mock.andDo(^(NSInvocation *invocation) {

__unsafe_unretained void (^handler)(FIRRemoteConfigFetchStatus status,

NSError *_Nullable error) = nil;

[invocation getArgument:&handler atIndex:3];

[self->_configFetch[i] fetchWithUserProperties:[[NSDictionary alloc] init]

completionHandler:handler];

});

_response[i] = @{@"state" : @"UPDATE", @"entries" : _entries[i]};

_responseData[i] = [NSJSONSerialization dataWithJSONObject:_response[i] options:0 error:nil];

_URLResponse[i] = [[NSHTTPURLResponse alloc]

initWithURL:[NSURL URLWithString:@"https://firebase.com"]

statusCode:200

HTTPVersion:nil

headerFields:@{@"etag" : [NSString stringWithFormat:@"etag1-%d", i]}];

id completionBlock =

[OCMArg invokeBlockWithArgs:_responseData[i], _URLResponse[i], [NSNull null], nil];

OCMExpect([_configFetch[i] URLSessionDataTaskWithContent:[OCMArg any]

completionHandler:completionBlock])

.andReturn(nil);

[_configInstances[i] updateWithNewInstancesForConfigFetch:_configFetch[i]

configContent:configContent

configSettings:settings

configExperiment:nil];

}

}

- (void)tearDown {

[_DBManager removeDatabaseOnDatabaseQueueAtPath:_DBPath];

[[NSUserDefaults standardUserDefaults] removePersistentDomainForName:_userDefaultsSuiteName];

[_DBManagerMock stopMocking];

_DBManagerMock = nil;

[_userDefaultsMock stopMocking];

_userDefaultsMock = nil;

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

[(id)_configInstances[i] stopMocking];

[(id)_configFetch[i] stopMocking];

}

[_configInstances removeAllObjects];

[_configFetch removeAllObjects];

_configInstances = nil;

_configFetch = nil;

[super tearDown];

}

- (void)testFetchConfigWithNilCallback {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Set defaults no callback expectation - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:nil];

dispatch_after(

dispatch_time(DISPATCH_TIME_NOW, (int64_t)(_checkCompletionTimeout * NSEC_PER_SEC)),

dispatch_get_main_queue(), ^{

XCTAssertEqual(self->_configInstances[i].lastFetchStatus,

FIRRemoteConfigFetchStatusSuccess);

[expectations[i] fulfill];

});

}

[self waitForExpectationsWithTimeout:_expectationTimeout handler:nil];

}

- (void)testFetchConfigsSuccessfully {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] =

[self expectationWithDescription:

[NSString stringWithFormat:@"Test fetch configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertTrue(changed);

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *value1 = [NSString stringWithFormat:@"value1-%d", i];

NSString *value2 = [NSString stringWithFormat:@"value2-%d", i];

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, value1);

XCTAssertEqualObjects(self->_configInstances[i][key2].stringValue, value2);

OCMVerify([self->_configInstances[i] objectForKeyedSubscript:key1]);

XCTAssertEqual(status, FIRRemoteConfigFetchStatusSuccess,

@"Callback of first successful config "

@"fetch. Status must equal to FIRRemoteConfigFetchStatusSuccessFresh.");

XCTAssertNotNil(self->_configInstances[i].lastFetchTime);

XCTAssertGreaterThan(self->_configInstances[i].lastFetchTime.timeIntervalSince1970, 0,

@"last fetch time interval should be set.");

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testFetchAndActivate {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] =

[self expectationWithDescription:

[NSString stringWithFormat:@"Test fetch configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchAndActivateCompletion fetchAndActivateCompletion = ^void(

FIRRemoteConfigFetchAndActivateStatus status, NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *value1 = [NSString stringWithFormat:@"value1-%d", i];

NSString *value2 = [NSString stringWithFormat:@"value2-%d", i];

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, value1);

XCTAssertEqualObjects(self->_configInstances[i][key2].stringValue, value2);

OCMVerify([self->_configInstances[i] objectForKeyedSubscript:key1]);

XCTAssertEqual(

status, FIRRemoteConfigFetchAndActivateStatusSuccessFetchedFromRemote,

@"Callback of first successful config "

@"fetchAndActivate status must equal to FIRRemoteConfigFetchAndStatusSuccessFromRemote.");

XCTAssertNotNil(self->_configInstances[i].lastFetchTime);

XCTAssertGreaterThan(self->_configInstances[i].lastFetchTime.timeIntervalSince1970, 0,

@"last fetch time interval should be set.");

[expectations[i] fulfill];

};

// Update the minimum fetch interval to 0. This disables the cache and forces a remote fetch

// request.

FIRRemoteConfigSettings *settings = [[FIRRemoteConfigSettings alloc] init];

settings.minimumFetchInterval = 0;

[_configInstances[i] setConfigSettings:settings];

[_configInstances[i] fetchAndActivateWithCompletionHandler:fetchAndActivateCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

// TODO: Try splitting into smaller tests.

- (void)testFetchConfigsSuccessfullyWithNewActivateMethodSignature {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] =

[self expectationWithDescription:

[NSString stringWithFormat:@"Test fetch configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertTrue(changed);

XCTAssertNil(error);

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *value1 = [NSString stringWithFormat:@"value1-%d", i];

NSString *value2 = [NSString stringWithFormat:@"value2-%d", i];

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, value1);

XCTAssertEqualObjects(self->_configInstances[i][key2].stringValue, value2);

OCMVerify([self->_configInstances[i] objectForKeyedSubscript:key1]);

XCTAssertEqual(status, FIRRemoteConfigFetchStatusSuccess,

@"Callback of first successful config "

@"fetch. Status must equal to FIRRemoteConfigFetchStatusSuccessFresh.");

XCTAssertNotNil(self->_configInstances[i].lastFetchTime);

XCTAssertGreaterThan(self->_configInstances[i].lastFetchTime.timeIntervalSince1970, 0,

@"last fetch time interval should be set.");

// A second activate should have no effect.

[self->_configInstances[i]

activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertFalse(changed);

XCTAssertNil(error);

}];

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testEnumeratingConfigResults {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Test enumerating configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertTrue(changed);

NSString *key5 = [NSString stringWithFormat:@"key5-%d", i];

NSString *key19 = [NSString stringWithFormat:@"key19-%d", i];

NSString *value5 = [NSString stringWithFormat:@"value5-%d", i];

NSString *value19 = [NSString stringWithFormat:@"value19-%d", i];

XCTAssertEqualObjects(self->_configInstances[i][key5].stringValue, value5);

XCTAssertEqualObjects(self->_configInstances[i][key19].stringValue, value19);

// Test enumerating config values.

for (NSString *key in self->_configInstances[i]) {

if ([key isEqualToString:key5]) {

XCTAssertEqualObjects(self->_configInstances[i][key5].stringValue, value5);

}

if ([key isEqualToString:key19]) {

XCTAssertEqualObjects(self->_configInstances[i][key19].stringValue, value19);

}

}

XCTAssertEqual(status, FIRRemoteConfigFetchStatusSuccess,

@"Callback of first successful config "

@"fetch. Status must equal to FIRRemoteConfigFetchStatusSuccessFresh.");

XCTAssertNotNil(self->_configInstances[i].lastFetchTime);

XCTAssertGreaterThan(self->_configInstances[i].lastFetchTime.timeIntervalSince1970, 0,

@"last fetch time interval should be set.");

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testFetchConfigsFailed {

// Override the setup values to return back an error status.

RCNConfigContent *configContent = [[RCNConfigContent alloc] initWithDBManager:_DBManager];

// Populate the default, second app, second namespace instances.

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

NSString *currentAppName = nil;

FIROptions *currentOptions = nil;

NSString *currentNamespace = nil;

switch (i) {

case RCNTestRCInstanceSecondNamespace:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsPerfNamespace;

break;

case RCNTestRCInstanceSecondApp:

currentAppName = RCNTestsSecondFIRAppName;

currentOptions = [self secondAppOptions];

currentNamespace = FIRNamespaceGoogleMobilePlatform;

break;

case RCNTestRCInstanceDefault:

default:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsFIRNamespace;

break;

}

NSString *fullyQualifiedNamespace =

[NSString stringWithFormat:@"%@:%@", currentNamespace, currentAppName];

RCNUserDefaultsManager *userDefaultsManager =

[[RCNUserDefaultsManager alloc] initWithAppName:currentAppName

bundleID:[NSBundle mainBundle].bundleIdentifier

namespace:fullyQualifiedNamespace];

userDefaultsManager.lastFetchTime = 0;

FIRRemoteConfig *config =

OCMPartialMock([[FIRRemoteConfig alloc] initWithAppName:currentAppName

FIROptions:currentOptions

namespace:currentNamespace

DBManager:_DBManager

configContent:configContent

analytics:nil]);

_configInstances[i] = config;

RCNConfigSettings *settings =

[[RCNConfigSettings alloc] initWithDatabaseManager:_DBManager

namespace:fullyQualifiedNamespace

firebaseAppName:currentAppName

googleAppID:currentOptions.googleAppID];

dispatch_queue_t queue = dispatch_queue_create(

[[NSString stringWithFormat:@"testqueue: %d", i] cStringUsingEncoding:NSUTF8StringEncoding],

DISPATCH_QUEUE_SERIAL);

_configFetch[i] = OCMPartialMock([[RCNConfigFetch alloc] initWithContent:configContent

DBManager:_DBManager

settings:settings

analytics:nil

experiment:nil

queue:queue

namespace:fullyQualifiedNamespace

options:currentOptions]);

OCMStub([_configFetch[i] fetchConfigWithExpirationDuration:43200 completionHandler:OCMOCK_ANY])

.andDo(^(NSInvocation *invocation) {

__unsafe_unretained void (^handler)(FIRRemoteConfigFetchStatus status,

NSError *_Nullable error) = nil;

// void (^handler)(FIRRemoteConfigFetchCompletion);

[invocation getArgument:&handler atIndex:3];

[self->_configFetch[i] fetchWithUserProperties:[[NSDictionary alloc] init]

completionHandler:handler];

});

_response[i] = @{};

_responseData[i] = [NSJSONSerialization dataWithJSONObject:_response[i] options:0 error:nil];

_URLResponse[i] =

[[NSHTTPURLResponse alloc] initWithURL:[NSURL URLWithString:@"https://firebase.com"]

statusCode:500

HTTPVersion:nil

headerFields:@{@"etag" : @"etag1"}];

id completionBlock =

[OCMArg invokeBlockWithArgs:_responseData[i], _URLResponse[i], [NSNull null], nil];

OCMExpect([_configFetch[i] URLSessionDataTaskWithContent:[OCMArg any]

completionHandler:completionBlock])

.andReturn(nil);

[_configInstances[i] updateWithNewInstancesForConfigFetch:_configFetch[i]

configContent:configContent

configSettings:settings

configExperiment:nil];

}

// Make the fetch calls for all instances.

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Test enumerating configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusFailure);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertFalse(changed);

XCTAssertNil(error);

FIRRemoteConfigValue *value = self->_configInstances[RCNTestRCInstanceDefault][@"key1"];

XCTAssertEqual((int)value.source, (int)FIRRemoteConfigSourceStatic);

XCTAssertEqualObjects(value.stringValue, @"");

XCTAssertEqual(value.boolValue, NO);

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

// TODO(mandard): Break up test with helper methods.

- (void)testFetchConfigsFailedErrorNoNetwork {

// Override the setup values to return back an error status.

RCNConfigContent *configContent = [[RCNConfigContent alloc] initWithDBManager:_DBManager];

// Populate the default, second app, second namespace instances.

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

NSString *currentAppName = nil;

FIROptions *currentOptions = nil;

NSString *currentNamespace = nil;

switch (i) {

case RCNTestRCInstanceSecondNamespace:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsPerfNamespace;

break;

case RCNTestRCInstanceSecondApp:

currentAppName = RCNTestsSecondFIRAppName;

currentOptions = [self secondAppOptions];

currentNamespace = FIRNamespaceGoogleMobilePlatform;

break;

case RCNTestRCInstanceDefault:

default:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsFIRNamespace;

break;

}

NSString *fullyQualifiedNamespace =

[NSString stringWithFormat:@"%@:%@", currentNamespace, currentAppName];

RCNUserDefaultsManager *userDefaultsManager =

[[RCNUserDefaultsManager alloc] initWithAppName:currentAppName

bundleID:[NSBundle mainBundle].bundleIdentifier

namespace:fullyQualifiedNamespace];

userDefaultsManager.lastFetchTime = 0;

FIRRemoteConfig *config =

OCMPartialMock([[FIRRemoteConfig alloc] initWithAppName:currentAppName

FIROptions:currentOptions

namespace:currentNamespace

DBManager:_DBManager

configContent:configContent

analytics:nil]);

_configInstances[i] = config;

RCNConfigSettings *settings =

[[RCNConfigSettings alloc] initWithDatabaseManager:_DBManager

namespace:fullyQualifiedNamespace

firebaseAppName:currentAppName

googleAppID:currentOptions.googleAppID];

dispatch_queue_t queue = dispatch_queue_create(

[[NSString stringWithFormat:@"testqueue: %d", i] cStringUsingEncoding:NSUTF8StringEncoding],

DISPATCH_QUEUE_SERIAL);

_configFetch[i] = OCMPartialMock([[RCNConfigFetch alloc] initWithContent:configContent

DBManager:_DBManager

settings:settings

analytics:nil

experiment:nil

queue:queue

namespace:fullyQualifiedNamespace

options:currentOptions]);

OCMStub([_configFetch[i] fetchConfigWithExpirationDuration:43200 completionHandler:OCMOCK_ANY])

.andDo(^(NSInvocation *invocation) {

__unsafe_unretained void (^handler)(FIRRemoteConfigFetchStatus status,

NSError *_Nullable error) = nil;

// void (^handler)(FIRRemoteConfigFetchCompletion);

[invocation getArgument:&handler atIndex:3];

[self->_configFetch[i] fetchWithUserProperties:[[NSDictionary alloc] init]

completionHandler:handler];

});

_response[i] = @{};

_responseData[i] = [NSJSONSerialization dataWithJSONObject:_response[i] options:0 error:nil];

// A no network error is accompanied with an HTTP status code of 0.

_URLResponse[i] =

[[NSHTTPURLResponse alloc] initWithURL:[NSURL URLWithString:@"https://firebase.com"]

statusCode:0

HTTPVersion:nil

headerFields:@{@"etag" : @"etag1"}];

id completionBlock =

[OCMArg invokeBlockWithArgs:_responseData[i], _URLResponse[i], [NSNull null], nil];

OCMExpect([_configFetch[i] URLSessionDataTaskWithContent:[OCMArg any]

completionHandler:completionBlock])

.andReturn(nil);

[_configInstances[i] updateWithNewInstancesForConfigFetch:_configFetch[i]

configContent:configContent

configSettings:settings

configExperiment:nil];

}

// Make the fetch calls for all instances.

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Test enumerating configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusFailure);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertFalse(changed);

XCTAssertNil(error);

// No such key, still return a static value.

FIRRemoteConfigValue *value = self->_configInstances[RCNTestRCInstanceDefault][@"key1"];

XCTAssertEqual((int)value.source, (int)FIRRemoteConfigSourceStatic);

XCTAssertEqualObjects(value.stringValue, @"");

XCTAssertEqual(value.boolValue, NO);

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

// Activate should return false if a fetch response returns 200 with NO_CHANGE as the response body.

- (void)testActivateOnFetchNoChangeStatus {

// Override the setup values to return back an error status.

RCNConfigContent *configContent = [[RCNConfigContent alloc] initWithDBManager:_DBManager];

// Populate the default, second app, second namespace instances.

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

NSString *currentAppName = nil;

FIROptions *currentOptions = nil;

NSString *currentNamespace = nil;

switch (i) {

case RCNTestRCInstanceSecondNamespace:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsPerfNamespace;

break;

case RCNTestRCInstanceSecondApp:

currentAppName = RCNTestsSecondFIRAppName;

currentOptions = [self secondAppOptions];

currentNamespace = FIRNamespaceGoogleMobilePlatform;

break;

case RCNTestRCInstanceDefault:

default:

currentAppName = RCNTestsDefaultFIRAppName;

currentOptions = [self firstAppOptions];

currentNamespace = RCNTestsFIRNamespace;

break;

}

NSString *fullyQualifiedNamespace =

[NSString stringWithFormat:@"%@:%@", currentNamespace, currentAppName];

RCNUserDefaultsManager *userDefaultsManager =

[[RCNUserDefaultsManager alloc] initWithAppName:currentAppName

bundleID:[NSBundle mainBundle].bundleIdentifier

namespace:fullyQualifiedNamespace];

userDefaultsManager.lastFetchTime = 10;

FIRRemoteConfig *config =

OCMPartialMock([[FIRRemoteConfig alloc] initWithAppName:currentAppName

FIROptions:currentOptions

namespace:currentNamespace

DBManager:_DBManager

configContent:configContent

analytics:nil]);

_configInstances[i] = config;

RCNConfigSettings *settings =

[[RCNConfigSettings alloc] initWithDatabaseManager:_DBManager

namespace:fullyQualifiedNamespace

firebaseAppName:currentAppName

googleAppID:currentOptions.googleAppID];

// Start the test with the assumption that we have some data that was fetched and activated.

settings.lastETag = @"etag1";

settings.lastETagUpdateTime = 100;

settings.lastApplyTimeInterval = 101;

dispatch_queue_t queue =

dispatch_queue_create([[NSString stringWithFormat:@"testNoStatusFetchQueue: %d", i]

cStringUsingEncoding:NSUTF8StringEncoding],

DISPATCH_QUEUE_SERIAL);

_configFetch[i] = OCMPartialMock([[RCNConfigFetch alloc] initWithContent:configContent

DBManager:_DBManager

settings:settings

analytics:nil

experiment:nil

queue:queue

namespace:fullyQualifiedNamespace

options:currentOptions]);

OCMStub([_configFetch[i] fetchConfigWithExpirationDuration:43200 completionHandler:OCMOCK_ANY])

.andDo(^(NSInvocation *invocation) {

__unsafe_unretained void (^handler)(FIRRemoteConfigFetchStatus status,

NSError *_Nullable error) = nil;

[invocation getArgument:&handler atIndex:3];

[self->_configFetch[i] fetchWithUserProperties:[[NSDictionary alloc] init]

completionHandler:handler];

});

_response[i] = @{@"state" : @"NO_CHANGE"};

_responseData[i] = [NSJSONSerialization dataWithJSONObject:_response[i] options:0 error:nil];

_URLResponse[i] =

[[NSHTTPURLResponse alloc] initWithURL:[NSURL URLWithString:@"https://firebase.com"]

statusCode:200

HTTPVersion:nil

headerFields:@{@"etag" : @"etag1"}];

id completionBlock =

[OCMArg invokeBlockWithArgs:_responseData[i], _URLResponse[i], [NSNull null], nil];

OCMExpect([_configFetch[i] URLSessionDataTaskWithContent:[OCMArg any]

completionHandler:completionBlock])

.andReturn(nil);

[_configInstances[i] updateWithNewInstancesForConfigFetch:_configFetch[i]

configContent:configContent

configSettings:settings

configExperiment:nil];

}

// Make the fetch calls for all instances.

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Test enumerating configs successfully - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

// Make sure activate returns false in fetch completion.

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertFalse(changed);

XCTAssertNil(error);

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testConfigValueForKey {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] =

[self expectationWithDescription:

[NSString stringWithFormat:@"Test configValueForKey: method - instance %d", i]];

XCTAssertEqual(_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusNoFetchYet);

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(status, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertTrue(changed);

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *key3 = [NSString stringWithFormat:@"key3-%d", i];

NSString *key7 = [NSString stringWithFormat:@"key7-%d", i];

NSString *value1 = [NSString stringWithFormat:@"value1-%d", i];

NSString *value2 = [NSString stringWithFormat:@"value2-%d", i];

NSString *value3 = [NSString stringWithFormat:@"value3-%d", i];

NSString *value7 = [NSString stringWithFormat:@"value7-%d", i];

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, value1);

XCTAssertEqualObjects(self->_configInstances[i][key2].stringValue, value2);

OCMVerify([self->_configInstances[i] objectForKeyedSubscript:key1]);

XCTAssertEqualObjects([self->_configInstances[i] configValueForKey:key3].stringValue,

value3);

if (i == RCNTestRCInstanceDefault) {

XCTAssertEqualObjects([self->_configInstances[i] configValueForKey:key7].stringValue,

value7);

}

XCTAssertEqualObjects([self->_configInstances[i] configValueForKey:key7].stringValue,

value7);

XCTAssertNotNil([self->_configInstances[i] configValueForKey:nil]);

XCTAssertEqual([self->_configInstances[i] configValueForKey:nil].source,

FIRRemoteConfigSourceStatic);

XCTAssertEqual([self->_configInstances[i] configValueForKey:nil].source,

FIRRemoteConfigSourceStatic);

XCTAssertEqual([self->_configInstances[i] configValueForKey:nil source:-1].source,

FIRRemoteConfigSourceStatic);

[expectations[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testFetchConfigWithDefaultSets {

NSMutableArray<XCTestExpectation *> *fetchConfigsExpectation =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

fetchConfigsExpectation[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Test fetch configs with defaults set - instance %d", i]];

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *key0 = [NSString stringWithFormat:@"key0-%d", i];

NSString *value1 = [NSString stringWithFormat:@"value1-%d", i];

NSString *value2 = [NSString stringWithFormat:@"value2-%d", i];

NSDictionary<NSString *, NSString *> *defaults = @{key1 : @"default key1", key0 : @"value0-0"};

[_configInstances[i] setDefaults:defaults];

FIRRemoteConfigFetchCompletion fetchCompletion = ^void(FIRRemoteConfigFetchStatus status,

NSError *error) {

XCTAssertEqual(self->_configInstances[i].lastFetchStatus, FIRRemoteConfigFetchStatusSuccess);

XCTAssertNil(error);

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, @"default key1");

XCTAssertEqual(self->_configInstances[i][key1].source, FIRRemoteConfigSourceDefault);

[self->_configInstances[i] activateWithCompletion:^(BOOL changed, NSError *_Nullable error) {

XCTAssertTrue(changed);

XCTAssertEqualObjects(self->_configInstances[i][key1].stringValue, value1);

XCTAssertEqual(self->_configInstances[i][key1].source, FIRRemoteConfigSourceRemote);

XCTAssertEqualObjects([self->_configInstances[i] defaultValueForKey:key1].stringValue,

@"default key1");

XCTAssertEqualObjects(self->_configInstances[i][key2].stringValue, value2);

XCTAssertEqualObjects(self->_configInstances[i][key0].stringValue, @"value0-0");

XCTAssertNil([self->_configInstances[i] defaultValueForKey:nil]);

OCMVerify([self->_configInstances[i] objectForKeyedSubscript:key1]);

XCTAssertEqual(status, FIRRemoteConfigFetchStatusSuccess,

@"Callback of first successful config "

@"fetch. Status must equal to FIRRemoteConfigFetchStatusSuccess.");

[fetchConfigsExpectation[i] fulfill];

}];

};

[_configInstances[i] fetchWithExpirationDuration:43200 completionHandler:fetchCompletion];

}

[self waitForExpectationsWithTimeout:_expectationTimeout

handler:^(NSError *error) {

XCTAssertNil(error);

}];

}

- (void)testDefaultsSetsOnly {

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

NSString *key1 = [NSString stringWithFormat:@"key1-%d", i];

NSString *key2 = [NSString stringWithFormat:@"key2-%d", i];

NSString *key3 = [NSString stringWithFormat:@"key3-%d", i];

NSString *key4 = [NSString stringWithFormat:@"key4-%d", i];

NSString *key5 = [NSString stringWithFormat:@"key5-%d", i];

NSString *defaultValue1 = @"default value1";

NSData *defaultValue2 = [defaultValue1 dataUsingEncoding:NSUTF8StringEncoding];

NSNumber *defaultValue3 = [NSNumber numberWithFloat:3.1415926];

NSDate *defaultValue4 = [NSDate date];

BOOL defaultValue5 = NO;

NSMutableDictionary<NSString *, id> *defaults = [NSMutableDictionary dictionaryWithDictionary:@{

key1 : defaultValue1,

key2 : defaultValue2,

key3 : defaultValue3,

key4 : defaultValue4,

key5 : @(defaultValue5),

}];

[_configInstances[i] setDefaults:defaults];

if (i == RCNTestRCInstanceSecondNamespace) {

[defaults setObject:@"2860" forKey:@"experience"];

[_configInstances[i] setDefaults:defaults];

}

// Remove objects right away to make sure dispatch_async gets the copy.

[defaults removeAllObjects];

FIRRemoteConfig *config = _configInstances[i];

XCTAssertEqualObjects(config[key1].stringValue, defaultValue1, @"Should support string format");

XCTAssertEqualObjects(config[key2].dataValue, defaultValue2, @"Should support data format");

XCTAssertEqual(config[key3].numberValue.longValue, defaultValue3.longValue,

@"Should support number format");

NSDateFormatter *dateFormatter = [[NSDateFormatter alloc] init];

[dateFormatter setDateFormat:@"yyyy-MM-dd HH:mm:ss"];

NSString *strValueOfDate = [dateFormatter stringFromDate:(NSDate *)defaultValue4];

XCTAssertEqualObjects(

config[key4].stringValue, strValueOfDate,

@"Date format can be set as an input from plist, but output coming out of "

@"defaultConfig as string.");

XCTAssertEqual(config[key5].boolValue, defaultValue5, @"Should support bool format");

if (i == RCNTestRCInstanceSecondNamespace) {

XCTAssertEqualObjects(

[_configInstances[i] configValueForKey:@"experience"].stringValue, @"2860",

@"Only default config has the key, must equal to default config value.");

}

// Reset default sets

[_configInstances[i] setDefaults:nil];

XCTAssertEqual([_configInstances[i] allKeysFromSource:FIRRemoteConfigSourceDefault].count, 0);

}

}

- (void)testSetDefaultsWithNilParams {

NSMutableArray<XCTestExpectation *> *expectations =

[[NSMutableArray alloc] initWithCapacity:RCNTestRCNumTotalInstances];

for (int i = 0; i < RCNTestRCNumTotalInstances; i++) {

expectations[i] = [self

expectationWithDescription:

[NSString stringWithFormat:@"Set defaults no callback expectation - instance %d", i]];

// Should work when passing nil.

[_configInstances[i] setDefaults:nil];

[_configInstances[i] setDefaults:nil];

dispatch_after(

dispatch_time(DISPATCH_TIME_NOW, (int64_t)(_checkCompletionTimeout * NSEC_PER_SEC)),

dispatch_get_main_queue(), ^{

XCTAssertEqual(

[self->_configInstances[i] allKeysFromSource:FIRRemoteConfigSourceDefault].count, 0);

[expectations[i] fulfill];