firebase-ios-sdk/Example/Database/Tests/Helpers/FDevice.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 "FDevice.h"
#import <XCTest/XCTest.h>

#import <FirebaseDatabase/FIRDatabaseReference.h>

#import "FIRDatabaseConfig_Private.h"
#import "FIRDatabaseReference_Private.h"
#import "FRepoManager.h"
#import "FTestHelpers.h"
#import "SenTest+FWaiter.h"

@interface FDevice () {
  FIRDatabaseConfig *config;
  NSString *url;
  BOOL isOnline;
  BOOL disposed;
}
@end

@implementation FDevice

- (id)initOnline {
  FIRDatabaseReference *ref = [FTestHelpers getRandomNode];
  return [self initOnlineWithUrl:[ref description]];
}

- (id)initOffline {
  FIRDatabaseReference *ref = [FTestHelpers getRandomNode];
  return [self initOfflineWithUrl:[ref description]];
}

- (id)initOnlineWithUrl:(NSString *)firebaseUrl {
  return [self initWithUrl:firebaseUrl andOnline:YES];
}

- (id)initOfflineWithUrl:(NSString *)firebaseUrl {
  return [self initWithUrl:firebaseUrl andOnline:NO];
}

static NSUInteger deviceId = 0;

- (id)initWithUrl:(NSString *)firebaseUrl andOnline:(BOOL)online {
  self = [super init];
  if (self) {
    config = [FTestHelpers
        configForName:[NSString stringWithFormat:@"device-%lu", (unsigned long)deviceId++]];
    config.persistenceEnabled = YES;
    url = firebaseUrl;
    isOnline = online;
  }
  return self;
}

- (void)dealloc {
  if (!self->disposed) {
    [NSException raise:NSInternalInconsistencyException format:@"Forgot to dispose device"];
  }
}

- (void)dispose {
  // TODO: clear persistence
  [FRepoManager disposeRepos:self->config];
  self->disposed = YES;
}

- (void)goOffline {
  isOnline = NO;
  [FRepoManager interrupt:config];
}

- (void)goOnline {
  isOnline = YES;
  [FRepoManager resume:config];
}

- (void)restartOnline {
  @autoreleasepool {
    [FRepoManager disposeRepos:config];
    isOnline = YES;
  }
}

- (void)restartOffline {
  @autoreleasepool {
    [FRepoManager disposeRepos:config];
    isOnline = NO;
  }
}

// Waits for us to connect and then does an extra round-trip to make sure all initial state
// restoration is completely done.
- (void)waitForIdleUsingWaiter:(XCTest *)waiter {
  [self do:^(FIRDatabaseReference *ref) {
    __block BOOL connected = NO;
    FIRDatabaseHandle handle =
        [[ref.root child:@".info/connected"] observeEventType:FIRDataEventTypeValue
                                                    withBlock:^(FIRDataSnapshot *snapshot) {
                                                      connected = [snapshot.value boolValue];
                                                    }];
    [waiter waitUntil:^BOOL {
      return connected;
    }];
    [ref.root removeObserverWithHandle:handle];

    // HACK: Do a deep setPriority (which we expect to fail because there's no data there) to do a
    // no-op roundtrip.
    __block BOOL done = NO;
    [[ref.root child:@"ENTOHTNUHOE/ONTEHNUHTOE"]
                setPriority:@"blah"
        withCompletionBlock:^(NSError *error, FIRDatabaseReference *ref) {
          done = YES;
        }];
    [waiter waitUntil:^BOOL {
      return done;
    }];
  }];
}

- (void)do:(void (^)(FIRDatabaseReference *))action {
  @autoreleasepool {
    FIRDatabaseReference *ref = [[[[FIRDatabaseReference alloc] initWithConfig:self->config]
        database] referenceFromURL:self->url];
    if (!isOnline) {
      [FRepoManager interrupt:config];
    }
    action(ref);
  }
}

@end