firebase-ios-sdk/FirebaseDatabase/Tests/Integration/FIRDatabaseQueryTests.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 "FirebaseDatabase/Tests/Integration/FIRDatabaseQueryTests.h"
#import "FirebaseCore/Sources/Public/FirebaseCore/FIROptions.h"
#import "FirebaseDatabase/Sources/Api/Private/FIRDatabaseQuery_Private.h"
#import "FirebaseDatabase/Sources/Constants/FConstants.h"
#import "FirebaseDatabase/Sources/Core/FQuerySpec.h"
#import "FirebaseDatabase/Sources/Utilities/FUtilities.h"
#import "FirebaseDatabase/Tests/Helpers/FIRFakeApp.h"
#import "FirebaseDatabase/Tests/Helpers/FTestExpectations.h"

@implementation FIRDatabaseQueryTests

- (void)testCanCreateBasicQueries {
  // Just make sure none of these throw anything

  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  [ref queryLimitedToFirst:10];
  [ref queryLimitedToLast:10];

  [[ref queryOrderedByKey] queryStartingAtValue:@"foo"];
  [[ref queryOrderedByKey] queryStartingAfterValue:@"foo"];
  [[ref queryOrderedByKey] queryEndingAtValue:@"foo"];
  [[ref queryOrderedByKey] queryEndingBeforeValue:@"foo"];
  [[ref queryOrderedByKey] queryEqualToValue:@"foo"];

  [[ref queryOrderedByChild:@"index"] queryStartingAtValue:@YES];
  [[ref queryOrderedByChild:@"index"] queryStartingAtValue:@1];
  [[ref queryOrderedByChild:@"index"] queryStartingAtValue:@"foo"];
  [[ref queryOrderedByChild:@"index"] queryStartingAtValue:nil];
  [[ref queryOrderedByChild:@"index"] queryStartingAfterValue:@YES];
  [[ref queryOrderedByChild:@"index"] queryStartingAfterValue:@1];
  [[ref queryOrderedByChild:@"index"] queryStartingAfterValue:@"foo"];
  [[ref queryOrderedByChild:@"index"] queryStartingAfterValue:nil];
  [[ref queryOrderedByChild:@"index"] queryEndingAtValue:@YES];
  [[ref queryOrderedByChild:@"index"] queryEndingAtValue:@1];
  [[ref queryOrderedByChild:@"index"] queryEndingAtValue:@"foo"];
  [[ref queryOrderedByChild:@"index"] queryEndingAtValue:nil];
  [[ref queryOrderedByChild:@"index"] queryEndingBeforeValue:@YES];
  [[ref queryOrderedByChild:@"index"] queryEndingBeforeValue:@1];
  [[ref queryOrderedByChild:@"index"] queryEndingBeforeValue:@"foo"];
  [[ref queryOrderedByChild:@"index"] queryEndingBeforeValue:nil];
  [[ref queryOrderedByChild:@"index"] queryEqualToValue:@YES];
  [[ref queryOrderedByChild:@"index"] queryEqualToValue:@1];
  [[ref queryOrderedByChild:@"index"] queryEqualToValue:@"foo"];
  [[ref queryOrderedByChild:@"index"] queryEqualToValue:nil];

  [[ref queryOrderedByPriority] queryStartingAtValue:@1];
  [[ref queryOrderedByPriority] queryStartingAtValue:@"foo"];
  [[ref queryOrderedByPriority] queryStartingAtValue:nil];
  [[ref queryOrderedByPriority] queryStartingAfterValue:@1];
  [[ref queryOrderedByPriority] queryStartingAfterValue:@"foo"];
  [[ref queryOrderedByPriority] queryStartingAfterValue:nil];
  [[ref queryOrderedByPriority] queryEndingAtValue:@1];
  [[ref queryOrderedByPriority] queryEndingAtValue:@"foo"];
  [[ref queryOrderedByPriority] queryEndingAtValue:nil];
  [[ref queryOrderedByPriority] queryEndingBeforeValue:@1];
  [[ref queryOrderedByPriority] queryEndingBeforeValue:@"foo"];
  [[ref queryOrderedByPriority] queryEndingBeforeValue:nil];
  [[ref queryOrderedByPriority] queryEqualToValue:@1];
  [[ref queryOrderedByPriority] queryEqualToValue:@"foo"];
  [[ref queryOrderedByPriority] queryEqualToValue:nil];
}

- (void)testInvalidQueryParams {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  XCTAssertThrows([[ref queryLimitedToFirst:100] queryLimitedToFirst:100]);
  XCTAssertThrows([[ref queryLimitedToFirst:100] queryLimitedToLast:100]);
  XCTAssertThrows([[ref queryLimitedToLast:100] queryLimitedToFirst:100]);
  XCTAssertThrows([[ref queryLimitedToLast:100] queryLimitedToLast:100]);
  XCTAssertThrows([[ref queryOrderedByPriority] queryOrderedByPriority]);
  XCTAssertThrows([[ref queryOrderedByPriority] queryOrderedByKey]);
  XCTAssertThrows([[ref queryOrderedByPriority] queryOrderedByChild:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByPriority] queryOrderedByValue]);
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByPriority]);
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByKey]);
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByChild:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByValue]);
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByPriority]);
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByChild:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByValue]);
  XCTAssertThrows([[ref queryOrderedByValue] queryOrderedByPriority]);
  XCTAssertThrows([[ref queryOrderedByValue] queryOrderedByKey]);
  XCTAssertThrows([[ref queryOrderedByValue] queryOrderedByChild:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByValue] queryOrderedByValue]);
  XCTAssertThrows([[ref queryStartingAtValue:@"foo"] queryStartingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryStartingAtValue:@"foo"] queryStartingAfterValue:@"foo"]);
  XCTAssertThrows([[ref queryStartingAtValue:@"foo"] queryEqualToValue:@"foo"]);
  XCTAssertThrows([[ref queryStartingAfterValue:@"foo"] queryStartingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryStartingAfterValue:@"foo"] queryEqualToValue:@"foo"]);
  XCTAssertThrows([[ref queryEndingAtValue:@"foo"] queryEndingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryEndingAtValue:@"foo"] queryEndingBeforeValue:@"foo"]);
  XCTAssertThrows([[ref queryEndingAtValue:@"foo"] queryEqualToValue:@"foo"]);
  XCTAssertThrows([[ref queryEndingBeforeValue:@"foo"] queryEndingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryEndingBeforeValue:@"foo"] queryEqualToValue:@"foo"]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo"] queryStartingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo"] queryStartingAfterValue:@"foo"]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo"] queryEndingAtValue:@"foo"]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo"] queryEndingBeforeValue:@"foo"]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo"] queryEqualToValue:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAtValue:@"foo" childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAfterValue:@"foo" childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingAtValue:@"foo" childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingBeforeValue:@"foo" childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEqualToValue:@"foo" childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAtValue:@1 childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAfterValue:@1 childKey:@"foo"]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAtValue:@YES]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAfterValue:@YES]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingAtValue:@1]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingAtValue:@YES]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingBeforeValue:@1]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingBeforeValue:@YES]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAtValue:nil]);
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAfterValue:nil]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingAtValue:nil]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingBeforeValue:nil]);
  XCTAssertThrows([[ref queryOrderedByKey] queryEqualToValue:nil]);
  XCTAssertThrows([[ref queryStartingAtValue:@"foo" childKey:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryStartingAfterValue:@"foo" childKey:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEndingAtValue:@"foo" childKey:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEndingBeforeValue:@"foo" childKey:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEqualToValue:@"foo" childKey:@"foo"] queryOrderedByKey]);
  XCTAssertThrows([[ref queryStartingAtValue:@1] queryOrderedByKey]);
  XCTAssertThrows([[ref queryStartingAfterValue:@1] queryOrderedByKey]);
  XCTAssertThrows([[ref queryStartingAtValue:@YES] queryOrderedByKey]);
  XCTAssertThrows([[ref queryStartingAfterValue:@YES] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEndingAtValue:@1] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEndingBeforeValue:@1] queryOrderedByKey]);
  XCTAssertThrows([[ref queryEndingBeforeValue:@YES] queryOrderedByKey]);
  XCTAssertThrows([ref queryStartingAtValue:@[]]);
  XCTAssertThrows([ref queryStartingAtValue:@{}]);
  XCTAssertThrows([ref queryStartingAfterValue:@[]]);
  XCTAssertThrows([ref queryStartingAfterValue:@{}]);
  XCTAssertThrows([ref queryEndingAtValue:@[]]);
  XCTAssertThrows([ref queryEndingAtValue:@{}]);
  XCTAssertThrows([ref queryEndingBeforeValue:@[]]);
  XCTAssertThrows([ref queryEndingBeforeValue:@{}]);
  XCTAssertThrows([ref queryEqualToValue:@[]]);
  XCTAssertThrows([ref queryEqualToValue:@{}]);

  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByPriority],
                  @"Cannot call orderBy multiple times");
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByPriority],
                  @"Cannot call orderBy multiple times");
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByKey],
                  @"Cannot call orderBy multiple times");
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByKey],
                  @"Cannot call orderBy multiple times");
  XCTAssertThrows([[ref queryOrderedByKey] queryOrderedByChild:@"foo"],
                  @"Cannot call orderBy multiple times");
  XCTAssertThrows([[ref queryOrderedByChild:@"foo"] queryOrderedByChild:@"foo"],
                  @"Cannot call orderBy multiple times");

  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAtValue:@"a" childKey:@"b"],
                  @"Cannot specify starting child name when ordering by key.");
  XCTAssertThrows([[ref queryOrderedByKey] queryStartingAfterValue:@"a" childKey:@"b"],
                  @"Cannot specify starting child name when ordering by key.");
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingAtValue:@"a" childKey:@"b"],
                  @"Cannot specify ending child name when ordering by key.");
  XCTAssertThrows([[ref queryOrderedByKey] queryEndingBeforeValue:@"a" childKey:@"b"],
                  @"Cannot specify ending child name when ordering by key.");
  XCTAssertThrows([[ref queryOrderedByKey] queryEqualToValue:@"a" childKey:@"b"],
                  @"Cannot specify equalTo child name when ordering by key.");

  XCTAssertThrows([[ref queryOrderedByPriority] queryStartingAtValue:@YES],
                  @"Can't pass booleans as start/end when using priority index.");
  XCTAssertThrows([[ref queryOrderedByPriority] queryStartingAfterValue:@YES],
                  @"Can't pass booleans as start/end when using priority index.");
  XCTAssertThrows([[ref queryOrderedByPriority] queryEndingAtValue:@NO],
                  @"Can't pass booleans as start/end when using priority index.");
  XCTAssertThrows([[ref queryOrderedByPriority] queryEndingBeforeValue:@NO],
                  @"Can't pass booleans as start/end when using priority index.");
  XCTAssertThrows([[ref queryOrderedByPriority] queryEqualToValue:@YES],
                  @"Can't pass booleans as start/end when using priority index.");
}

- (void)testLimitRanges {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  XCTAssertThrows([ref queryLimitedToLast:0], @"Can't pass zero as limit");
  XCTAssertThrows([ref queryLimitedToFirst:0], @"Can't pass zero as limit");
  XCTAssertThrows([ref queryLimitedToLast:0], @"Can't pass zero as limit");
  uint64_t MAX_ALLOWED_VALUE = (uint64_t)(1l << 31) - 1;
  [ref queryLimitedToFirst:(NSUInteger)MAX_ALLOWED_VALUE];
  [ref queryLimitedToLast:(NSUInteger)MAX_ALLOWED_VALUE];
  XCTAssertThrows([ref queryLimitedToFirst:(NSUInteger)(MAX_ALLOWED_VALUE + 1)],
                  @"Can't pass limits that don't fit into 32 bit signed integer range");
  XCTAssertThrows([ref queryLimitedToLast:(NSUInteger)(MAX_ALLOWED_VALUE + 1)],
                  @"Can't pass limits that don't fit into 32 bit signed integer range");
}

- (void)testInvalidKeys {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSArray* badKeys = @[
    @".test", @"test.", @"fo$o", @"[what", @"ever]", @"ha#sh", @"/thing", @"th/ing", @"thing/"
  ];

  for (NSString* badKey in badKeys) {
    XCTAssertThrows([[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:badKey],
                    @"Setting bad key");
    XCTAssertThrows([[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:badKey],
                    @"Setting bad key");
    XCTAssertThrows([[ref queryOrderedByPriority] queryEndingAtValue:nil childKey:badKey],
                    @"Setting bad key");
    XCTAssertThrows([[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:badKey],
                    @"Setting bad key");
  }
}

- (void)testOffCanBeCalledOnDefault {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL called = NO;
  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  if (called) {
                    XCTFail(@"Should not be called twice");
                  } else {
                    called = YES;
                  }
                }];

  [ref setValue:@{@"a" : @5, @"b" : @6}];

  [self waitUntil:^BOOL {
    return called;
  }];

  called = NO;

  [ref removeAllObservers];

  __block BOOL complete = NO;
  [ref setValue:@{@"a" : @6, @"b" : @7}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        complete = YES;
      }];

  [self waitUntil:^BOOL {
    return complete;
  }];

  XCTAssertFalse(called, @"Should not have been called again");
}

- (void)testOffCanBeCalledOnHandle {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL called = NO;
  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  FIRDatabaseHandle handle = [query observeEventType:FIRDataEventTypeValue
                                           withBlock:^(FIRDataSnapshot* snapshot) {
                                             if (called) {
                                               XCTFail(@"Should not be called twice");
                                             } else {
                                               called = YES;
                                             }
                                           }];

  [ref setValue:@{@"a" : @5, @"b" : @6}];

  [self waitUntil:^BOOL {
    return called;
  }];

  called = NO;

  [ref removeObserverWithHandle:handle];

  __block BOOL complete = NO;
  [ref setValue:@{@"a" : @6, @"b" : @7}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        complete = YES;
      }];

  [self waitUntil:^BOOL {
    return complete;
  }];

  XCTAssertFalse(called, @"Should not have been called again");
}

- (void)testOffCanBeCalledOnSpecificQuery {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL called = NO;
  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  FIRDatabaseHandle handle = [query observeEventType:FIRDataEventTypeValue
                                           withBlock:^(FIRDataSnapshot* snapshot) {
                                             if (called) {
                                               XCTFail(@"Should not be called twice");
                                             } else {
                                               called = YES;
                                             }
                                           }];

  [ref setValue:@{@"a" : @5, @"b" : @6}];

  [self waitUntil:^BOOL {
    return called;
  }];

  called = NO;

  [query removeObserverWithHandle:handle];

  __block BOOL complete = NO;
  [ref setValue:@{@"a" : @6, @"b" : @7}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        complete = YES;
      }];

  [self waitUntil:^BOOL {
    return complete;
  }];

  XCTAssertFalse(called, @"Should not have been called again");
}

- (void)testOffCanBeCalledOnMultipleQueries {
  FIRDatabaseQuery* query = [[FTestHelpers getRandomNode] queryLimitedToFirst:10];
  FIRDatabaseHandle handle1 = [query observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  FIRDatabaseHandle handle2 = [query observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  [query removeObserverWithHandle:handle1];
  [query removeObserverWithHandle:handle2];
}

- (void)testOffCanBeCalledWithoutHandle {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL called1 = NO;
  __block BOOL called2 = NO;
  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                called1 = YES;
              }];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  called2 = YES;
                }];

  [ref setValue:@{@"a" : @5, @"b" : @6}];

  [self waitUntil:^BOOL {
    return called1 && called2;
  }];

  called1 = NO;
  called2 = NO;

  [ref removeAllObservers];

  __block BOOL complete = NO;
  [ref setValue:@{@"a" : @6, @"b" : @7}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        complete = YES;
      }];

  [self waitUntil:^BOOL {
    return complete;
  }];

  XCTAssertFalse(called1 || called2, @"Should not have called either callback");
}

- (void)testEnsureOnly5ItemsAreKept {
  __block FIRDataSnapshot* snap = nil;
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  __block int count = 0;
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  snap = snapshot;
                  count++;
                }];

  [ref setValue:nil];
  for (int i = 0; i < 10; ++i) {
    [[ref childByAutoId] setValue:[NSNumber numberWithInt:i]];
  }

  [self waitUntil:^BOOL {
    // The initial set triggers the callback, so we need to wait for 11 events
    return count == 11;
  }];

  count = 5;
  for (FIRDataSnapshot* snapshot in snap.children) {
    NSNumber* num = [snapshot value];
    NSNumber* current = [NSNumber numberWithInt:count];
    XCTAssertTrue([num isEqualToNumber:current], @"Expect children in order");
    count++;
  }

  XCTAssertTrue(count == 10, @"Expected 5 children");
}

- (void)testOnlyLast5SentFromServer {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block int count = 0;

  [ref setValue:nil];

  for (int i = 0; i < 10; ++i) {
    [[ref childByAutoId] setValue:[NSNumber numberWithInt:i]
              withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
                count++;
              }];
  }

  [self waitUntil:^BOOL {
    return count == 10;
  }];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:5];
  count = 5;
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  for (FIRDataSnapshot* child in snapshot.children) {
                    NSNumber* num = [child value];
                    NSNumber* current = [NSNumber numberWithInt:count];
                    XCTAssertTrue([num isEqualToNumber:current], @"Expect children to be in order");
                    count++;
                  }
                }];

  [self waitUntil:^BOOL {
    return count == 10;
  }];
}

- (void)testVariousLimits {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[ref queryLimitedToLast:1] withExpectation:@{@"c" : @3}];
  [expectations
             addQuery:[[[ref queryOrderedByPriority] queryEndingAtValue:nil] queryLimitedToLast:1]
      withExpectation:@{@"c" : @3}];
  [expectations
             addQuery:[[[ref queryOrderedByPriority] queryEndingAtValue:nil] queryLimitedToLast:2]
      withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations
             addQuery:[[[ref queryOrderedByPriority] queryEndingAtValue:nil] queryLimitedToLast:3]
      withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3}];
  [expectations
             addQuery:[[[ref queryOrderedByPriority] queryEndingAtValue:nil] queryLimitedToLast:4]
      withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3}];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  [expectations validate];
}

- (void)testVariousLimitsWithEndBefore {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryEndingBeforeValue:nil]
                             queryLimitedToLast:1]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"d"]
                             queryLimitedToLast:2]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"c"]
                             queryLimitedToLast:2]
         withExpectation:@{@"a" : @1, @"b" : @2}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"b"]
                             queryLimitedToLast:2]
         withExpectation:@{@"a" : @1}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"a"]
                             queryLimitedToLast:1]
         withExpectation:@{}];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  [expectations validate];
}

- (void)testSetLimitsWithStartAt {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil]
                             queryLimitedToFirst:1]
         withExpectation:@{@"a" : @1}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"c"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"b"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"b" : @2}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"b"]
                             queryLimitedToFirst:2]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"b"]
                             queryLimitedToFirst:3]
         withExpectation:@{@"b" : @2, @"c" : @3}];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  [expectations validate];
}

- (void)testSetLimitsWithStartAfter {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil]
                             queryLimitedToFirst:1]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"c"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:2]
         withExpectation:@{@"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:3]
         withExpectation:@{@"c" : @3, @"d" : @4}];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  [expectations validate];
}

- (void)testLimitsAndStartAtWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:nil]
                             queryLimitedToFirst:1]
         withExpectation:@{@"a" : @1}];

  /*params = [[FQueryParams alloc] init];
  params = [params setStartPriority:nil andName:@"c"];
  params = [params limitTo:1];
  [expectations addQuery:[ref queryWithParams:params] withExpectation:@{@"c": @3}];

  params = [[FQueryParams alloc] init];
  params = [params setStartPriority:nil andName:@"b"];
  params = [params limitTo:1];
  [expectations addQuery:[ref queryWithParams:params] withExpectation:@{@"b": @2}];

  params = [[FQueryParams alloc] init];
  params = [params setStartPriority:nil andName:@"b"];
  params = [params limitTo:2];
  [expectations addQuery:[ref queryWithParams:params] withExpectation:@{@"b": @2, @"c": @3}];

  params = [[FQueryParams alloc] init];
  params = [params setStartPriority:nil andName:@"b"];
  params = [params limitTo:3];
  [expectations addQuery:[ref queryWithParams:params] withExpectation:@{@"b": @2, @"c": @3}];*/

  [self waitUntil:^BOOL {
    return expectations.isReady;
  }];
  [expectations validate];
  [ref removeAllObservers];
}

- (void)testLimitsAndStartAfterWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil]
                             queryLimitedToFirst:1]
         withExpectation:@{}];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"c"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:1]
         withExpectation:@{@"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:2]
         withExpectation:@{@"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"b"]
                             queryLimitedToFirst:3]
         withExpectation:@{@"c" : @3, @"d" : @4}];

  [self waitUntil:^BOOL {
    return expectations.isReady;
  }];
  [expectations validate];
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHit {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeChildAdded
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       [added addObject:[snapshot key]];
                                     }];
  [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeChildRemoved
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       [removed addObject:[snapshot key]];
                                     }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"d"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"d" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add d");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHitWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  FIRDatabaseQuery* query = [ref queryLimitedToLast:2];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 2;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"d"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"d" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add d");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHitWithStart {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"a"] queryLimitedToFirst:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"a", @"b" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"aa"] setValue:@4
          withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
            ready = YES;
          }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"aa" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add aa");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHitWithStartAfter {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                    childKey:@"a"] queryLimitedToFirst:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"aa"] setValue:@4
          withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
            ready = YES;
          }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"c" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"aa" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add aa");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHitWithStartAndServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"a"] queryLimitedToFirst:2];
  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 2;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"a", @"b" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"aa"] setValue:@4
          withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
            ready = YES;
          }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"aa" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add aa");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenLimitIsHitWithStartAfterAndServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                    childKey:@"a"] queryLimitedToFirst:2];
  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 2;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"aa"] setValue:@4
          withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
            ready = YES;
          }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"c" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"aa" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add aa");
  [ref removeAllObservers];
}

- (void)testStartAndLimitWithIncompleteWindow {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"a"] queryLimitedToFirst:2];
  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready && [added count] >= 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Expected to remove nothing");
  expected = @[ @"b" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add b");
  [ref removeAllObservers];
}

- (void)testStartAfterAndLimitWithIncompleteWindow {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                    childKey:@"a"] queryLimitedToFirst:2];
  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready && [added count] >= 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"a"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Expected to remove nothing");
  expected = @[];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected _not_ to add a");
  [ref removeAllObservers];
}

- (void)testStartAndLimitWithIncompleteWindowAndServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"a"] queryLimitedToFirst:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Expected to remove nothing");
  expected = @[ @"b" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add b");
  [ref removeAllObservers];
}

- (void)testStartAfterAndLimitWithIncompleteWindowAndServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                    childKey:@"a"] queryLimitedToFirst:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"a"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([removed count] == 0, @"Expected to remove nothing");
  expected = @[];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add b");
  [ref removeAllObservers];
}

- (void)testChildEventsFiredWhenItemDeleted {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready && [added count] >= 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  expected = @[ @"a" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Expected to add a");
  [ref removeAllObservers];
}

- (void)testChildEventsAreFiredWhenItemDeletedAtServer {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNodeWithoutPersistence];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 2;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertEqualObjects(removed, (@[ @"b" ]), @"Expected to remove b");
  XCTAssertEqualObjects(added, (@[ @"a" ]), @"Expected to add a");
  [ref removeAllObservers];
}

- (void)testRemoveFiredWhenItemDeleted {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:2];
  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready && [added count] >= 1;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have one item");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  XCTAssertTrue([added count] == 0, @"Expected to add nothing");
  [ref removeAllObservers];
}

- (void)testRemoveFiredWhenItemDeletedAtServer {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{@"b" : @2, @"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  FIRDatabaseQuery* query = [ref queryLimitedToLast:2];

  NSMutableArray* added = [[NSMutableArray alloc] init];
  NSMutableArray* removed = [[NSMutableArray alloc] init];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [added addObject:[snapshot key]];
                }];
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  [removed addObject:[snapshot key]];
                }];

  [self waitUntil:^BOOL {
    return [added count] == 2;
  }];

  XCTAssertTrue([removed count] == 0, @"Nothing should be removed from our window");
  NSArray* expected = @[ @"b", @"c" ];
  XCTAssertTrue([added isEqualToArray:expected], @"Should have two items");

  [added removeAllObjects];
  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"b" ];
  XCTAssertTrue([removed isEqualToArray:expected], @"Expected to remove b");
  XCTAssertTrue([added count] == 0, @"Expected to add nothing");
  [ref removeAllObservers];
}

- (void)testStartAtPriorityAndEndAtPriorityWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingAtValue:@"y"]
         withExpectation:@{@"b" : @2, @"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingAtValue:@"w"]
         withExpectation:@{@"d" : @4}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAtValue:@"a"] queryEndingAtValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAfterWithOrderByKey {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabaseReference* childOne = [ref childByAutoId];
  FIRDatabaseReference* childTwo = [ref childByAutoId];

  [self waitForCompletionOf:childOne setValue:@1];
  [self waitForCompletionOf:childTwo setValue:@2];

  __block BOOL done = NO;

  [[[ref queryOrderedByKey] queryStartingAfterValue:[childOne key]]
      getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
        XCTAssert([[snapshot value] isKindOfClass:[NSDictionary class]]);
        NSDictionary* data = (NSDictionary*)[snapshot value];
        XCTAssertEqualObjects([data allKeys], @[ [childTwo key] ]);
        XCTAssertEqualObjects([data allValues], @[ @2 ]);
        done = YES;
      }];

  WAIT_FOR(done);
}

- (void)testEndBeforeWithOrderByKey {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabaseReference* childOne = [ref childByAutoId];
  FIRDatabaseReference* childTwo = [ref childByAutoId];

  [self waitForCompletionOf:childOne setValue:@1];
  [self waitForCompletionOf:childTwo setValue:@2];

  __block BOOL done = NO;

  [[[ref queryOrderedByKey] queryEndingBeforeValue:[childTwo key]]
      getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
        XCTAssert([[snapshot value] isKindOfClass:[NSDictionary class]]);
        NSDictionary* data = (NSDictionary*)[snapshot value];
        XCTAssertEqualObjects([data allKeys], @[ [childOne key] ]);
        XCTAssertEqualObjects([data allValues], @[ @1 ]);
        done = YES;
      }];

  WAIT_FOR(done);
}

- (void)testStartAfterWithOrderByKeyOverlappingListener {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabaseReference* childOne = [ref childByAutoId];
  FIRDatabaseReference* childTwo = [ref childByAutoId];

  [self waitForCompletionOf:childOne setValue:@1];
  [self waitForCompletionOf:childTwo setValue:@2];

  __block BOOL done = NO;

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                done = YES;
              }];

  WAIT_FOR(done);
  done = NO;

  [[[ref queryOrderedByKey] queryStartingAfterValue:[childOne key]]
      getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
        XCTAssert([[snapshot value] isKindOfClass:[NSDictionary class]]);
        NSDictionary* data = (NSDictionary*)[snapshot value];
        XCTAssertEqualObjects([data allKeys], @[ [childTwo key] ]);
        XCTAssertEqualObjects([data allValues], @[ @2 ]);
        done = YES;
      }];

  WAIT_FOR(done);
  [ref removeAllObservers];
}

- (void)testEndBeforeWithOrderByKeyOverlappingListener {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabaseReference* childOne = [ref childByAutoId];
  FIRDatabaseReference* childTwo = [ref childByAutoId];

  [self waitForCompletionOf:childOne setValue:@1];
  [self waitForCompletionOf:childTwo setValue:@2];

  __block BOOL done = NO;

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                done = YES;
              }];

  WAIT_FOR(done);
  done = NO;

  [[[ref queryOrderedByKey] queryEndingBeforeValue:[childTwo key]]
      getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
        XCTAssert([[snapshot value] isKindOfClass:[NSDictionary class]]);
        NSDictionary* data = (NSDictionary*)[snapshot value];
        XCTAssertEqualObjects([data allKeys], @[ [childOne key] ]);
        XCTAssertEqualObjects([data allValues], @[ @1 ]);
        done = YES;
      }];

  WAIT_FOR(done);
  [ref removeAllObservers];
}

- (void)testStartAfterPriorityAndEndAtPriorityWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"y"]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"w"]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"x"]
         withExpectation:@{@"c" : @3}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAfterValue:@"a"] queryEndingAtValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAtPriorityAndEndBeforePriorityWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingBeforeValue:@"y"]
         withExpectation:@{@"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"x"]
                             queryEndingBeforeValue:@"z"]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"z"]
                             queryEndingBeforeValue:@"z"]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingBeforeValue:@"w"]
         withExpectation:@{}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAtValue:@"a"] queryEndingBeforeValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAtPriorityAndEndAtPriorityWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingAtValue:@"y"]
         withExpectation:@{@"b" : @2, @"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingAtValue:@"w"]
         withExpectation:@{@"d" : @4}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAtValue:@"a"] queryEndingAtValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAfterPriorityAndEndAtPriorityWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"y"]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"w"]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAfterValue:@"w"]
                             queryEndingAtValue:@"x"]
         withExpectation:@{@"c" : @3}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAfterValue:@"a"] queryEndingAtValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAtPriorityAndEndBeforePriorityWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingBeforeValue:@"y"]
         withExpectation:@{@"c" : @3, @"d" : @4}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"x"]
                             queryEndingBeforeValue:@"z"]
         withExpectation:@{@"b" : @2, @"c" : @3}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"z"]
                             queryEndingBeforeValue:@"z"]
         withExpectation:@{}];
  [expectations addQuery:[[[ref queryOrderedByPriority] queryStartingAtValue:@"w"]
                             queryEndingBeforeValue:@"w"]
         withExpectation:@{}];

  __block id nullSnap = @"dummy";
  [[[[ref queryOrderedByPriority] queryStartingAtValue:@"a"] queryEndingBeforeValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testStartAtAndEndAtPriorityAndNameWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"a"]
      queryEndingAtValue:@2
                childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"b"] queryEndingAtValue:@2
                                                                              childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"b" : @2, @"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"c"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"c" : @3, @"d" : @4}];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAfterAndEndAtPriorityAndNameWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1 childKey:@"a"]
      queryEndingAtValue:@2
                childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1
                                                        childKey:@"b"] queryEndingAtValue:@2
                                                                                 childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1
                                                        childKey:@"c"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"c" : @3, @"d" : @4}];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAtAndEndBeforePriorityAndNameWork {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"a"]
      queryEndingBeforeValue:@2
                    childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"b"] queryEndingBeforeValue:@2
                                                                                  childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"b" : @2}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"c"] queryEndingBeforeValue:@2];
  [expectations addQuery:query withExpectation:@{}];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAtAndEndAtPriorityAndNameWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"a"]
      queryEndingAtValue:@2
                childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"b"] queryEndingAtValue:@2
                                                                              childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"b" : @2, @"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"c"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"c" : @3, @"d" : @4}];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAfterAndEndAtPriorityAndNameWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1 childKey:@"a"]
      queryEndingAtValue:@2
                childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1
                                                        childKey:@"b"] queryEndingAtValue:@2
                                                                                 childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAfterValue:@1
                                                        childKey:@"c"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"c" : @3, @"d" : @4}];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAtAndEndBeforePriorityAndNameWorkWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"a"]
      queryEndingBeforeValue:@2
                    childKey:@"d"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"b"] queryEndingBeforeValue:@2
                                                                                  childKey:@"c"];
  [expectations addQuery:query withExpectation:@{@"b" : @2}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"c"] queryEndingBeforeValue:@2];
  [expectations addQuery:query withExpectation:@{}];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAtAndEndAtPriorityAndNameWork2 {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"c"]
      queryEndingAtValue:@2
                childKey:@"b"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"d"] queryEndingAtValue:@2
                                                                              childKey:@"a"];
  [expectations addQuery:query withExpectation:@{@"d" : @4, @"a" : @1}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"e"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2}];

  [ref setValue:@{
    @"c" : @{@".value" : @3, @".priority" : @1},
    @"d" : @{@".value" : @4, @".priority" : @1},
    @"a" : @{@".value" : @1, @".priority" : @2},
    @"b" : @{@".value" : @2, @".priority" : @2}
  }];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testStartAtAndEndAtPriorityAndNameWorkWithServerData2 {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block BOOL ready = NO;
  [ref setValue:@{
    @"c" : @{@".value" : @3, @".priority" : @1},
    @"d" : @{@".value" : @4, @".priority" : @1},
    @"a" : @{@".value" : @1, @".priority" : @2},
    @"b" : @{@".value" : @2, @".priority" : @2}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1 childKey:@"c"]
      queryEndingAtValue:@2
                childKey:@"b"];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"d"] queryEndingAtValue:@2
                                                                              childKey:@"a"];
  [expectations addQuery:query withExpectation:@{@"d" : @4, @"a" : @1}];

  query = [[[ref queryOrderedByPriority] queryStartingAtValue:@1
                                                     childKey:@"e"] queryEndingAtValue:@2];
  [expectations addQuery:query withExpectation:@{@"a" : @1, @"b" : @2}];

  WAIT_FOR(expectations.isReady);

  [expectations validate];
}

- (void)testEqualToPriorityWorks {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[ref queryOrderedByPriority] queryEqualToValue:@"w"]
         withExpectation:@{@"d" : @4}];

  __block id nullSnap = @"dummy";
  [[[ref queryOrderedByPriority] queryEqualToValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testEqualToPriorityWorksWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @"z"},
    @"b" : @{@".value" : @2, @".priority" : @"y"},
    @"c" : @{@".value" : @3, @".priority" : @"x"},
    @"d" : @{@".value" : @4, @".priority" : @"w"}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  [expectations addQuery:[[ref queryOrderedByPriority] queryEqualToValue:@"w"]
         withExpectation:@{@"d" : @4}];

  __block id nullSnap = @"dummy";
  [[[ref queryOrderedByPriority] queryEqualToValue:@"c"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testEqualToPriorityAndNameWorks {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[ref queryOrderedByPriority] queryEqualToValue:@1 childKey:@"a"];
  [expectations addQuery:query withExpectation:@{@"a" : @1}];

  __block id nullSnap = @"dummy";
  [[[ref queryOrderedByPriority] queryEqualToValue:@"1" childKey:@"z"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testEqualToPriorityAndNameWorksWithServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block BOOL ready = NO;
  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @1},
    @"c" : @{@".value" : @3, @".priority" : @2},
    @"d" : @{@".value" : @4, @".priority" : @2}
  }
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  FTestExpectations* expectations = [[FTestExpectations alloc] initFrom:self];

  FIRDatabaseQuery* query = [[ref queryOrderedByPriority] queryEqualToValue:@1 childKey:@"a"];
  [expectations addQuery:query withExpectation:@{@"a" : @1}];

  __block id nullSnap = @"dummy";
  [[[ref queryOrderedByPriority] queryEqualToValue:@"1" childKey:@"z"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               nullSnap = [snapshot value];
             }];

  WAIT_FOR(expectations.isReady && [nullSnap isEqual:[NSNull null]]);

  [expectations validate];
}

- (void)testPrevNameWorks {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* added = [[NSMutableArray alloc] init];

  [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeChildAdded
                andPreviousSiblingKeyWithBlock:^(FIRDataSnapshot* snapshot, NSString* prevName) {
                  [added addObject:snapshot.key];
                  if (prevName) {
                    [added addObject:prevName];
                  } else {
                    [added addObject:@"null"];
                  }
                }];

  [[ref child:@"a"] setValue:@1];
  [self waitUntil:^BOOL {
    NSArray* expected = @[ @"a", @"null" ];
    return [added isEqualToArray:expected];
  }];

  [added removeAllObjects];

  [[ref child:@"c"] setValue:@3];
  [self waitUntil:^BOOL {
    NSArray* expected = @[ @"c", @"a" ];
    return [added isEqualToArray:expected];
  }];

  [added removeAllObjects];

  [[ref child:@"b"] setValue:@2];
  [self waitUntil:^BOOL {
    NSArray* expected = @[ @"b", @"null" ];
    return [added isEqualToArray:expected];
  }];

  [added removeAllObjects];

  [[ref child:@"d"] setValue:@3];
  [self waitUntil:^BOOL {
    NSArray* expected = @[ @"d", @"c" ];
    return [added isEqualToArray:expected];
  }];
}

// Dropping some of the server data tests here, around prevName. They don't really test anything
// new, and mostly don't even test server data

- (void)testPrevNameWorksWithMoves {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* moved = [[NSMutableArray alloc] init];

  [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeChildMoved
                andPreviousSiblingKeyWithBlock:^(FIRDataSnapshot* snapshot, NSString* prevName) {
                  [moved addObject:snapshot.key];
                  if (prevName) {
                    [moved addObject:prevName];
                  } else {
                    [moved addObject:@"null"];
                  }
                }];

  [ref setValue:@{
    @"a" : @{@".value" : @"a", @".priority" : @10},
    @"b" : @{@".value" : @"b", @".priority" : @20},
    @"c" : @{@".value" : @"c", @".priority" : @30},
    @"d" : @{@".value" : @"d", @".priority" : @40}
  }];

  __block BOOL ready = NO;
  [[ref child:@"c"] setPriority:@50
            withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
              ready = YES;
            }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSArray* expected = @[ @"c", @"d" ];
  XCTAssertTrue([moved isEqualToArray:expected], @"Expected changed node and prevChild");

  [moved removeAllObjects];
  ready = NO;
  [[ref child:@"c"] setPriority:@35
            withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
              ready = YES;
            }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[ @"c", @"null" ];
  XCTAssertTrue([moved isEqualToArray:expected], @"Expected changed node and prevChild");

  [moved removeAllObjects];
  ready = NO;
  [[ref child:@"b"] setPriority:@33
            withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
              ready = YES;
            }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  expected = @[];
  XCTAssertTrue([moved isEqualToArray:expected],
                @"Expected changed node and prevChild to be empty");
}

- (void)testLocalEvents {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* events = [[NSMutableArray alloc] init];
  [[ref queryLimitedToLast:2]
      observeEventType:FIRDataEventTypeChildAdded
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSString* eventString = [NSString stringWithFormat:@"%@ added", [snapshot value]];
               [events addObject:eventString];
             }];

  [[ref queryLimitedToLast:2]
      observeEventType:FIRDataEventTypeChildRemoved
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSString* eventString = [NSString stringWithFormat:@"%@ removed", [snapshot value]];
               [events addObject:eventString];
             }];

  __block BOOL ready = NO;
  for (int i = 0; i < 5; ++i) {
    [[ref childByAutoId] setValue:[NSNumber numberWithInt:i]
              withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
                if (i == 4) {
                  ready = YES;
                }
              }];
  }

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSArray* expected = @[
    @"0 added", @"1 added", @"0 removed", @"2 added", @"1 removed", @"3 added", @"2 removed",
    @"4 added"
  ];
  XCTAssertTrue([events isEqualToArray:expected], @"Expecting window to stay at two nodes");
}

- (void)testRemoteEvents {
  FTupleFirebase* pair = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = pair.one;
  FIRDatabaseReference* reader = pair.two;

  NSMutableArray* events = [[NSMutableArray alloc] init];

  [[reader queryLimitedToLast:2]
      observeEventType:FIRDataEventTypeChildAdded
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSString* eventString = [NSString stringWithFormat:@"%@ added", [snapshot value]];
               [events addObject:eventString];
             }];

  [[reader queryLimitedToLast:2]
      observeEventType:FIRDataEventTypeChildRemoved
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSString* oldEventString = [NSString stringWithFormat:@"%@ added", [snapshot value]];
               [events removeObject:oldEventString];
             }];

  for (int i = 0; i < 5; ++i) {
    [[writer childByAutoId] setValue:[NSNumber numberWithInt:i]];
  }

  NSArray* expected = @[ @"3 added", @"4 added" ];
  [self waitUntil:^BOOL {
    return [events isEqualToArray:expected];
  }];
}

- (void)testLimitOnEmptyNodeFiresValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  [[ref queryLimitedToLast:1] observeEventType:FIRDataEventTypeValue
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       ready = YES;
                                     }];

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testFilteringToNullPriorities {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  // Note: cannot set nil in a dictionary, just leave out priority
  [ref setValue:@{
    @"a" : @0,
    @"b" : @1,
    @"c" : @{@".priority" : @2, @".value" : @2},
    @"d" : @{@".priority" : @3, @".value" : @3},
    @"e" : @{@".priority" : @"hi", @".value" : @4}
  }];

  __block BOOL ready = NO;
  [[[[ref queryOrderedByPriority] queryStartingAtValue:nil] queryEndingAtValue:nil]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSDictionary* expected = @{@"a" : @0, @"b" : @1};
               NSDictionary* val = [snapshot value];
               XCTAssertTrue([val isEqualToDictionary:expected],
                             @"Expected only null priority keys");
               ready = YES;
             }];

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testNullPrioritiesIncludedInEndAt {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  // Note: cannot set nil in a dictionary, just leave out priority
  [ref setValue:@{
    @"a" : @0,
    @"b" : @1,
    @"c" : @{@".priority" : @2, @".value" : @2},
    @"d" : @{@".priority" : @3, @".value" : @3},
    @"e" : @{@".priority" : @"hi", @".value" : @4}
  }];

  __block BOOL ready = NO;
  [[[ref queryOrderedByPriority] queryEndingAtValue:@2]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSDictionary* expected = @{@"a" : @0, @"b" : @1, @"c" : @2};
               NSDictionary* val = [snapshot value];
               XCTAssertTrue([val isEqualToDictionary:expected], @"Expected up to priority 2");
               ready = YES;
             }];

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testNullPrioritiesIncludedInEndBefore {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  // Note: cannot set nil in a dictionary, just leave out priority
  [ref setValue:@{
    @"a" : @0,
    @"b" : @1,
    @"c" : @{@".priority" : @2, @".value" : @2},
    @"d" : @{@".priority" : @3, @".value" : @3},
    @"e" : @{@".priority" : @"hi", @".value" : @4}
  }];

  __block BOOL ready = NO;
  [[[ref queryOrderedByPriority] queryEndingBeforeValue:@2]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               NSDictionary* expected = @{@"a" : @0, @"b" : @1};
               NSDictionary* val = [snapshot value];
               XCTAssertTrue([val isEqualToDictionary:expected], @"Expected up to priority 2");
               ready = YES;
             }];

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (NSSet*)dumpListensForRef:(FIRDatabaseReference*)ref {
  NSMutableSet* dumpPieces = [[NSMutableSet alloc] init];
  NSDictionary* listens = [ref.repo dumpListens];

  FPath* nodePath = ref.path;
  [listens enumerateKeysAndObjectsUsingBlock:^(FQuerySpec* spec, id obj, BOOL* stop) {
    if ([nodePath contains:spec.path]) {
      FPath* relative = [FPath relativePathFrom:nodePath to:spec.path];
      [dumpPieces addObject:[[FQuerySpec alloc] initWithPath:relative params:spec.params]];
    }
  }];

  return dumpPieces;
}

- (NSSet*)expectDefaultListenerAtPath:(FPath*)path {
  return [self expectParams:[FQueryParams defaultInstance] atPath:path];
}

- (NSSet*)expectParamssetValue:(NSSet*)paramsSet atPath:(FPath*)path {
  NSMutableSet* all = [NSMutableSet set];
  [paramsSet enumerateObjectsUsingBlock:^(FQueryParams* params, BOOL* stop) {
    [all addObject:[[FQuerySpec alloc] initWithPath:path params:params]];
  }];
  return all;
}

- (NSSet*)expectParams:(FQueryParams*)params atPath:(FPath*)path {
  return [self expectParamssetValue:[NSSet setWithObject:params] atPath:path];
}

- (void)testDedupesListensOnChild {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block NSSet* listens = [self dumpListensForRef:ref];
  XCTAssertTrue(listens.count == 0, @"No Listens yet");

  [[ref child:@"a"] observeEventType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot){
                           }];
  __block BOOL ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [NSSet setWithObject:[FQuerySpec defaultQueryAtPath:PATH(@"a")]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected child listener");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot){
              }];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [NSSet setWithObject:[FQuerySpec defaultQueryAtPath:PATH(@"")]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected parent listener");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [NSSet setWithObject:[FQuerySpec defaultQueryAtPath:PATH(@"a")]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Child listener should be back");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [[ref child:@"a"] removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No more listeners");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testDedupeListensOnGrandchild {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block NSSet* listens;
  __block BOOL ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No Listens yet");
    ready = YES;
  });
  WAIT_FOR(ready);

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot){
              }];

  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath empty]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected one listener");
    ready = YES;
  });
  WAIT_FOR(ready);

  [[ref child:@"a/aa"] observeEventType:FIRDataEventTypeValue
                              withBlock:^(FIRDataSnapshot* snapshot){
                              }];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath empty]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected parent listener to override");
    ready = YES;
  });
  WAIT_FOR(ready);

  [ref removeAllObservers];
  [[ref child:@"a/aa"] removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No more listeners");
    ready = YES;
  });
  WAIT_FOR(ready);
}

- (void)testListenOnGrandparentOfTwoChildren {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block NSSet* listens = [self dumpListensForRef:ref];
  XCTAssertTrue(listens.count == 0, @"No Listens yet");

  [[ref child:@"a/aa"] observeEventType:FIRDataEventTypeValue
                              withBlock:^(FIRDataSnapshot* snapshot){
                              }];
  __block BOOL ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/aa"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected grandchild");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [[ref child:@"a/bb"] observeEventType:FIRDataEventTypeValue
                              withBlock:^(FIRDataSnapshot* snapshot){
                              }];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expecteda = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/aa"]];
    NSSet* expectedb = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/bb"]];
    NSMutableSet* expected = [NSMutableSet setWithSet:expecteda];
    [expected unionSet:expectedb];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected two grandchildren");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot){
              }];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath empty]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected parent listener to override");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expecteda = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/aa"]];
    NSSet* expectedb = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/bb"]];
    NSMutableSet* expected = [NSMutableSet setWithSet:expecteda];
    [expected unionSet:expectedb];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected grandchild listeners to return");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [[ref child:@"a/aa"] removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath pathWithString:@"/a/bb"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Expected one listener");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [[ref child:@"a/bb"] removeAllObservers];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No more listeners");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testDedupingMultipleListenQueries {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block NSSet* listens = [self dumpListensForRef:ref];
  XCTAssertTrue(listens.count == 0, @"No Listens yet");

  __block BOOL ready = NO;
  FIRDatabaseQuery* aLim1 = [[ref child:@"a"] queryLimitedToLast:1];
  FIRDatabaseHandle handle1 = [aLim1 observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams = [[FQueryParams alloc] init];
    expectedParams = [expectedParams limitTo:1];
    NSSet* expected = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/a"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Single query");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseQuery* rootLim1 = [ref queryLimitedToLast:1];
  FIRDatabaseHandle handle2 = [rootLim1 observeEventType:FIRDataEventTypeValue
                                               withBlock:^(FIRDataSnapshot* snapshot){
                                               }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams = [[FQueryParams alloc] init];
    expectedParams = [expectedParams limitTo:1];
    NSSet* rootExpected = [self expectParams:expectedParams atPath:[FPath empty]];
    NSSet* childExpected = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/a"]];
    NSMutableSet* expected = [NSMutableSet setWithSet:rootExpected];
    [expected unionSet:childExpected];
    XCTAssertTrue([expected isEqualToSet:listens], @"Two queries");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseQuery* aLim5 = [[ref child:@"a"] queryLimitedToLast:5];
  FIRDatabaseHandle handle3 = [aLim5 observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams1 = [[FQueryParams alloc] init];
    expectedParams1 = [expectedParams1 limitTo:1];
    NSSet* rootExpected = [self expectParams:expectedParams1 atPath:[FPath empty]];

    FQueryParams* expectedParams2 = [[FQueryParams alloc] init];
    expectedParams2 = [expectedParams2 limitTo:5];
    NSSet* childExpected =
        [self expectParamssetValue:[NSSet setWithObjects:expectedParams1, expectedParams2, nil]
                            atPath:[FPath pathWithString:@"/a"]];
    NSMutableSet* expected = [NSMutableSet setWithSet:childExpected];
    [expected unionSet:rootExpected];
    XCTAssertTrue([expected isEqualToSet:listens], @"Three queries");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref removeObserverWithHandle:handle2];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams1 = [[FQueryParams alloc] init];
    expectedParams1 = [expectedParams1 limitTo:1];
    FQueryParams* expectedParams2 = [[FQueryParams alloc] init];
    expectedParams2 = [expectedParams2 limitTo:5];
    NSSet* expected =
        [self expectParamssetValue:[NSSet setWithObjects:expectedParams1, expectedParams2, nil]
                            atPath:[FPath pathWithString:@"/a"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Two queries");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [aLim1 removeObserverWithHandle:handle1];
  [aLim5 removeObserverWithHandle:handle3];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No more listeners");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testListenOnParentOfQueriedChildren {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  __block NSSet* listens = [self dumpListensForRef:ref];
  XCTAssertTrue(listens.count == 0, @"No Listens yet");

  __block BOOL ready = NO;
  FIRDatabaseQuery* aLim1 = [[ref child:@"a"] queryLimitedToLast:1];
  FIRDatabaseHandle handle1 = [aLim1 observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams = [[FQueryParams alloc] init];
    expectedParams = [expectedParams limitTo:1];
    NSSet* expected = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/a"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Single query");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseQuery* bLim1 = [[ref child:@"b"] queryLimitedToLast:1];
  FIRDatabaseHandle handle2 = [bLim1 observeEventType:FIRDataEventTypeValue
                                            withBlock:^(FIRDataSnapshot* snapshot){
                                            }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams = [[FQueryParams alloc] init];
    expectedParams = [expectedParams limitTo:1];
    NSSet* expecteda = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/a"]];
    NSSet* expectedb = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/b"]];
    NSMutableSet* expected = [NSMutableSet setWithSet:expecteda];
    [expected unionSet:expectedb];
    XCTAssertTrue([expected isEqualToSet:listens], @"Two queries");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseHandle handle3 = [ref observeEventType:FIRDataEventTypeValue
                                          withBlock:^(FIRDataSnapshot* snapshot){
                                          }];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath empty]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Parent should override");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  // remove in slightly random order
  [aLim1 removeObserverWithHandle:handle1];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    NSSet* expected = [self expectDefaultListenerAtPath:[FPath empty]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Parent should override");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  [ref removeObserverWithHandle:handle3];
  ready = NO;
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    FQueryParams* expectedParams = [[FQueryParams alloc] init];
    expectedParams = [expectedParams limitTo:1];
    NSSet* expected = [self expectParams:expectedParams atPath:[FPath pathWithString:@"/b"]];
    XCTAssertTrue([expected isEqualToSet:listens], @"Single query");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  [bLim1 removeObserverWithHandle:handle2];
  dispatch_async([FIRDatabaseQuery sharedQueue], ^{
    listens = [self dumpListensForRef:ref];
    XCTAssertTrue(listens.count == 0, @"No more listeners");
    ready = YES;
  });

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testLimitWithMixOfNullAndNonNullPriorities {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* children = [[NSMutableArray alloc] init];

  [[ref queryLimitedToLast:5] observeEventType:FIRDataEventTypeChildAdded
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       [children addObject:[snapshot key]];
                                     }];

  __block BOOL ready = NO;
  NSDictionary* toSet = @{
    @"Vikrum" : @{@".priority" : @1000, @"score" : @1000, @"name" : @"Vikrum"},
    @"Mike" : @{@".priority" : @500, @"score" : @500, @"name" : @"Mike"},
    @"Andrew" : @{@".priority" : @50, @"score" : @50, @"name" : @"Andrew"},
    @"James" : @{@".priority" : @7, @"score" : @7, @"name" : @"James"},
    @"Sally" : @{@".priority" : @-7, @"score" : @-7, @"name" : @"Sally"},
    @"Fred" : @{@"score" : @0, @"name" : @"Fred"}
  };

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSArray* expected = @[ @"Sally", @"James", @"Andrew", @"Mike", @"Vikrum" ];
  XCTAssertTrue([children isEqualToArray:expected], @"Null priority should be left out");
}

- (void)testLimitWithMixOfNullAndNonNullPrioritiesOnServerData {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  NSDictionary* toSet = @{
    @"Vikrum" : @{@".priority" : @1000, @"score" : @1000, @"name" : @"Vikrum"},
    @"Mike" : @{@".priority" : @500, @"score" : @500, @"name" : @"Mike"},
    @"Andrew" : @{@".priority" : @50, @"score" : @50, @"name" : @"Andrew"},
    @"James" : @{@".priority" : @7, @"score" : @7, @"name" : @"James"},
    @"Sally" : @{@".priority" : @-7, @"score" : @-7, @"name" : @"Sally"},
    @"Fred" : @{@"score" : @0, @"name" : @"Fred"}
  };

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  __block int count = 0;
  NSMutableArray* children = [[NSMutableArray alloc] init];

  [[ref queryLimitedToLast:5] observeEventType:FIRDataEventTypeChildAdded
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       [children addObject:[snapshot key]];
                                       count++;
                                     }];

  [self waitUntil:^BOOL {
    return count == 5;
  }];

  NSArray* expected = @[ @"Sally", @"James", @"Andrew", @"Mike", @"Vikrum" ];
  XCTAssertTrue([children isEqualToArray:expected], @"Null priority should be left out");
}

// Skipping context tests. Context is not implemented on iOS

/* DISABLING for now, since I'm not 100% sure what the right behavior is.
   Perhaps a merge at /foo should shadow server updates at /foo instead of
   just the modified children?  Not sure.
- (void) testHandleUpdateThatDeletesEntireWindow {
    Firebase* ref = [FTestHelpers getRandomNode];

    NSMutableArray* snaps = [[NSMutableArray alloc] init];

    [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeValue withBlock:^(FIRDataSnapshot
*snapshot) { id val = [snapshot value]; if (val == nil) { [snaps addObject:[NSNull null]]; } else {
            [snaps addObject:val];
        }
    }];

    NSDictionary* toSet = @{
    @"a": @{@".priority": @1, @".value": @1},
    @"b": @{@".priority": @2, @".value": @2},
    @"c": @{@".priority": @3, @".value": @3}
    };

    [ref setValue:toSet];

    __block BOOL ready = NO;
    toSet = @{@"b": [NSNull null], @"c": [NSNull null]};
    [ref updateChildValues:toSet withCompletionBlock:^(NSError* err, Firebase* ref) {
        ready = YES;
    }];

    [self waitUntil:^BOOL{
        return ready;
    }];

    NSArray* expected = @[@{@"b": @2, @"c": @3}, [NSNull null], @{@"a": @1}];
    STAssertTrue([snaps isEqualToArray:expected], @"Expected %@ to equal %@", snaps, expected);
}
*/

- (void)testHandlesAnOutOfViewQueryOnAChild {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSDictionary* parent = nil;
  [[ref queryLimitedToLast:1] observeEventType:FIRDataEventTypeValue
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       parent = [snapshot value];
                                     }];

  __block NSNumber* child = nil;
  [[ref child:@"a"] observeEventType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             child = [snapshot value];
                           }];

  __block BOOL ready = NO;
  NSDictionary* toSet = @{@"a" : @1, @"b" : @2};
  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSDictionary* parentExpected = @{@"b" : @2};
  NSNumber* childExpected = [NSNumber numberWithInt:1];
  XCTAssertTrue([parent isEqualToDictionary:parentExpected], @"Expected last element");
  XCTAssertTrue([child isEqualToNumber:childExpected], @"Expected value of a");

  ready = NO;
  [ref updateChildValues:@{@"c" : @3}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  parentExpected = @{@"c" : @3};
  XCTAssertTrue([parent isEqualToDictionary:parentExpected], @"Expected last element");
  XCTAssertTrue([child isEqualToNumber:childExpected], @"Expected value of a");
}

- (void)testHandlesAChildQueryGoingOutOfViewOfTheParent {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSDictionary* parent = nil;
  [[ref queryLimitedToLast:1] observeEventType:FIRDataEventTypeValue
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       parent = [snapshot value];
                                     }];

  __block NSNumber* child = nil;
  [[ref child:@"a"] observeEventType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             child = [snapshot value];
                           }];

  __block BOOL ready = NO;
  NSDictionary* toSet = @{@"a" : @1};
  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  WAIT_FOR(ready);

  NSDictionary* parentExpected = @{@"a" : @1};
  NSNumber* childExpected = [NSNumber numberWithInt:1];
  XCTAssertTrue([parent isEqualToDictionary:parentExpected], @"Expected last element");
  XCTAssertTrue([child isEqualToNumber:childExpected], @"Expected value of a");

  ready = NO;
  [[ref child:@"b"] setValue:@2
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  WAIT_FOR(ready);

  parentExpected = @{@"b" : @2};
  XCTAssertTrue([parent isEqualToDictionary:parentExpected], @"Expected last element");
  XCTAssertTrue([child isEqualToNumber:childExpected], @"Expected value of a");

  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  parentExpected = @{@"a" : @1};
  XCTAssertTrue([parent isEqualToDictionary:parentExpected], @"Expected last element");
  XCTAssertTrue([child isEqualToNumber:childExpected], @"Expected value of a");
}

- (void)testHandlesDivergingViews {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSDictionary* cVal = nil;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryEndingAtValue:nil childKey:@"c"] queryLimitedToLast:1];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  cVal = [snapshot value];
                }];

  __block NSDictionary* dVal = nil;
  query = [[[ref queryOrderedByPriority] queryEndingAtValue:nil
                                                   childKey:@"d"] queryLimitedToLast:1];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  dVal = [snapshot value];
                }];

  __block BOOL ready = NO;
  NSDictionary* toSet = @{@"a" : @1, @"b" : @2, @"c" : @3};
  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSDictionary* expected = @{@"c" : @3};
  XCTAssertTrue([cVal isEqualToDictionary:expected], @"should be c");
  XCTAssertTrue([dVal isEqualToDictionary:expected], @"should be c");

  ready = NO;
  [[ref child:@"d"] setValue:@4
         withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([cVal isEqualToDictionary:expected], @"should be c");
  expected = @{@"d" : @4};
  XCTAssertTrue([dVal isEqualToDictionary:expected], @"should be d");
}

- (void)testHandlesRemovingAQueriedElement {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSNumber* val = nil;
  [[ref queryLimitedToLast:1] observeEventType:FIRDataEventTypeChildAdded
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       id newVal = [snapshot value];
                                       if (newVal != nil) {
                                         val = [snapshot value];
                                       }
                                     }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@2], @"Expected last element in window");

  ready = NO;
  [[ref child:@"b"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@1], @"Should now be the next element in the window");
}

- (void)testStartAtAndLimit1Works {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSNumber* val = nil;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil] queryLimitedToFirst:1];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  id newVal = [snapshot value];
                  if (newVal != nil) {
                    val = [snapshot value];
                  }
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@1], @"Expected first element in window");
}

// See case 1664
- (void)testStartAtAndLimit1AndRemoveFirstChild {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSNumber* val = nil;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:nil] queryLimitedToFirst:1];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  id newVal = [snapshot value];
                  if (newVal != nil) {
                    val = [snapshot value];
                  }
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@1], @"Expected first element in window");

  ready = NO;
  [[ref child:@"a"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@2], @"Expected next element in window");
}

- (void)testStartAfterAndLimit1AndRemoveFirstChild {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block NSNumber* val = nil;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                    childKey:@"a"] queryLimitedToFirst:1];
  [query observeEventType:FIRDataEventTypeChildAdded
                withBlock:^(FIRDataSnapshot* snapshot) {
                  id newVal = [snapshot value];
                  if (newVal != nil) {
                    val = [snapshot value];
                  }
                }];

  __block BOOL ready = NO;
  [ref setValue:@{@"a" : @1, @"b" : @2}
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@2], @"Expected first element in window");

  ready = NO;
  [[ref child:@"a"] removeValueWithCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
    ready = YES;
  }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([val isEqualToNumber:@2], @"Expected next element in window");
}

// See case 1169
- (void)testStartAtWithTwoArgumentsWorks {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  NSMutableArray* children = [[NSMutableArray alloc] init];

  NSDictionary* toSet = @{
    @"Walker" : @{@"name" : @"Walker", @"score" : @20, @".priority" : @20},
    @"Michael" : @{@"name" : @"Michael", @"score" : @100, @".priority" : @100}
  };

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAtValue:@20
                                                 childKey:@"Walker"] queryLimitedToFirst:2];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  for (FIRDataSnapshot* child in snapshot.children) {
                    [children addObject:child.key];
                  }
                  ready = YES;
                }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSArray* expected = @[ @"Walker", @"Michael" ];
  XCTAssertTrue([children isEqualToArray:expected], @"Expected both children");
}

- (void)testStartAfterWithTwoArgumentsWorks {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;
  NSMutableArray* children = [[NSMutableArray alloc] init];

  NSDictionary* toSet = @{
    @"Walker" : @{@"name" : @"Walker", @"score" : @20, @".priority" : @20},
    @"Michael" : @{@"name" : @"Michael", @"score" : @100, @".priority" : @100}
  };

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryStartingAfterValue:@20
                                                    childKey:@"Walker"] queryLimitedToFirst:2];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  for (FIRDataSnapshot* child in snapshot.children) {
                    [children addObject:child.key];
                  }
                  ready = YES;
                }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSArray* expected = @[ @"Michael" ];
  XCTAssertTrue([children isEqualToArray:expected], @"Expected single child");
}

- (void)testHandlesMultipleQueriesOnSameNode {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;

  NSDictionary* toSet = @{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4, @"e" : @5, @"f" : @6};

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  __block BOOL called = NO;
  [[ref queryLimitedToLast:2]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               // we got the initial data
               XCTAssertFalse(called,
                              @"This should only get called once, we don't update data after this");
               called = YES;
               ready = YES;
             }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  __block NSDictionary* snap = nil;
  // now do nested once calls
  [[ref queryLimitedToLast:1]
      observeSingleEventOfType:FIRDataEventTypeValue
                     withBlock:^(FIRDataSnapshot* snapshot) {
                       [[ref queryLimitedToLast:1]
                           observeSingleEventOfType:FIRDataEventTypeValue
                                          withBlock:^(FIRDataSnapshot* snapshot) {
                                            snap = [snapshot value];
                                            ready = YES;
                                          }];
                     }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSDictionary* expected = @{@"f" : @6};
  XCTAssertTrue([snap isEqualToDictionary:expected], @"Expected the correct data");
}

- (void)testHandlesOnceCalledOnNodeWithDefaultListener {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;

  NSDictionary* toSet = @{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4, @"e" : @5, @"f" : @6};

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                // we got the initial data
                ready = YES;
              }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;

  __block NSNumber* snap = nil;
  [[ref queryLimitedToLast:1] observeSingleEventOfType:FIRDataEventTypeChildAdded
                                             withBlock:^(FIRDataSnapshot* snapshot) {
                                               snap = [snapshot value];
                                               ready = YES;
                                             }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  XCTAssertTrue([snap isEqualToNumber:@6], @"Got once response");
}

- (void)testHandlesOnceCalledOnNodeWithDefaultListenerAndNonCompleteLimit {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL ready = NO;

  NSDictionary* toSet = @{@"a" : @1, @"b" : @2, @"c" : @3};

  [ref setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  // do first listen
  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                ready = YES;
              }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;

  __block NSDictionary* snap = nil;
  [[ref queryLimitedToLast:5] observeSingleEventOfType:FIRDataEventTypeValue
                                             withBlock:^(FIRDataSnapshot* snapshot) {
                                               snap = [snapshot value];
                                               ready = YES;
                                             }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  NSDictionary* expected = @{@"a" : @1, @"b" : @2, @"c" : @3};
  XCTAssertTrue([snap isEqualToDictionary:expected], @"Got once response");
}

- (void)testRemoveTriggersRemoteEvents {
  FTupleFirebase* tuple = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = tuple.one;
  FIRDatabaseReference* reader = tuple.two;

  __block BOOL ready = NO;

  NSDictionary* toSet = @{@"a" : @"a", @"b" : @"b", @"c" : @"c", @"d" : @"d", @"e" : @"e"};

  [writer setValue:toSet
      withCompletionBlock:^(NSError* err, FIRDatabaseReference* ref) {
        ready = YES;
      }];

  [self waitUntil:^BOOL {
    return ready;
  }];

  ready = NO;
  __block int count = 0;

  [[reader queryLimitedToLast:5]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               count++;
               if (count == 1) {
                 NSDictionary* val = [snapshot value];
                 NSDictionary* expected =
                     @{@"a" : @"a", @"b" : @"b", @"c" : @"c", @"d" : @"d", @"e" : @"e"};
                 XCTAssertTrue([val isEqualToDictionary:expected],
                               @"First callback, expect all the data");
                 [[writer child:@"c"] removeValue];
               } else {
                 XCTAssertTrue(count == 2, @"Should only get called twice");
                 NSDictionary* val = [snapshot value];
                 NSDictionary* expected = @{@"a" : @"a", @"b" : @"b", @"d" : @"d", @"e" : @"e"};
                 XCTAssertTrue([val isEqualToDictionary:expected],
                               @"Second callback, expect all the remaining data");
                 ready = YES;
               }
             }];

  [self waitUntil:^BOOL {
    return ready;
  }];
}

- (void)testEndingAtNameReturnsCorrectChildren {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSDictionary* toSet = @{
    @"a" : @"a",
    @"b" : @"b",
    @"c" : @"c",
    @"d" : @"d",
    @"e" : @"e",
    @"f" : @"f",
    @"g" : @"g",
    @"h" : @"h"
  };

  [self waitForCompletionOf:ref setValue:toSet];

  __block NSDictionary* snap = nil;
  __block BOOL done = NO;
  FIRDatabaseQuery* query =
      [[[ref queryOrderedByPriority] queryEndingAtValue:nil childKey:@"f"] queryLimitedToLast:5];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  snap = [snapshot value];
                  done = YES;
                }];

  [self waitUntil:^BOOL {
    return done;
  }];

  NSDictionary* expected = @{@"b" : @"b", @"c" : @"c", @"d" : @"d", @"e" : @"e", @"f" : @"f"};
  XCTAssertTrue([snap isEqualToDictionary:expected], @"Expected 5 elements, ending at f");
}

- (void)testEndingBeforeNameReturnsCorrectChildren {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSDictionary* toSet = @{
    @"a" : @"a",
    @"b" : @"b",
    @"c" : @"c",
    @"d" : @"d",
    @"e" : @"e",
    @"f" : @"f",
    @"g" : @"g",
    @"h" : @"h"
  };

  [self waitForCompletionOf:ref setValue:toSet];

  __block NSDictionary* snap = nil;
  __block BOOL done = NO;
  FIRDatabaseQuery* query = [[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"f"]
      queryLimitedToLast:5];
  [query observeEventType:FIRDataEventTypeValue
                withBlock:^(FIRDataSnapshot* snapshot) {
                  snap = [snapshot value];
                  done = YES;
                }];

  [self waitUntil:^BOOL {
    return done;
  }];

  NSDictionary* expected = @{@"a" : @"a", @"b" : @"b", @"c" : @"c", @"d" : @"d", @"e" : @"e"};
  XCTAssertTrue([snap isEqualToDictionary:expected], @"Expected 5 elements, ending at e");
}

- (void)testListenForChildAddedWithLimitEnsureEventsFireProperly {
  FTupleFirebase* refs = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = refs.one;
  FIRDatabaseReference* reader = refs.two;

  __block BOOL done = NO;

  NSDictionary* toSet =
      @{@"a" : @1, @"b" : @"b", @"c" : @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}}};
  [writer setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        done = YES;
      }];

  WAIT_FOR(done);

  __block int count = 0;
  [[reader queryLimitedToLast:3]
      observeEventType:FIRDataEventTypeChildAdded
             withBlock:^(FIRDataSnapshot* snapshot) {
               count++;
               if (count == 1) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"a"], @"Got first child");
                 XCTAssertTrue([snapshot.value isEqualToNumber:@1], @"Got correct value");
               } else if (count == 2) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"b"], @"Got second child");
                 XCTAssertTrue([snapshot.value isEqualToString:@"b"], @"got correct value");
               } else if (count == 3) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"c"], @"Got third child");
                 NSDictionary* expected = @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}};
                 XCTAssertTrue([snapshot.value isEqualToDictionary:expected], @"Got deep object");
               } else {
                 XCTFail(@"wrong event count");
               }
             }];

  WAIT_FOR(count == 3);
}

#ifdef FLAKY_TEST
- (void)testListenForChildChangedWithLimitEnsureEventsFireProperly {
  FTupleFirebase* refs = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = refs.one;
  FIRDatabaseReference* reader = refs.two;

  __block BOOL done = NO;

  NSDictionary* toSet = @{@"a" : @"something", @"b" : @"we'll", @"c" : @"overwrite"};
  [writer setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        done = YES;
      }];

  WAIT_FOR(done);

  __block int count = 0;
  [reader
      observeEventType:FIRDataEventTypeChildChanged
             withBlock:^(FIRDataSnapshot* snapshot) {
               count++;
               if (count == 1) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"a"], @"Got first child");
                 XCTAssertTrue([snapshot.value isEqualToNumber:@1], @"Got correct value");
               } else if (count == 2) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"b"], @"Got second child");
                 XCTAssertTrue([snapshot.value isEqualToString:@"b"], @"got correct value");
               } else if (count == 3) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"c"], @"Got third child");
                 NSDictionary* expected = @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}};
                 XCTAssertTrue([snapshot.value isEqualToDictionary:expected], @"Got deep object");
               } else {
                 XCTFail(@"wrong event count");
               }
             }];
  toSet = @{@"a" : @1, @"b" : @"b", @"c" : @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}}};
  [writer setValue:toSet];

  WAIT_FOR(count == 3);
}
#endif

- (void)testListenForChildRemovedWithLimitEnsureEventsFireProperly {
  FTupleFirebase* refs = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = refs.one;
  FIRDatabaseReference* reader = refs.two;

  __block BOOL done = NO;

  NSDictionary* toSet =
      @{@"a" : @1, @"b" : @"b", @"c" : @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}}};
  [writer setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        done = YES;
      }];

  WAIT_FOR(done);

  __block int count = 0;
  [reader
      observeEventType:FIRDataEventTypeChildRemoved
             withBlock:^(FIRDataSnapshot* snapshot) {
               count++;
               if (count == 1) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"a"], @"Got first child");
                 XCTAssertTrue([snapshot.value isEqualToNumber:@1], @"Got correct value");
               } else if (count == 2) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"b"], @"Got second child");
                 XCTAssertTrue([snapshot.value isEqualToString:@"b"], @"got correct value");
               } else if (count == 3) {
                 XCTAssertTrue([snapshot.key isEqualToString:@"c"], @"Got third child");
                 NSDictionary* expected = @{@"deep" : @"path", @"of" : @{@"stuff" : @YES}};
                 XCTAssertTrue([snapshot.value isEqualToDictionary:expected], @"Got deep object");
               } else {
                 XCTFail(@"wrong event count");
               }
             }];

  done = NO;
  [reader observeEventType:FIRDataEventTypeValue
                 withBlock:^(FIRDataSnapshot* snapshot) {
                   // Load the data first
                   done = snapshot.value != [NSNull null] &&
                          [snapshot.value isEqualToDictionary:toSet];
                 }];

  WAIT_FOR(done);

  // Now do the removes
  [[writer child:@"a"] removeValue];
  [[writer child:@"b"] removeValue];
  [[writer child:@"c"] removeValue];

  WAIT_FOR(count == 3);
}

- (void)testQueriesBehaveProperlyAfterOnceCall {
  FTupleFirebase* refs = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = refs.one;
  FIRDatabaseReference* reader = refs.two;

  __block BOOL done = NO;
  NSDictionary* toSet = @{@"a" : @1, @"b" : @2, @"c" : @3, @"d" : @4};
  [writer setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        done = YES;
      }];

  WAIT_FOR(done);

  done = NO;
  [reader observeSingleEventOfType:FIRDataEventTypeValue
                         withBlock:^(FIRDataSnapshot* snapshot) {
                           done = YES;
                         }];

  WAIT_FOR(done);

  // Ok, now do some queries
  __block int startCount = 0;
  __block int defaultCount = 0;
  [[[reader queryOrderedByPriority] queryStartingAtValue:nil childKey:@"d"]
      observeEventType:FIRDataEventTypeChildAdded
             withBlock:^(FIRDataSnapshot* snapshot) {
               startCount++;
             }];

  [reader observeEventType:FIRDataEventTypeChildAdded
                 withBlock:^(FIRDataSnapshot* snapshot) {
                   defaultCount++;
                 }];

  [reader observeEventType:FIRDataEventTypeChildRemoved
                 withBlock:^(FIRDataSnapshot* snapshot) {
                   XCTFail(@"Should not remove any children");
                 }];

  WAIT_FOR(startCount == 1 && defaultCount == 4);
}

- (void)testIntegerKeysBehaveNumerically1 {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority] queryStartingAtValue:nil childKey:@"80"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{@"80" : @YES, @"550" : @YES, @"600" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumericallyWithStartAfter {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority] queryStartingAfterValue:nil childKey:@"80"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{@"550" : @YES, @"600" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumericallyWithStartAfterOverflow {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES,
    @"a" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority]
            queryStartingAfterValue:nil
                           childKey:[NSString stringWithFormat:@"%ld", (long)[FUtilities int32max]]]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{@"a" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumericallyWithEndBefore {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority] queryEndingBeforeValue:nil childKey:@"80"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{
                               @"1" : @YES,
                               @"6" : @YES,
                               @"8" : @YES,
                               @"50" : @YES,
                               @"70" : @YES
                             };
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumericallyWithEndBeforeWithUnderflow {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{@"1" : @YES};
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority]
            queryEndingBeforeValue:nil
                          childKey:[NSString stringWithFormat:@"%ld", (long)[FUtilities int32min]]]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             XCTAssertEqual(snapshot.value, [NSNull null]);
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumerically2 {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[ref queryOrderedByPriority] queryEndingAtValue:nil childKey:@"50"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected =
                                 @{@"1" : @YES,
                                   @"6" : @YES,
                                   @"8" : @YES,
                                   @"50" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumerically3 {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[[ref queryOrderedByPriority] queryStartingAtValue:nil
                                                    childKey:@"50"] queryEndingAtValue:nil
                                                                              childKey:@"80"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{@"50" : @YES, @"70" : @YES, @"80" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testIntegerKeysBehaveNumericallyWithStartAfterEndAt {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  NSDictionary* toSet = @{
    @"1" : @YES,
    @"50" : @YES,
    @"550" : @YES,
    @"6" : @YES,
    @"600" : @YES,
    @"70" : @YES,
    @"8" : @YES,
    @"80" : @YES
  };
  __block BOOL done = NO;
  [ref setValue:toSet
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        [[[[ref queryOrderedByPriority] queryStartingAfterValue:nil
                                                       childKey:@"50"] queryEndingAtValue:nil
                                                                                 childKey:@"80"]
            observeSingleEventOfType:FIRDataEventTypeValue
                           withBlock:^(FIRDataSnapshot* snapshot) {
                             NSDictionary* expected = @{@"70" : @YES, @"80" : @YES};
                             XCTAssertTrue([snapshot.value isEqualToDictionary:expected],
                                           @"Got correct result.");
                             done = YES;
                           }];
      }];
  WAIT_FOR(done);
}

- (void)testItemsPulledIntoLimitCorrectly {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  NSMutableArray* snaps = [[NSMutableArray alloc] init];

  // Just so everything is cached locally.
  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot){

              }];

  [[ref queryLimitedToLast:2] observeEventType:FIRDataEventTypeValue
                                     withBlock:^(FIRDataSnapshot* snapshot) {
                                       id val = [snapshot value];
                                       [snaps addObject:val];
                                     }];

  [ref setValue:@{
    @"a" : @{@".value" : @1, @".priority" : @1},
    @"b" : @{@".value" : @2, @".priority" : @2},
    @"c" : @{@".value" : @3, @".priority" : @3}
  }];

  __block BOOL ready = NO;
  [[ref child:@"b"] setValue:[NSNull null]
         withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
           ready = YES;
         }];

  WAIT_FOR(ready);

  NSArray* expected = @[ @{@"b" : @2, @"c" : @3}, @{@"a" : @1, @"c" : @3} ];
  XCTAssertEqualObjects(snaps, expected, @"Incorrect snapshots.");
}

- (void)testChildChangedCausesChildRemovedEvent {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  [[ref child:@"l/a"] setValue:@"1" andPriority:@"a"];
  [[ref child:@"l/b"] setValue:@"2" andPriority:@"b"];
  FIRDatabaseQuery* query = [[[[ref child:@"l"] queryOrderedByPriority] queryStartingAtValue:@"b"]
      queryEndingAtValue:@"d"];
  __block BOOL removed = NO;
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  XCTAssertEqualObjects(snapshot.value, @"2", @"Incorrect snapshot");
                  removed = YES;
                }];

  [[ref child:@"l/b"] setValue:@"4" andPriority:@"a"];

  WAIT_FOR(removed);
}

- (void)testChildChangedCausesChildRemovedEventWithStartAfter {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  [[ref child:@"l/a"] setValue:@"1" andPriority:@"a"];
  [[ref child:@"l/b"] setValue:@"2" andPriority:@"b"];
  FIRDatabaseQuery* query = [[[[ref child:@"l"] queryOrderedByPriority]
      queryStartingAfterValue:@"a"] queryEndingAtValue:@"d"];
  __block BOOL removed = NO;
  [query observeEventType:FIRDataEventTypeChildRemoved
                withBlock:^(FIRDataSnapshot* snapshot) {
                  XCTAssertEqualObjects(snapshot.value, @"2", @"Incorrect snapshot");
                  removed = YES;
                }];

  [[ref child:@"l/b"] setValue:@"4" andPriority:@"a"];

  WAIT_FOR(removed);
}

- (void)testQueryHasRef {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabaseQuery* query = [ref queryOrderedByKey];
  XCTAssertEqualObjects([query.ref path], [ref path], @"Should have same path");
}

- (void)testQuerySnapshotChildrenRespectDefaultOrdering {
  FTupleFirebase* pair = [FTestHelpers getRandomNodePair];
  FIRDatabaseReference* writer = pair.one;
  FIRDatabaseReference* reader = pair.two;
  __block BOOL done = NO;

  NSDictionary* list = @{
    @"a" : @{
      @"thisvaluefirst" : @{@".value" : @true, @".priority" : @1},
      @"name" : @{@".value" : @"Michael", @".priority" : @2},
      @"thisvaluelast" : @{@".value" : @true, @".priority" : @3},
    },
    @"b" : @{
      @"thisvaluefirst" : @{@".value" : @true},
      @"name" : @{@".value" : @"Rob", @".priority" : @2},
      @"thisvaluelast" : @{@".value" : @true, @".priority" : @3},
    },
    @"c" : @{
      @"thisvaluefirst" : @{@".value" : @true, @".priority" : @1},
      @"name" : @{@".value" : @"Jonny", @".priority" : @2},
      @"thisvaluelast" : @{@".value" : @true, @".priority" : @"somestring"},
    }
  };

  [writer setValue:list
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        done = YES;
      }];
  WAIT_FOR(done);

  done = NO;
  [[reader queryOrderedByChild:@"name"]
      observeSingleEventOfType:FIRDataEventTypeValue
                     withBlock:^(FIRDataSnapshot* snapshot) {
                       NSArray* expectedKeys = @[ @"thisvaluefirst", @"name", @"thisvaluelast" ];
                       NSArray* expectedNames = @[ @"Jonny", @"Michael", @"Rob" ];

                       // Validate that snap.child() resets order to default for child snaps
                       NSMutableArray* orderedKeys = [[NSMutableArray alloc] init];
                       for (FIRDataSnapshot* childSnap in [snapshot childSnapshotForPath:@"b"]
                                .children) {
                         [orderedKeys addObject:childSnap.key];
                       }
                       XCTAssertEqualObjects(expectedKeys, orderedKeys,
                                             @"Should have matching ordered lists of keys");

                       // Validate that snap.forEach() resets ordering to default for child snaps
                       NSMutableArray* orderedNames = [[NSMutableArray alloc] init];
                       for (FIRDataSnapshot* childSnap in snapshot.children) {
                         [orderedNames addObject:[childSnap childSnapshotForPath:@"name"].value];
                         [orderedKeys removeAllObjects];
                         for (FIRDataSnapshot* grandchildSnap in childSnap.children) {
                           [orderedKeys addObject:grandchildSnap.key];
                         }
                         XCTAssertEqualObjects(expectedKeys, orderedKeys,
                                               @"Should have matching ordered lists of keys");
                       }
                       XCTAssertEqualObjects(expectedNames, orderedNames,
                                             @"Should have matching ordered lists of names");

                       done = YES;
                     }];
  WAIT_FOR(done);
}

- (void)testAddingListensForTheSamePathDoesNotCheckFail {
  // This bug manifests itself if there's a hierarchy of query listener, default listener and
  // one-time listener underneath. In Java implementation, during one-time listener registration,
  // sync-tree traversal stopped as soon as it found a complete server cache (this is the case for
  // not indexed query view). The problem is that the same traversal was looking for a ancestor
  // default view, and the early exit prevented from finding the default listener above the one-time
  // listener. Event removal code path wasn't removing the listener because it stopped as soon as it
  // found the default view. This left the zombie one-time listener and check failed on the second
  // attempt to create a listener for the same path (asana#61028598952586).

  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [[ref child:@"child"] setValue:@{@"name" : @"John"}];
  [[[ref queryOrderedByChild:@"name"] queryEqualToValue:@"John"]
      observeEventType:FIRDataEventTypeValue
             withBlock:^(FIRDataSnapshot* snapshot) {
               done = YES;
             }];
  WAIT_FOR(done);

  done = NO;
  [[[ref child:@"child"] child:@"favoriteToy"]
      observeSingleEventOfType:FIRDataEventTypeValue
                     withBlock:^(FIRDataSnapshot* snapshot) {
                       done = YES;
                     }];
  WAIT_FOR(done);

  done = NO;
  [[[ref child:@"child"] child:@"favoriteToy"]
      observeSingleEventOfType:FIRDataEventTypeValue
                     withBlock:^(FIRDataSnapshot* snapshot) {
                       done = YES;
                     }];
  WAIT_FOR(done);
}

- (void)testGetForEmptyNodeReturnsNull {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssert([snapshot.value isEqual:[NSNull null]]);
    done = YES;
  }];

  WAIT_FOR(done);
}

- (void)testGetInvokesCallbacksOnMainThread {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssert([[NSThread currentThread] isMainThread]);
    done = YES;
  }];

  WAIT_FOR(done);
}

- (void)testGetForNodeReturnsCorrectValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [ref setValue:@1
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
        done = YES;
      }];

  WAIT_FOR(done);
  done = NO;

  [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssertNil(err);
    XCTAssertEqualObjects([snapshot value], @1);
    done = YES;
  }];

  WAIT_FOR(done);
}

- (void)testGetForChildReturnsCorrectValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [ref setValue:@{@"a" : @1, @"b" : @2}
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
        done = YES;
      }];

  WAIT_FOR(done);

  [[ref child:@"a"] getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssertNil(err);
    XCTAssertEqualObjects([snapshot value], @1);
    done = YES;
  }];
}

- (void)testGetForParentReturnsCorrectValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];

  __block BOOL done = NO;

  [[ref child:@"a"] setValue:@1
         withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
           XCTAssertNil(error);
           done = YES;
         }];

  WAIT_FOR(done);

  [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssertNil(err);
    XCTAssertEqualObjects([snapshot value], @{@"a" : @1});
    done = YES;
  }];
}

- (void)testGetForNodeWithPendingWritesReturnsCorrectValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabase* db = [ref database];

  __block BOOL done = NO;

  @try {
    [db goOffline];
    [ref setValue:@42];

    [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNil(err);
      XCTAssertEqualObjects(snapshot.value, @42);
      done = YES;
    }];

    WAIT_FOR(done);
  } @finally {
    [db goOnline];
  }
}

- (void)testGetForChildOfPendingWritesReturnsCorrectValue {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabase* db = [ref database];

  __block BOOL done = NO;

  @try {
    [db goOffline];
    [ref setValue:@{@"foo" : @"bar"}];

    [[ref child:@"foo"] getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNil(err);
      XCTAssertEqualObjects(snapshot.value, @"bar");
      done = YES;
    }];

    WAIT_FOR(done);
  } @finally {
    [db goOnline];
  }
}

- (void)testGetReturnsErrorForUnknownNodeWhenOffline {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabase* db = [ref database];

  __block BOOL done = NO;

  @try {
    [db goOffline];

    [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNotNil(err);
      XCTAssertEqualObjects([err localizedFailureReason], kPersistentConnectionOffline);
      done = YES;
    }];

    WAIT_FOR(done);
  } @finally {
    [db goOnline];
  }
}

- (void)testGetProbesInMemoryCacheForActiveListenerWhenOffline {
  FIRDatabaseReference* ref = [FTestHelpers getRandomNode];
  FIRDatabase* db = [ref database];

  __block BOOL done = NO;

  [ref observeEventType:FIRDataEventTypeValue
              withBlock:^(FIRDataSnapshot* snapshot) {
                id value = [snapshot value];
                if (value != nil && [value isEqualToNumber:@42]) {
                  done = YES;
                }
              }];

  [ref setValue:@42];

  WAIT_FOR(done);
  done = NO;

  @try {
    [db goOffline];

    [ref getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNil(err);
      XCTAssertEqualObjects(snapshot.value, @42);
      done = YES;
    }];

    WAIT_FOR(done);
  } @finally {
    [db goOnline];
  }
}

- (FIRDatabase*)databaseForURL:(NSString*)url name:(NSString*)name {
  NSString* defaultDatabaseURL = [NSString stringWithFormat:@"url:%@", self.databaseURL];
  if ([url isEqualToString:self.databaseURL] && [name isEqualToString:defaultDatabaseURL]) {
    // Use the default app for the default URL to avoid getting out of sync with FRepoManager
    // when calling ensureRepo during tests that don't create their own FIRFakeApp.
    return [FTestHelpers defaultDatabase];
  } else {
    id app = [[FIRFakeApp alloc] initWithName:name URL:url];
    return [FIRDatabase databaseForApp:app];
  }
}

- (void)testGetReturnsPersistenceCachedValueWhenOffline {
  FIRDatabase* db = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];
  FIRDatabase* db2 = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];

  [db2 setPersistenceEnabled:true];

  NSString* uuidPath = [[NSUUID UUID] UUIDString];

  FIRDatabaseReference* writeRef = [db referenceWithPath:uuidPath];
  FIRDatabaseReference* readRef = [db2 referenceWithPath:uuidPath];

  __block BOOL done = NO;

  [writeRef setValue:@42
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
        done = YES;
      }];

  WAIT_FOR(done);
  done = NO;

  [readRef observeEventType:FIRDataEventTypeValue
                  withBlock:^(FIRDataSnapshot* snapshot) {
                    XCTAssertEqualObjects(snapshot.value, @42);
                    done = YES;
                  }];

  WAIT_FOR(done);

  [readRef removeAllObservers];

  @try {
    [db2 goOffline];
    [readRef getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNil(err);
      XCTAssertEqualObjects(snapshot.value, @42);
      done = YES;
    }];
  } @finally {
    [db2 goOnline];
  }
}

- (void)testGetThrowsExceptionNewNodeWhenOfflineWithPersistenceEnabled {
  FIRDatabase* db = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];
  [db setPersistenceEnabled:true];

  NSString* uuidPath = [[NSUUID UUID] UUIDString];
  FIRDatabaseReference* readRef = [db referenceWithPath:uuidPath];

  [db goOffline];
  @try {
    __block BOOL done = NO;
    [readRef getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
      XCTAssertNotNil(err);
      XCTAssertEqualObjects([err localizedFailureReason], kPersistentConnectionOffline);
      done = YES;
    }];
    WAIT_FOR(done);
  } @finally {
    [db goOnline];
  }
}

- (void)testGetUpdatesPersistenceCacheWhenEnabled {
  FIRDatabase* db = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];
  FIRDatabase* db2 = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];

  [db2 setPersistenceEnabled:true];

  NSString* uuidPath = [[NSUUID UUID] UUIDString];

  FIRDatabaseReference* writeRef = [db referenceWithPath:uuidPath];
  FIRDatabaseReference* readRef = [db2 referenceWithPath:uuidPath];

  __block BOOL done = NO;

  [writeRef setValue:@42
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
        done = YES;
      }];

  WAIT_FOR(done);
  done = NO;

  [readRef getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssertNil(err);
    XCTAssertEqualObjects(snapshot.value, @42);
    done = YES;
  }];

  WAIT_FOR(done);
  done = NO;

  @try {
    [db2 goOffline];

    [readRef observeEventType:FIRDataEventTypeValue
                    withBlock:^(FIRDataSnapshot* snapshot) {
                      XCTAssertEqualObjects(snapshot.value, @42);
                      done = YES;
                    }];

    WAIT_FOR(done);
  } @finally {
    [db2 goOnline];
  }
}

- (void)testGetSkipsPersistenceCacheWhenOnline {
  FIRDatabase* db = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];
  FIRDatabase* db2 = [self databaseForURL:self.databaseURL name:[[NSUUID UUID] UUIDString]];

  [db2 setPersistenceEnabled:true];

  NSString* uuidPath = [[NSUUID UUID] UUIDString];

  FIRDatabaseReference* writeRef = [db referenceWithPath:uuidPath];
  FIRDatabaseReference* readRef = [db2 referenceWithPath:uuidPath];

  __block BOOL done = NO;

  [writeRef setValue:@42
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
      }];

  [readRef observeEventType:FIRDataEventTypeValue
                  withBlock:^(FIRDataSnapshot* snapshot) {
                    XCTAssertEqualObjects(snapshot.value, @42);
                    done = YES;
                  }];

  WAIT_FOR(done);
  done = NO;

  [readRef removeAllObservers];

  [writeRef setValue:@43
      withCompletionBlock:^(NSError* error, FIRDatabaseReference* ref) {
        XCTAssertNil(error);
        done = YES;
      }];

  WAIT_FOR(done);
  done = NO;

  [readRef getDataWithCompletionBlock:^(NSError* err, FIRDataSnapshot* snapshot) {
    XCTAssertNil(err);
    XCTAssertEqualObjects(snapshot.value, @43);
    done = YES;
  }];

  WAIT_FOR(done);
}

@end