firebase-ios-sdk/Example/Database/Tests/Helpers/FIRFakeApp.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 "FIRFakeApp.h"

#import "FIRAuthInteropFake.h"
#import "FIRComponentTestUtilities.h"
#import "FIRDatabaseComponent.h"

@interface FIRFakeOptions: NSObject
@property(nonatomic, readonly, copy) NSString *databaseURL;
- (instancetype)initWithURL:(NSString *)url;
@end

@implementation FIRFakeOptions
- (instancetype)initWithURL:(NSString *)url {
  self = [super init];
  if (self) {
    _databaseURL = url;
  }
  return self;
}
@end

@interface FIRDatabaseComponent (Internal)
- (instancetype)initWithApp:(FIRApp *)app;
@end

@interface FIRComponentContainer (TestInternal)
@property(nonatomic, strong) NSMutableDictionary<NSString *, FIRComponentCreationBlock> *components;
@end

@interface FIRComponentContainer (TestInternalImplementations)
- (instancetype)initWithApp:(FIRApp *)app
                 components:(NSDictionary<NSString *, FIRComponentCreationBlock> *)components;
@end

@implementation FIRComponentContainer (TestInternalImplementations)

- (instancetype)initWithApp:(FIRApp *)app
                 components:(NSDictionary<NSString *, FIRComponentCreationBlock> *)components {
  self = [self initWithApp:app registrants:[[NSMutableSet alloc] init]];
  if (self) {
    self.components = [components mutableCopy];
  }
  return self;
}
@end

@implementation FIRFakeApp

- (instancetype)initWithName:(NSString *)name URL:(NSString *)url {
  self = [super init];
  if (self) {
    _name = name;
    _options = [[FIRFakeOptions alloc] initWithURL:url];
    _Nullable id (^authBlock)(FIRComponentContainer *, BOOL *) =
        ^(FIRComponentContainer *container, BOOL *isCacheable) {
          return [[FIRAuthInteropFake alloc] initWithToken:nil userID:nil error:nil];
        };
    FIRComponentCreationBlock databaseBlock =
        ^id _Nullable(FIRComponentContainer *container, BOOL *isCacheable) {
          *isCacheable = YES;
          return [[FIRDatabaseComponent alloc] initWithApp:container.app];
        };
    NSDictionary<NSString *, FIRComponentCreationBlock> *components =
        @{NSStringFromProtocol(@protocol(FIRAuthInterop)) : authBlock,
          NSStringFromProtocol(@protocol(FIRDatabaseProvider)) : databaseBlock};
    _container = [[FIRComponentContainer alloc] initWithApp:(FIRApp *)self components:components];
  }
  return self;
}
@end