Files
firebase-ios-sdk/Example/Messaging/Tests/FIRMessagingTestUtilities.m
T
Maksym MalyhinandChen Liang f62c7ddcec IID: version dependent on FirebaseInstallations (#4533)
* Firebase Installations: Instance ID now uses FIS under the hood. (#3398)

* Fix merge errors

* Use modular import for FIRInstanceID_Private.h

* Use modular imports for FirebaseInstanceID.h (#3509)

* Changes missing during the fix.

* Changes missing during merge.

* Fix header import.

* ./scripts/style.sh

* Fix FCM tests. (#3888)

* Fix IID FCM token check.

* FirebaseInstallation: merge master into FIS integration branch (#4076)

* Fix IID FCM token check.

* Unused constant deleted.

* FCM tests: fix bad merge.

* Remote Config test app: use local IID and FIS. (#4089)

* token request should pass through if token is not fresh (#4104)

* Merge errors fix.

* IID with FIS: send FIS auth token with IID token request (#4152)

* FIRInstanceIDTokenOperation: add FIS auth token to the IID token request.

* IID tests: don't use FIS private headers.

* IID: indication of compatibility with FIS (#4262)

* FIS: IID incompatible version exception.

* FirebaseInstallations.podspec: set preprocessor flags based on env vars.

* FIS Travis: set required env vars

* /scripts/style.sh

* IID: method to indicate FIS usege added.

* FIS: FIRInstantiationTimingEagerInDefaultApp

* FIS: FIRInstantiationTimingAlwaysEager

* IID: make compatibility method static.

* Make -[FIRInstanceID appInstanceID:] safe to be called from any thread (#4482)

* IID: use a cached value of FID for -[FIRInstanceID appInstanceID:]

* TODO comment

* TODO updated

* FirebaseInstallations: prepare IID with FIS dependency for release (#4522)

* Bump FIrebaseInstallations to 1.0.0

* Travis: remove mm/fis-integration-master branch

* Travis: if_changed script updated to reflect IID dependency on FIS.

* Revert "Travis: remove mm/fis-integration-master branch"

This reverts commit 534dff990a.

* Travis: remove mm/fis-integration-master branch (#4526)

* Formatting and spelling.

* Firebase example proj: pod deintegrate

Co-authored-by: Chen Liang <chliang@google.com>
2019-12-20 10:54:11 -05:00

110 lines
3.5 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 <OCMock/OCMock.h>
#import "Example/Messaging/Tests/FIRMessagingTestUtilities.h"
#import <FirebaseAnalyticsInterop/FIRAnalyticsInterop.h>
#import <FirebaseInstanceID/FirebaseInstanceID.h>
#import <GoogleUtilities/GULUserDefaults.h>
#import <FirebaseInstallations/FIRInstallations.h>
#import "Firebase/Messaging/FIRMessagingPubSub.h"
#import "Firebase/Messaging/FIRMessagingRmqManager.h"
NS_ASSUME_NONNULL_BEGIN
static NSString *const kFIRMessagingDefaultsTestDomain = @"com.messaging.tests";
@interface FIRInstanceID (ExposedForTest)
/// Private initializer to avoid singleton usage.
- (FIRInstanceID *)initPrivately;
/// Starts fetching and configuration of InstanceID. This is necessary after the `initPrivately`
/// call.
- (void)start;
@end
@interface FIRMessaging (ExposedForTest)
@property(nonatomic, readwrite, strong) FIRMessagingPubSub *pubsub;
@property(nonatomic, readwrite, strong) FIRMessagingRmqManager *rmq2Manager;
/// Surface internal initializer to avoid singleton usage during tests.
- (instancetype)initWithAnalytics:(nullable id<FIRAnalyticsInterop>)analytics
withInstanceID:(FIRInstanceID *)instanceID
withUserDefaults:(GULUserDefaults *)defaults;
/// Kicks off required calls for some messaging tests.
- (void)start;
- (void)setupRmqManager;
@end
@interface FIRMessagingRmqManager (ExposedForTest)
- (void)removeDatabase;
@end
@implementation FIRMessagingTestUtilities
- (instancetype)initWithUserDefaults:(GULUserDefaults *)userDefaults withRMQManager:(BOOL)withRMQManager {
self = [super init];
if (self) {
// `+[FIRInstallations installations]` supposed to be used on `-[FIRInstanceID start]` to get
// `FIRInstallations` default instance. Need to stub it before.
_mockInstallations = OCMClassMock([FIRInstallations class]);
OCMStub([self.mockInstallations installations]).andReturn(self.mockInstallations);
_instanceID = [[FIRInstanceID alloc] initPrivately];
[_instanceID start];
// Create the messaging instance and call `start`.
_messaging = [[FIRMessaging alloc] initWithAnalytics:nil
withInstanceID:_instanceID
withUserDefaults:userDefaults];
if (withRMQManager) {
[_messaging start];
}
_mockMessaging = OCMPartialMock(_messaging);
if (!withRMQManager) {
OCMStub([_mockMessaging setupRmqManager]).andDo(nil);
[(FIRMessaging *)_mockMessaging start];
}
_mockInstanceID = OCMPartialMock(_instanceID);
_mockPubsub = OCMPartialMock(_messaging.pubsub);
}
return self;
}
- (void)cleanupAfterTest {
[_messaging.rmq2Manager removeDatabase];
[_messaging.messagingUserDefaults removePersistentDomainForName:kFIRMessagingDefaultsTestDomain];
_messaging.shouldEstablishDirectChannel = NO;
[_mockPubsub stopMocking];
[_mockMessaging stopMocking];
[_mockInstanceID stopMocking];
}
@end
NS_ASSUME_NONNULL_END