firebase-ios-sdk/FirebaseCore/Tests/Unit/FIROptionsTest.m

// 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 "FirebaseCore/Tests/Unit/FIRTestCase.h"

#import "FirebaseCore/Extension/FIRAppInternal.h"
#import "FirebaseCore/Extension/FIROptionsInternal.h"
#import "FirebaseCore/Sources/FIRBundleUtil.h"
#import "FirebaseCore/Sources/Public/FirebaseCore/FIRVersion.h"
#import "SharedTestUtilities/FIROptionsMock.h"

extern NSString *const kFIRIsMeasurementEnabled;
extern NSString *const kFIRIsAnalyticsCollectionEnabled;
extern NSString *const kFIRIsAnalyticsCollectionDeactivated;
extern NSString *const kFIRLibraryVersionID;

@interface FIROptions (Test)

- (nullable NSDictionary *)analyticsOptionsDictionaryWithInfoDictionary:
    (nullable NSDictionary *)infoDictionary;

@end

@interface FIROptionsTest : FIRTestCase

@end

@implementation FIROptionsTest

- (void)setUp {
  [super setUp];
  [FIROptions resetDefaultOptions];
}

- (void)testInit {
  [FIROptionsMock mockFIROptions];
  NSDictionary *optionsDictionary = [FIROptions defaultOptionsDictionary];
  FIROptions *options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  [self assertOptionsMatchDefaults:options];
  XCTAssertNil(options.deepLinkURLScheme);
  XCTAssertTrue(options.usingOptionsFromDefaultPlist);

  options.deepLinkURLScheme = kDeepLinkURLScheme;
  XCTAssertEqualObjects(options.deepLinkURLScheme, kDeepLinkURLScheme);
}

- (void)testDefaultOptionsDictionaryWithNilFilePath {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMStub([mockBundleUtil optionsDictionaryPathWithResourceName:kJsonFileName
                                                    andFileType:kJsonFileType
                                                      inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn(nil);
  XCTAssertNil([FIROptions defaultOptionsDictionary]);
}

- (void)testDefaultOptionsDictionaryWithInvalidSourceFile {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMStub([mockBundleUtil optionsDictionaryPathWithResourceName:kJsonFileName
                                                    andFileType:kJsonFileType
                                                      inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn(@"invalid.plist");
  XCTAssertNil([FIROptions defaultOptionsDictionary]);
}

- (void)testDefaultOptions {
  [FIROptionsMock mockFIROptions];
  FIROptions *options = [FIROptions defaultOptions];
  [self assertOptionsMatchDefaults:options];
  XCTAssertNil(options.deepLinkURLScheme);
  XCTAssertTrue(options.usingOptionsFromDefaultPlist);

  options.deepLinkURLScheme = kDeepLinkURLScheme;
  XCTAssertEqualObjects(options.deepLinkURLScheme, kDeepLinkURLScheme);
}

#ifndef SWIFT_PACKAGE
// Another strategy is needed for these tests to pass with SWIFT_PACKAGE, since the mocking
// prevents the file path traversals.

- (void)testDefaultOptionsAreInitializedOncePlist {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMExpect([mockBundleUtil optionsDictionaryPathWithResourceName:kServiceInfoFileName
                                                      andFileType:kServiceInfoFileType
                                                        inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn([self validGoogleServicesInfoPlistPath]);
  XCTAssertNotNil([FIROptions defaultOptions]);
  OCMVerifyAll(mockBundleUtil);

  OCMReject([mockBundleUtil optionsDictionaryPathWithResourceName:OCMOCK_ANY
                                                      andFileType:OCMOCK_ANY
                                                        inBundles:OCMOCK_ANY]);
  XCTAssertNotNil([FIROptions defaultOptions]);
  OCMVerifyAll(mockBundleUtil);
}

- (void)testDefaultOptionsAreInitializedOnceJson {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMExpect([mockBundleUtil optionsDictionaryPathWithResourceName:kJsonFileName
                                                      andFileType:kJsonFileType
                                                        inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn([self validJsonConfigPath]);
  XCTAssertNotNil([FIROptions defaultOptions]);
  OCMVerifyAll(mockBundleUtil);

  OCMReject([mockBundleUtil optionsDictionaryPathWithResourceName:OCMOCK_ANY
                                                      andFileType:OCMOCK_ANY
                                                        inBundles:OCMOCK_ANY]);
  XCTAssertNotNil([FIROptions defaultOptions]);
  OCMVerifyAll(mockBundleUtil);
}

- (void)testDefaultOptionsDictionaryIsInitializedOncePlist {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMExpect([mockBundleUtil optionsDictionaryPathWithResourceName:kServiceInfoFileName
                                                      andFileType:kServiceInfoFileType
                                                        inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn([self validGoogleServicesInfoPlistPath]);
  XCTAssertNotNil([FIROptions defaultOptionsDictionary]);
  OCMVerifyAll(mockBundleUtil);

  OCMReject([mockBundleUtil optionsDictionaryPathWithResourceName:OCMOCK_ANY
                                                      andFileType:OCMOCK_ANY
                                                        inBundles:OCMOCK_ANY]);
  XCTAssertNotNil([FIROptions defaultOptionsDictionary]);
  OCMVerifyAll(mockBundleUtil);
}

- (void)testDefaultOptionsDictionaryIsInitializedOnceJson {
  id mockBundleUtil = OCMClassMock([FIRBundleUtil class]);
  OCMExpect([mockBundleUtil optionsDictionaryPathWithResourceName:kJsonFileName
                                                      andFileType:kJsonFileType
                                                        inBundles:[FIRBundleUtil relevantBundles]])
      .andReturn([self validJsonConfigPath]);
  XCTAssertNotNil([FIROptions defaultOptionsDictionary]);
  OCMVerifyAll(mockBundleUtil);

  OCMReject([mockBundleUtil optionsDictionaryPathWithResourceName:OCMOCK_ANY
                                                      andFileType:OCMOCK_ANY
                                                        inBundles:OCMOCK_ANY]);
  XCTAssertNotNil([FIROptions defaultOptionsDictionary]);
  OCMVerifyAll(mockBundleUtil);
}

- (void)testInitWithContentsOfPlistFile {
  NSString *filePath = [self validGoogleServicesInfoPlistPath];
  FIROptions *options = [[FIROptions alloc] initWithContentsOfFile:filePath];
  [self assertOptionsMatchDefaults:options];
  XCTAssertNil(options.deepLinkURLScheme);
  XCTAssertFalse(options.usingOptionsFromDefaultPlist);

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wnonnull"
  FIROptions *emptyOptions = [[FIROptions alloc] initWithContentsOfFile:nil];
#pragma clang diagnostic pop
  XCTAssertNil(emptyOptions);

  FIROptions *invalidOptions = [[FIROptions alloc] initWithContentsOfFile:@"invalid.plist"];
  XCTAssertNil(invalidOptions);
}

- (void)testInitWithContentsOfJsonFile {
  NSString *filePath = [self validJsonConfigPath];
  FIROptions *options = [[FIROptions alloc] initWithContentsOfFile:filePath];
  [self assertOptionsMatchDefaults:options];
  XCTAssertNil(options.deepLinkURLScheme);
  XCTAssertFalse(options.usingOptionsFromDefaultPlist);

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wnonnull"
  FIROptions *emptyOptions = [[FIROptions alloc] initWithContentsOfFile:nil];
#pragma clang diagnostic pop
  XCTAssertNil(emptyOptions);

  FIROptions *invalidOptions = [[FIROptions alloc] initWithContentsOfFile:@"invalid.json"];
  XCTAssertNil(invalidOptions);
}
#endif

- (void)testInitCustomizedOptionsPlist {
  FIROptions *options = [self oldAppOptions];
  options.APIKey = kAPIKey;
  options.bundleID = kBundleID;
  options.clientID = kClientID;
  options.databaseURL = kDatabaseURL;
  options.deepLinkURLScheme = kDeepLinkURLScheme;
  options.projectID = kProjectID;
  options.storageBucket = kStorageBucket;
  [self assertOptionsMatchDefaults:options];
  XCTAssertEqualObjects(options.deepLinkURLScheme, kDeepLinkURLScheme);
  XCTAssertFalse(options.usingOptionsFromDefaultPlist);
}

- (void)testInitCustomizedOptionsJson {
  FIROptions *options = [self appOptions];
  options.bundleID = kBundleID;
  options.clientID = kClientID;
  options.databaseURL = kDatabaseURL;
  options.deepLinkURLScheme = kDeepLinkURLScheme;
  options.storageBucket = kStorageBucket;
  [self assertOptionsMatchDefaults:options];
  XCTAssertEqualObjects(options.deepLinkURLScheme, kDeepLinkURLScheme);
  XCTAssertFalse(options.usingOptionsFromDefaultPlist);
}

- (void)assertOptionsMatchDefaults:(FIROptions *)options {
  XCTAssertEqualObjects(options.googleAppID, kGoogleAppID);
  XCTAssertEqualObjects(options.APIKey, kAPIKey);
  XCTAssertEqualObjects(options.clientID, kClientID);
  XCTAssertEqualObjects(options.GCMSenderID, kProjectNumber);
  XCTAssertEqualObjects(options.libraryVersionID, kFIRLibraryVersionID);
  XCTAssertEqualObjects(options.databaseURL, kDatabaseURL);
  XCTAssertEqualObjects(options.storageBucket, kStorageBucket);
  XCTAssertEqualObjects(options.bundleID, kBundleID);
  XCTAssertEqualObjects(options.projectID, kProjectID);
}

- (void)testCopyingPropertiesPlist {
  NSMutableString *mutableString;
  FIROptions *options = [self appOptions];
  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.APIKey = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.APIKey, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.bundleID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.bundleID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.clientID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.clientID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.GCMSenderID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.GCMSenderID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.projectID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.projectID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.googleAppID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.googleAppID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.databaseURL = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.databaseURL, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.deepLinkURLScheme = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.deepLinkURLScheme, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.storageBucket = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.storageBucket, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.appGroupID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.appGroupID, @"1");
}

- (void)testCopyingPropertiesJSON {
  NSMutableString *mutableString;
  FIROptions *options = [self appOptions];
  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.APIKey = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.APIKey, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.bundleID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.bundleID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.clientID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.clientID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.GCMSenderID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.GCMSenderID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.projectID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.projectID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.googleAppID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.googleAppID, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.databaseURL = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.databaseURL, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.deepLinkURLScheme = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.deepLinkURLScheme, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.storageBucket = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.storageBucket, @"1");

  mutableString = [[NSMutableString alloc] initWithString:@"1"];
  options.appGroupID = mutableString;
  [mutableString appendString:@"2"];
  XCTAssertEqualObjects(options.appGroupID, @"1");
}

- (void)testCopyWithZone {
  [FIROptionsMock mockFIROptions];
  // default options
  FIROptions *options = [FIROptions defaultOptions];
  options.deepLinkURLScheme = kDeepLinkURLScheme;
  XCTAssertEqualObjects(options.deepLinkURLScheme, kDeepLinkURLScheme);

  FIROptions *newOptions = [options copy];
  XCTAssertEqualObjects(newOptions.deepLinkURLScheme, kDeepLinkURLScheme);

  [options setDeepLinkURLScheme:kNewDeepLinkURLScheme];
  XCTAssertEqualObjects(options.deepLinkURLScheme, kNewDeepLinkURLScheme);
  XCTAssertEqualObjects(newOptions.deepLinkURLScheme, kDeepLinkURLScheme);

  // customized options with plist
  FIROptions *customizedOptions = [self oldAppOptions];
  customizedOptions.deepLinkURLScheme = kDeepLinkURLScheme;
  FIROptions *copyCustomizedOptions = [customizedOptions copy];
  [copyCustomizedOptions setDeepLinkURLScheme:kNewDeepLinkURLScheme];
  XCTAssertEqualObjects(customizedOptions.deepLinkURLScheme, kDeepLinkURLScheme);
  XCTAssertEqualObjects(copyCustomizedOptions.deepLinkURLScheme, kNewDeepLinkURLScheme);

  // customized options with json
  FIROptions *customizedOptionsJSON = [self appOptions];
  customizedOptionsJSON.deepLinkURLScheme = kDeepLinkURLScheme;
  FIROptions *copyCustomizedOptionsJSON = [customizedOptionsJSON copy];
  [copyCustomizedOptionsJSON setDeepLinkURLScheme:kNewDeepLinkURLScheme];
  XCTAssertEqualObjects(customizedOptionsJSON.deepLinkURLScheme, kDeepLinkURLScheme);
  XCTAssertEqualObjects(copyCustomizedOptionsJSON.deepLinkURLScheme, kNewDeepLinkURLScheme);
}

- (void)testAnalyticsConstants {
  // The keys are public values and should never change.
  XCTAssertEqualObjects(kFIRIsMeasurementEnabled, @"IS_MEASUREMENT_ENABLED");
  XCTAssertEqualObjects(kFIRIsAnalyticsCollectionEnabled, @"FIREBASE_ANALYTICS_COLLECTION_ENABLED");
  XCTAssertEqualObjects(kFIRIsAnalyticsCollectionDeactivated,
                        @"FIREBASE_ANALYTICS_COLLECTION_DEACTIVATED");
}

- (void)testAnalyticsOptions {
  // No keys anywhere.
  NSDictionary *optionsDictionary = nil;
  FIROptions *options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  NSDictionary *mainDictionary = nil;
  NSDictionary *expectedAnalyticsOptions = @{};
  NSDictionary *analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:nil];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  optionsDictionary = @{};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{};
  expectedAnalyticsOptions = @{};
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  // Main has no keys.
  optionsDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{};
  expectedAnalyticsOptions = optionsDictionary;
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  // Main overrides all the keys.
  optionsDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @NO,
    kFIRIsAnalyticsCollectionEnabled : @NO,
    kFIRIsMeasurementEnabled : @NO
  };
  expectedAnalyticsOptions = mainDictionary;
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  // Keys exist only in main.
  optionsDictionary = @{};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  expectedAnalyticsOptions = mainDictionary;
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  // Main overrides single keys.
  optionsDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{kFIRIsAnalyticsCollectionDeactivated : @NO};
  expectedAnalyticsOptions = @{
    kFIRIsAnalyticsCollectionDeactivated : @NO,  // override
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  optionsDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{kFIRIsAnalyticsCollectionEnabled : @NO};
  expectedAnalyticsOptions = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @NO,  // override
    kFIRIsMeasurementEnabled : @YES
  };
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

  optionsDictionary = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @YES
  };
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  mainDictionary = @{kFIRIsMeasurementEnabled : @NO};
  expectedAnalyticsOptions = @{
    kFIRIsAnalyticsCollectionDeactivated : @YES,
    kFIRIsAnalyticsCollectionEnabled : @YES,
    kFIRIsMeasurementEnabled : @NO  // override
  };
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
  XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);
}

- (void)testAnalyticsOptions_combinatorial {
  // Complete combinatorial test.

  // Possible values for the flags in the plist, where NSNull means the flag is not present.
  NSArray *values = @[ [NSNull null], @NO, @YES ];

  // Sanity checks for the combination generation.
  int combinationCount = 0;
  NSMutableArray *uniqueMainCombinations = [[NSMutableArray alloc] init];
  NSMutableArray *uniqueOptionsCombinations = [[NSMutableArray alloc] init];

  // Generate all optout flag combinations for { main plist X GoogleService-info options plist }.
  // Options present in the main plist should override options of the same key in the service plist.
  for (id mainDeactivated in values) {
    for (id mainAnalyticsEnabled in values) {
      for (id mainMeasurementEnabled in values) {
        for (id optionsDeactivated in values) {
          for (id optionsAnalyticsEnabled in values) {
            for (id optionsMeasurementEnabled in values) {
              @autoreleasepool {
                // Fill the GoogleService-info options plist dictionary.
                NSMutableDictionary *optionsDictionary = [[NSMutableDictionary alloc] init];
                if (![optionsDeactivated isEqual:[NSNull null]]) {
                  optionsDictionary[kFIRIsAnalyticsCollectionDeactivated] = optionsDeactivated;
                }
                if (![optionsAnalyticsEnabled isEqual:[NSNull null]]) {
                  optionsDictionary[kFIRIsAnalyticsCollectionEnabled] = optionsAnalyticsEnabled;
                }
                if (![optionsMeasurementEnabled isEqual:[NSNull null]]) {
                  optionsDictionary[kFIRIsMeasurementEnabled] = optionsMeasurementEnabled;
                }

                FIROptions *options =
                    [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
                if (![uniqueOptionsCombinations containsObject:optionsDictionary]) {
                  [uniqueOptionsCombinations addObject:optionsDictionary];
                }

                // Fill the main plist dictionary.
                NSMutableDictionary *mainDictionary = [[NSMutableDictionary alloc] init];
                if (![mainDeactivated isEqual:[NSNull null]]) {
                  mainDictionary[kFIRIsAnalyticsCollectionDeactivated] = mainDeactivated;
                }
                if (![mainAnalyticsEnabled isEqual:[NSNull null]]) {
                  mainDictionary[kFIRIsAnalyticsCollectionEnabled] = mainAnalyticsEnabled;
                }
                if (![mainMeasurementEnabled isEqual:[NSNull null]]) {
                  mainDictionary[kFIRIsMeasurementEnabled] = mainMeasurementEnabled;
                }

                // Add mainDictionary to uniqueMainCombinations if it isn't included yet.
                if (![uniqueMainCombinations containsObject:mainDictionary]) {
                  [uniqueMainCombinations addObject:mainDictionary];
                }

                // Generate the expected options by adding main values on top of the service options
                // values. The main values will replace any existing options values with the same
                // key. This is a different way of combining the two sets of flags from the actual
                // implementation in FIROptions, with equivalent output.
                NSMutableDictionary *expectedAnalyticsOptions =
                    [[NSMutableDictionary alloc] initWithDictionary:optionsDictionary];
                [expectedAnalyticsOptions addEntriesFromDictionary:mainDictionary];

                NSDictionary *analyticsOptions =
                    [options analyticsOptionsDictionaryWithInfoDictionary:mainDictionary];
                XCTAssertEqualObjects(analyticsOptions, expectedAnalyticsOptions);

                combinationCount++;
              }
            }
          }
        }
      }
    }
  }

  // Verify the sanity checks.
  XCTAssertEqual(combinationCount, 729);  // = 3^6.

  XCTAssertEqual(uniqueOptionsCombinations.count, 27);
  int optionsSizeCount[4] = {0};
  for (NSDictionary *dictionary in uniqueOptionsCombinations) {
    optionsSizeCount[dictionary.count]++;
  }
  XCTAssertEqual(optionsSizeCount[0], 1);
  XCTAssertEqual(optionsSizeCount[1], 6);
  XCTAssertEqual(optionsSizeCount[2], 12);
  XCTAssertEqual(optionsSizeCount[3], 8);

  XCTAssertEqual(uniqueMainCombinations.count, 27);
  int mainSizeCount[4] = {0};
  for (NSDictionary *dictionary in uniqueMainCombinations) {
    mainSizeCount[dictionary.count]++;
  }
  XCTAssertEqual(mainSizeCount[0], 1);
  XCTAssertEqual(mainSizeCount[1], 6);
  XCTAssertEqual(mainSizeCount[2], 12);
  XCTAssertEqual(mainSizeCount[3], 8);
}

- (void)testAnalyticsCollectionGlobalSwitchEnabled {
  // Stub the default app, and set the global switch to YES.
  id appMock = OCMClassMock([FIRApp class]);
  OCMStub([appMock isDefaultAppConfigured]).andReturn(YES);
  OCMStub([appMock defaultApp]).andReturn(appMock);
  OCMStub([appMock isDataCollectionDefaultEnabled]).andReturn(YES);

  // With no other settings, Analytics collection should default to the app's flag.
  FIROptions *options = [[FIROptions alloc] initInternalWithOptionsDictionary:@{}];
  XCTAssertTrue(options.isAnalyticsCollectionEnabled);
  XCTAssertTrue(options.isMeasurementEnabled);

  [appMock stopMocking];
}

- (void)testAnalyticsCollectionGlobalSwitchDisabled {
  // Stub the default app, and set the global switch to NO.
  id appMock = OCMClassMock([FIRApp class]);
  OCMStub([appMock isDefaultAppConfigured]).andReturn(YES);
  OCMStub([appMock defaultApp]).andReturn(appMock);
  OCMStub([appMock isDataCollectionDefaultEnabled]).andReturn(NO);

  // With no other settings, Analytics collection should default to the app's flag.
  FIROptions *options = [[FIROptions alloc] initInternalWithOptionsDictionary:@{}];
  XCTAssertFalse(options.isAnalyticsCollectionEnabled);
  XCTAssertFalse(options.isMeasurementEnabled);

  [appMock stopMocking];
}

- (void)testAnalyticsCollectionGlobalSwitchOverrideToDisable {
  // Stub the default app, and set the global switch to YES.
  id appMock = OCMClassMock([FIRApp class]);
  OCMStub([appMock isDefaultAppConfigured]).andReturn(YES);
  OCMStub([appMock defaultApp]).andReturn(appMock);
  OCMStub([appMock isDataCollectionDefaultEnabled]).andReturn(YES);

  // Test the three Analytics flags that override to disable Analytics collection.
  FIROptions *collectionEnabledOptions = [[FIROptions alloc]
      initInternalWithOptionsDictionary:@{kFIRIsAnalyticsCollectionEnabled : @NO}];
  XCTAssertFalse(collectionEnabledOptions.isAnalyticsCollectionEnabled);

  FIROptions *collectionDeactivatedOptions = [[FIROptions alloc]
      initInternalWithOptionsDictionary:@{kFIRIsAnalyticsCollectionDeactivated : @YES}];
  XCTAssertFalse(collectionDeactivatedOptions.isAnalyticsCollectionEnabled);

  FIROptions *measurementEnabledOptions =
      [[FIROptions alloc] initInternalWithOptionsDictionary:@{kFIRIsMeasurementEnabled : @NO}];
  XCTAssertFalse(measurementEnabledOptions.isAnalyticsCollectionEnabled);
}

- (void)testAnalyticsCollectionGlobalSwitchOverrideToEnable {
  // Stub the default app, and set the global switch to YES.
  id appMock = OCMClassMock([FIRApp class]);
  OCMStub([appMock isDefaultAppConfigured]).andReturn(YES);
  OCMStub([appMock defaultApp]).andReturn(appMock);
  OCMStub([appMock isDataCollectionDefaultEnabled]).andReturn(NO);

  // Test the two Analytics flags that can override and enable collection.
  FIROptions *collectionEnabledOptions = [[FIROptions alloc]
      initInternalWithOptionsDictionary:@{kFIRIsAnalyticsCollectionEnabled : @YES}];
  XCTAssertTrue(collectionEnabledOptions.isAnalyticsCollectionEnabled);

  FIROptions *measurementEnabledOptions =
      [[FIROptions alloc] initInternalWithOptionsDictionary:@{kFIRIsMeasurementEnabled : @YES}];
  XCTAssertTrue(measurementEnabledOptions.isAnalyticsCollectionEnabled);
}

- (void)testAnalyticsCollectionExplicitlySet {
  NSDictionary *optionsDictionary = @{};
  FIROptions *options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  NSDictionary *analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:@{}];
  XCTAssertFalse([options isAnalyticsCollectionExplicitlySet]);

  // Test deactivation flag.
  optionsDictionary = @{kFIRIsAnalyticsCollectionDeactivated : @YES};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:@{}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);

  // If "deactivated" == NO, that doesn't mean it's explicitly set / enabled so it should be treated
  // as if it's not set.
  optionsDictionary = @{kFIRIsAnalyticsCollectionDeactivated : @NO};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:@{}];
  XCTAssertFalse([options isAnalyticsCollectionExplicitlySet]);

  // Test the collection enabled flag.
  optionsDictionary = @{kFIRIsAnalyticsCollectionEnabled : @YES};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:@{}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);

  optionsDictionary = @{kFIRIsAnalyticsCollectionEnabled : @NO};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  analyticsOptions = [options analyticsOptionsDictionaryWithInfoDictionary:@{}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);

  // Test the old measurement flag.
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:@{}];
  analyticsOptions =
      [options analyticsOptionsDictionaryWithInfoDictionary:@{kFIRIsMeasurementEnabled : @YES}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);

  options = [[FIROptions alloc] initInternalWithOptionsDictionary:@{}];
  analyticsOptions =
      [options analyticsOptionsDictionaryWithInfoDictionary:@{kFIRIsMeasurementEnabled : @NO}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);

  // For good measure, a combination of all 3 (even if they conflict).
  optionsDictionary =
      @{kFIRIsAnalyticsCollectionDeactivated : @YES, kFIRIsAnalyticsCollectionEnabled : @YES};
  options = [[FIROptions alloc] initInternalWithOptionsDictionary:optionsDictionary];
  analyticsOptions =
      [options analyticsOptionsDictionaryWithInfoDictionary:@{kFIRIsMeasurementEnabled : @NO}];
  XCTAssertTrue([options isAnalyticsCollectionExplicitlySet]);
}

- (void)testModifyingOptionsThrows {
  FIROptions *options = [self appOptions];
  options.editingLocked = YES;

  // Modification to every property should result in an exception.
  XCTAssertThrows(options.APIKey = @"should_throw");
  XCTAssertThrows(options.bundleID = @"should_throw");
  XCTAssertThrows(options.clientID = @"should_throw");
  XCTAssertThrows(options.databaseURL = @"should_throw");
  XCTAssertThrows(options.deepLinkURLScheme = @"should_throw");
  XCTAssertThrows(options.GCMSenderID = @"should_throw");
  XCTAssertThrows(options.googleAppID = @"should_throw");
  XCTAssertThrows(options.projectID = @"should_throw");
  XCTAssertThrows(options.storageBucket = @"should_throw");
}

- (void)testVersionFormat {
  // `kFIRLibraryVersionID` is `nil` until `libraryVersion` is called on `FIROptions`.
  FIROptions *options = [self appOptions];
  __unused NSString *libraryVersion = options.libraryVersionID;
  options = nil;

  NSRegularExpression *sLibraryVersionRegex =
      [NSRegularExpression regularExpressionWithPattern:@"^[0-9]{8,}$" options:0 error:NULL];
  NSUInteger numberOfMatches =
      [sLibraryVersionRegex numberOfMatchesInString:kFIRLibraryVersionID
                                            options:0
                                              range:NSMakeRange(0, kFIRLibraryVersionID.length)];
  XCTAssertEqual(numberOfMatches, 1, @"Incorrect library version format.");
}

- (void)testVersionConsistency {
  // `kFIRLibraryVersionID` is `nil` until `libraryVersion` is called on `FIROptions`.
  FIROptions *options = [self appOptions];
  __unused NSString *libraryVersion = options.libraryVersionID;
  options = nil;

  // Now `kFIRLibraryVersionID` is assigned, test that it is formatted correctly.
  const char *versionString = [kFIRLibraryVersionID UTF8String];
  int major = (versionString[0] - '0') * 10 + versionString[1] - '0';
  int minor = (versionString[2] - '0') * 10 + versionString[3] - '0';
  int patch = (versionString[4] - '0') * 10 + versionString[5] - '0';
  NSString *str = [NSString stringWithFormat:@"%d.%d.%d", major, minor, patch];
  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
}

// Repeat test with more Objective-C.
- (void)testVersionConsistency2 {
  // `kFIRLibraryVersionID` is `nil` until `libraryVersion` is called on `FIROptions`.
  FIROptions *options = [self appOptions];
  __unused NSString *libraryVersion = options.libraryVersionID;
  options = nil;

  NSRange major = NSMakeRange(0, 2);
  NSRange minor = NSMakeRange(2, 2);
  NSRange patch = NSMakeRange(4, 2);
  NSString *str = [NSString
      stringWithFormat:@"%d.%d.%d", [[kFIRLibraryVersionID substringWithRange:major] intValue],
                       [[kFIRLibraryVersionID substringWithRange:minor] intValue],
                       [[kFIRLibraryVersionID substringWithRange:patch] intValue]];
  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
}

#pragma mark - Helpers

- (NSString *)validGoogleServicesInfoPlistPath {
  NSString *filePath = [[NSBundle mainBundle] pathForResource:@"GoogleService-Info"
                                                       ofType:@"plist"];
  if (filePath == nil) {
    // Use bundleForClass to allow GoogleService-Info.plist to be in the test target's bundle.
    NSBundle *bundle = [NSBundle bundleForClass:[self class]];
    filePath = [bundle pathForResource:@"GoogleService-Info" ofType:@"plist"];
  }

  XCTAssertNotNil(filePath);
  return filePath;
}

- (NSString *)validJsonConfigPath {
  NSString *filePath = [[NSBundle mainBundle] pathForResource:@"firebase-sdk-config-apple"
                                                       ofType:@"json"];
  if (filePath == nil) {
    // Use bundleForClass to allow GoogleService-Info.plist to be in the test target's bundle.
    NSBundle *bundle = [NSBundle bundleForClass:[self class]];
    filePath = [bundle pathForResource:@"firebase-sdk-config-apple" ofType:@"json"];
  }

  XCTAssertNotNil(filePath);
  return filePath;
}

- (FIROptions *)appOptions {
  return [[FIROptions alloc] initWithAppID:kGoogleAppID
                             projectNumber:kProjectNumber
                                 projectID:kProjectID
                                    apiKey:kAPIKey];
}

- (FIROptions *)oldAppOptions {
#pragma clang diagnostic ignored "-Wdeprecated-declarations"
  return [[FIROptions alloc] initWithGoogleAppID:kGoogleAppID GCMSenderID:kProjectNumber];
#pragma clang diagnostic pop
}

@end