firebase-ios-sdk/Crashlytics/UnitTests/Mocks/FIRMockInstallations.m

// Copyright 2020 Google LLC
//
// 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 "Crashlytics/UnitTests/Mocks/FIRMockInstallations.h"

#import <Foundation/Foundation.h>

@interface FIRMockInstallationsImpl : NSObject

@property(nonatomic, copy) NSString *installationID;
@property(nonatomic, strong) NSError *error;

// the init function is not public for the token result, use as a placeholder to mock the token
// completion block
@property(nonatomic, strong) FIRInstallationsAuthTokenResult *tokenResult;

@property(nonatomic) BOOL authTokenFinished;
@property(nonatomic) BOOL installationIDFinished;

@end

@implementation FIRMockInstallationsImpl

- (void)authTokenWithCompletion:(FIRInstallationsTokenHandler)completion {
  self.authTokenFinished = true;
  completion(self.tokenResult, self.error);
}

- (void)installationIDWithCompletion:(FIRInstallationsIDHandler)completion {
  self.installationIDFinished = true;
  completion(self.installationID, self.error);
}

@end

@interface FIRMockInstallations ()

@end

@implementation FIRMockInstallations

- (instancetype)initWithFID:(NSString *)installationID {
  FIRMockInstallationsImpl *mock = [[FIRMockInstallationsImpl alloc] init];
  mock.installationID = [installationID copy];
  mock.error = nil;
  mock.authTokenFinished = false;
  mock.installationIDFinished = false;
  self = (id)mock;
  return self;
}

- (instancetype)initWithError:(NSError *)error {
  FIRMockInstallationsImpl *mock = [[FIRMockInstallationsImpl alloc] init];
  mock.installationID = nil;
  mock.error = error;
  mock.authTokenFinished = false;
  mock.installationIDFinished = false;
  self = (id)mock;
  return self;
}

@end