Files
firebase-ios-sdk/Firestore/Example/Tests/API/FIRQuerySnapshotTests.mm
T
Gil c2a3d04e6d Port FSTDocumentSet to C++ (#2608)
* Replace FSTDocumentSet's guts with immutable::SortedSet.

* Port FSTDocumentSet to C++ DocumentSet.

* Replace implementation of FSTDocumentSet with C++ DocumentSet

* Update all usages of FSTDocumentSet

* Remove FSTDocumentSet

* Add Objective-C++ guard

* Use absl::c_equal instead of hand-rolling it.

* Split SortedContainer out of SortedMapBase

... and use it

* Fix NSNotFound -> npos

* Remove = default constructors that aren't true.

These are implicitly deleted because DocumentSet is a member and isn't
default constructible.

* Fix usage of FSTQuery.limit
2019-03-25 11:04:21 -07:00

152 lines
6.5 KiB
Plaintext

/*
* 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 <FirebaseFirestore/FIRQuerySnapshot.h>
#import <XCTest/XCTest.h>
#include <utility>
#include <vector>
#import "Firestore/Example/Tests/API/FSTAPIHelpers.h"
#import "Firestore/Example/Tests/Util/FSTHelpers.h"
#import "Firestore/Source/API/FIRDocumentChange+Internal.h"
#import "Firestore/Source/API/FIRDocumentSnapshot+Internal.h"
#import "Firestore/Source/API/FIRFirestore+Internal.h"
#import "Firestore/Source/API/FIRQuerySnapshot+Internal.h"
#import "Firestore/Source/API/FIRSnapshotMetadata+Internal.h"
#import "Firestore/Source/Model/FSTDocument.h"
#include "Firestore/core/src/firebase/firestore/core/view_snapshot.h"
#include "Firestore/core/src/firebase/firestore/model/document_set.h"
#include "Firestore/core/src/firebase/firestore/util/string_apple.h"
namespace util = firebase::firestore::util;
using firebase::firestore::core::DocumentViewChange;
using firebase::firestore::core::ViewSnapshot;
using firebase::firestore::model::DocumentKeySet;
using firebase::firestore::model::DocumentSet;
NS_ASSUME_NONNULL_BEGIN
@interface FIRDocumentChange ()
// Expose initializer for testing.
- (instancetype)initWithType:(FIRDocumentChangeType)type
document:(FIRQueryDocumentSnapshot *)document
oldIndex:(NSUInteger)oldIndex
newIndex:(NSUInteger)newIndex;
@end
@interface FIRQuerySnapshotTests : XCTestCase
@end
@implementation FIRQuerySnapshotTests
- (void)testEquals {
FIRQuerySnapshot *foo = FSTTestQuerySnapshot("foo", @{}, @{@"a" : @{@"a" : @1}}, true, false);
FIRQuerySnapshot *fooDup = FSTTestQuerySnapshot("foo", @{}, @{@"a" : @{@"a" : @1}}, true, false);
FIRQuerySnapshot *differentPath =
FSTTestQuerySnapshot("bar", @{}, @{@"a" : @{@"a" : @1}}, true, false);
FIRQuerySnapshot *differentDoc =
FSTTestQuerySnapshot("foo", @{@"a" : @{@"b" : @1}}, @{}, true, false);
FIRQuerySnapshot *noPendingWrites =
FSTTestQuerySnapshot("foo", @{}, @{@"a" : @{@"a" : @1}}, false, false);
FIRQuerySnapshot *fromCache =
FSTTestQuerySnapshot("foo", @{}, @{@"a" : @{@"a" : @1}}, true, true);
XCTAssertEqualObjects(foo, fooDup);
XCTAssertNotEqualObjects(foo, differentPath);
XCTAssertNotEqualObjects(foo, differentDoc);
XCTAssertNotEqualObjects(foo, noPendingWrites);
XCTAssertNotEqualObjects(foo, fromCache);
XCTAssertEqual([foo hash], [fooDup hash]);
XCTAssertNotEqual([foo hash], [differentPath hash]);
XCTAssertNotEqual([foo hash], [differentDoc hash]);
XCTAssertNotEqual([foo hash], [noPendingWrites hash]);
XCTAssertNotEqual([foo hash], [fromCache hash]);
}
- (void)testIncludeMetadataChanges {
FSTDocument *doc1Old = FSTTestDoc("foo/bar", 1, @{@"a" : @"b"}, FSTDocumentStateLocalMutations);
FSTDocument *doc1New = FSTTestDoc("foo/bar", 1, @{@"a" : @"b"}, FSTDocumentStateSynced);
FSTDocument *doc2Old = FSTTestDoc("foo/baz", 1, @{@"a" : @"b"}, FSTDocumentStateSynced);
FSTDocument *doc2New = FSTTestDoc("foo/baz", 1, @{@"a" : @"c"}, FSTDocumentStateSynced);
DocumentSet oldDocuments = FSTTestDocSet(FSTDocumentComparatorByKey, @[ doc1Old, doc2Old ]);
DocumentSet newDocuments = FSTTestDocSet(FSTDocumentComparatorByKey, @[ doc2New, doc2New ]);
std::vector<DocumentViewChange> documentChanges{
DocumentViewChange{doc1New, DocumentViewChange::Type::kMetadata},
DocumentViewChange{doc2New, DocumentViewChange::Type::kModified},
};
Firestore *firestore = FSTTestFirestore().wrapped;
FSTQuery *query = FSTTestQuery("foo");
ViewSnapshot viewSnapshot{query,
newDocuments,
oldDocuments,
std::move(documentChanges),
/*mutated_keys=*/DocumentKeySet{},
/*from_cache=*/false,
/*sync_state_changed=*/true,
/*excludes_metadata_changes=*/false};
SnapshotMetadata metadata(/*pending_writes=*/false, /*from_cache=*/false);
FIRQuerySnapshot *snapshot = [[FIRQuerySnapshot alloc] initWithFirestore:firestore
originalQuery:query
snapshot:std::move(viewSnapshot)
metadata:std::move(metadata)];
FIRQueryDocumentSnapshot *doc1Snap =
[[FIRQueryDocumentSnapshot alloc] initWithFirestore:firestore
documentKey:doc1New.key
document:doc1New
fromCache:false
hasPendingWrites:false];
FIRQueryDocumentSnapshot *doc2Snap =
[[FIRQueryDocumentSnapshot alloc] initWithFirestore:firestore
documentKey:doc2New.key
document:doc2New
fromCache:false
hasPendingWrites:false];
NSArray<FIRDocumentChange *> *changesWithoutMetadata = @[
[[FIRDocumentChange alloc] initWithType:FIRDocumentChangeTypeModified
document:doc2Snap
oldIndex:1
newIndex:1],
];
XCTAssertEqualObjects(snapshot.documentChanges, changesWithoutMetadata);
NSArray<FIRDocumentChange *> *changesWithMetadata = @[
[[FIRDocumentChange alloc] initWithType:FIRDocumentChangeTypeModified
document:doc1Snap
oldIndex:0
newIndex:0],
[[FIRDocumentChange alloc] initWithType:FIRDocumentChangeTypeModified
document:doc2Snap
oldIndex:1
newIndex:1],
];
XCTAssertEqualObjects([snapshot documentChangesWithIncludeMetadataChanges:YES],
changesWithMetadata);
}
@end
NS_ASSUME_NONNULL_END