mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-10 22:45:59 +05:45
Maksym Malyhin
af6f4e512b
* FIRInstallationsIDControllerTests: test refreshed token is stored. * FIRInstallationsIDControllerTests: all required tests updated to expect refreshed token to be stored. * Tests: assert all FBLWaitForPromisesWithTimeout * FIRInstallationsIDController: saving refreshed token implemented. * FIRInstallationsIDController: duplicated code removed.
117 lines
3.3 KiB
Objective-C
117 lines
3.3 KiB
Objective-C
/*
|
|
* 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 <FirebaseInstanceID/FirebaseInstanceID.h>
|
|
#import "FBLPromise+Testing.h"
|
|
#import "FIRTestKeychain.h"
|
|
|
|
#import "FIRInstallationsIIDStore.h"
|
|
|
|
@interface FIRInstanceID (Tests)
|
|
+ (FIRInstanceID *)instanceIDForTests;
|
|
@end
|
|
|
|
@interface FIRInstallationsIIDStoreTests : XCTestCase
|
|
@property(nonatomic) FIRInstanceID *instanceID;
|
|
@property(nonatomic) FIRInstallationsIIDStore *IIDStore;
|
|
#if TARGET_OS_OSX
|
|
@property(nonatomic) FIRTestKeychain *privateKeychain;
|
|
#endif // TARGET_OSX
|
|
@end
|
|
|
|
@implementation FIRInstallationsIIDStoreTests
|
|
|
|
- (void)setUp {
|
|
self.instanceID = [FIRInstanceID instanceIDForTests];
|
|
self.IIDStore = [[FIRInstallationsIIDStore alloc] init];
|
|
|
|
#if TARGET_OS_OSX
|
|
self.privateKeychain = [[FIRTestKeychain alloc] init];
|
|
self.IIDStore.keychainRef = self.privateKeychain.testKeychainRef;
|
|
#endif // TARGET_OSX
|
|
}
|
|
|
|
- (void)tearDown {
|
|
self.instanceID = nil;
|
|
#if TARGET_OS_OSX
|
|
self.privateKeychain = nil;
|
|
#endif // TARGET_OSX
|
|
}
|
|
|
|
// TODO: Configure the tests to run on macOS without requesting the keychain password.
|
|
#if !TARGET_OS_OSX
|
|
- (void)testExistingIIDSuccess {
|
|
NSString *existingIID = [self readExistingIID];
|
|
|
|
FBLPromise<NSString *> *IIDPromise = [self.IIDStore existingIID];
|
|
|
|
XCTAssert(FBLWaitForPromisesWithTimeout(0.5));
|
|
|
|
XCTAssertNil(IIDPromise.error);
|
|
XCTAssertEqualObjects(IIDPromise.value, existingIID);
|
|
NSLog(@"Existing IID: %@", IIDPromise.value);
|
|
}
|
|
|
|
- (void)testDeleteExistingIID {
|
|
// 1. Generate IID.
|
|
NSString *existingIID1 = [self readExistingIID];
|
|
|
|
// 2. Delete IID.
|
|
FBLPromise<NSNull *> *deletePromise = [self.IIDStore deleteExistingIID];
|
|
XCTAssert(FBLWaitForPromisesWithTimeout(0.5));
|
|
|
|
XCTAssertNil(deletePromise.error);
|
|
XCTAssertTrue(deletePromise.isFulfilled);
|
|
|
|
// 3. Check there is no IID.
|
|
FBLPromise<NSString *> *IIDPromise = [self.IIDStore existingIID];
|
|
XCTAssert(FBLWaitForPromisesWithTimeout(0.5));
|
|
|
|
XCTAssertNotNil(IIDPromise.error);
|
|
XCTAssertTrue(IIDPromise.isRejected);
|
|
|
|
// 4. Re-instantiate IID instance to reset its in-memory cache.
|
|
self.instanceID = [FIRInstanceID instanceIDForTests];
|
|
|
|
// 5. Generate a new IID and check it is different.
|
|
NSString *existingIID2 = [self readExistingIID];
|
|
XCTAssertNotEqualObjects(existingIID1, existingIID2);
|
|
}
|
|
|
|
#endif // !TARGET_OSX
|
|
|
|
#pragma mark - Helpers
|
|
|
|
- (NSString *)readExistingIID {
|
|
__block NSString *existingIID;
|
|
|
|
XCTestExpectation *IIDExpectation = [self expectationWithDescription:@"IIDExpectation"];
|
|
[self.instanceID getIDWithHandler:^(NSString *_Nullable identity, NSError *_Nullable error) {
|
|
XCTAssertNil(error);
|
|
XCTAssertNotNil(identity);
|
|
existingIID = identity;
|
|
[IIDExpectation fulfill];
|
|
}];
|
|
|
|
[self waitForExpectations:@[ IIDExpectation ] timeout:20];
|
|
|
|
return existingIID;
|
|
}
|
|
|
|
@end
|