mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-09 22:36:01 +05:45
Michael Lehenbauer
a3c5064797
This is safer and better matches our design (and the other platforms), since classes like api::DocumentReference will have references to api::Firsetore but the lifetime is controlled by the developer, not us.
131 lines
5.8 KiB
Plaintext
131 lines
5.8 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 <memory>
|
|
#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::api::DocumentChange;
|
|
using firebase::firestore::api::DocumentSnapshot;
|
|
using firebase::firestore::api::Firestore;
|
|
using firebase::firestore::api::SnapshotMetadata;
|
|
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 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),
|
|
};
|
|
|
|
std::shared_ptr<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)];
|
|
|
|
DocumentSnapshot doc1Snap(firestore, doc1New.key, doc1New, SnapshotMetadata());
|
|
DocumentSnapshot doc2Snap(firestore, doc2New.key, doc2New, SnapshotMetadata());
|
|
|
|
NSArray<FIRDocumentChange *> *changesWithoutMetadata = @[
|
|
[[FIRDocumentChange alloc]
|
|
initWithDocumentChange:DocumentChange(DocumentChange::Type::Modified, doc2Snap,
|
|
/*old_index=*/1, /*new_index=*/1)],
|
|
];
|
|
XCTAssertEqualObjects(snapshot.documentChanges, changesWithoutMetadata);
|
|
|
|
NSArray<FIRDocumentChange *> *changesWithMetadata = @[
|
|
[[FIRDocumentChange alloc]
|
|
initWithDocumentChange:DocumentChange(DocumentChange::Type::Modified, doc1Snap,
|
|
/*old_index=*/0, /*new_index=*/0)],
|
|
[[FIRDocumentChange alloc]
|
|
initWithDocumentChange:DocumentChange(DocumentChange::Type::Modified, doc2Snap,
|
|
/*old_index=*/1, /*new_index=*/1)],
|
|
];
|
|
XCTAssertEqualObjects([snapshot documentChangesWithIncludeMetadataChanges:YES],
|
|
changesWithMetadata);
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|