firebase-ios-sdk/Example/InstanceID/Tests/FIRInstanceIDCheckinPreferencesTest.m

/*
 * 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 "Firebase/InstanceID/FIRInstanceIDCheckinPreferences+Internal.h"
#import "Firebase/InstanceID/FIRInstanceIDCheckinPreferences_Private.h"
#import "Firebase/InstanceID/FIRInstanceIDCheckinService.h"

static NSString *const kDeviceAuthId = @"1234";
static NSString *const kSecretToken = @"567890";

@interface FIRInstanceIDCheckinPreferencesTest : XCTestCase

@end

@implementation FIRInstanceIDCheckinPreferencesTest

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

- (void)tearDown {
  [super tearDown];
}

- (void)testInvalidCheckinInfo {
  FIRInstanceIDCheckinPreferences *preferences =
      [[FIRInstanceIDCheckinPreferences alloc] initWithDeviceID:@"" secretToken:@""];
  XCTAssertFalse([preferences hasValidCheckinInfo]);
}

- (void)testCheckinPreferencesReset {
  FIRInstanceIDCheckinPreferences *checkin =
      [[FIRInstanceIDCheckinPreferences alloc] initWithDeviceID:kDeviceAuthId
                                                    secretToken:kSecretToken];
  [checkin reset];
  XCTAssertNil(checkin.deviceID);
  XCTAssertNil(checkin.secretToken);
  XCTAssertFalse([checkin hasValidCheckinInfo]);
}

- (void)testInvalidCheckinInfoDueToLocaleChanged {
  // Set to a different locale than the current locale.
  [[NSUserDefaults standardUserDefaults] setObject:@"zh-Hant"
                                            forKey:kFIRInstanceIDUserDefaultsKeyLocale];
  FIRInstanceIDCheckinPreferences *checkin =
      [[FIRInstanceIDCheckinPreferences alloc] initWithDeviceID:kDeviceAuthId
                                                    secretToken:kSecretToken];
  XCTAssertFalse([checkin hasValidCheckinInfo],
                 @"Should consider checkin info invalid as locale has changed.");
  // set back the original locale
  [[NSUserDefaults standardUserDefaults] setObject:FIRInstanceIDCurrentLocale()
                                            forKey:kFIRInstanceIDUserDefaultsKeyLocale];
}

- (void)testCheckinPreferenceRefreshTokenWeekly {
  FIRInstanceIDCheckinPreferences *checkin =
      [[FIRInstanceIDCheckinPreferences alloc] initWithDeviceID:kDeviceAuthId
                                                    secretToken:kSecretToken];
  int64_t now = FIRInstanceIDCurrentTimestampInMilliseconds();
  [checkin updateWithCheckinPlistContents:@{kFIRInstanceIDLastCheckinTimeKey : @(now)}];

  XCTAssertTrue([checkin hasValidCheckinInfo]);

  // Set last checkin time long time ago.
  now = FIRInstanceIDCurrentTimestampInMilliseconds();
  [checkin updateWithCheckinPlistContents:@{
    kFIRInstanceIDLastCheckinTimeKey : @(now - kFIRInstanceIDDefaultCheckinInterval * 1000 - 1)
  }];

  XCTAssertFalse([checkin hasValidCheckinInfo]);
}

@end