Files
firebase-ios-sdk/Firestore/Example/Tests/Model/FSTDocumentSetTests.m
T
Gil 4fbc96904f Use fully qualified imports in Firestore (#467)
This simplifies the import process back into google3 and allows us to add additional directories to the project without needing to update the project files for each directory we add. This shows up most clearly in the change to Firestore/Example/Firestore.xcodeproj/project.pbxproj: this no longer needs to list essentially every directory in the project as a header search path.

* Clang-format configuration for C++ files
* Add support C++-only sources to the podspec
* Podspec support for fully qualified imports
* xcodeproj changes for fully-qualified imports
* Use fully-qualified imports in Firestore sources
2017-11-16 20:52:01 -08:00

143 lines
4.8 KiB
Objective-C

/*
* 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 "Firestore/Source/Model/FSTDocumentSet.h"
#import <XCTest/XCTest.h>
#import "Firestore/Source/Model/FSTDocument.h"
#import "Firestore/Example/Tests/Util/FSTHelpers.h"
NS_ASSUME_NONNULL_BEGIN
@interface FSTDocumentSetTests : XCTestCase
@end
@implementation FSTDocumentSetTests {
NSComparator _comp;
FSTDocument *_doc1;
FSTDocument *_doc2;
FSTDocument *_doc3;
}
- (void)setUp {
[super setUp];
_comp = FSTTestDocComparator(@"sort");
_doc1 = FSTTestDoc(@"docs/1", 0, @{ @"sort" : @2 }, NO);
_doc2 = FSTTestDoc(@"docs/2", 0, @{ @"sort" : @3 }, NO);
_doc3 = FSTTestDoc(@"docs/3", 0, @{ @"sort" : @1 }, NO);
}
- (void)testCount {
XCTAssertEqual([FSTTestDocSet(_comp, @[]) count], 0);
XCTAssertEqual([FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]) count], 3);
}
- (void)testHasKey {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2 ]);
XCTAssertTrue([set containsKey:_doc1.key]);
XCTAssertTrue([set containsKey:_doc2.key]);
XCTAssertFalse([set containsKey:_doc3.key]);
}
- (void)testDocumentForKey {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2 ]);
XCTAssertEqualObjects([set documentForKey:_doc1.key], _doc1);
XCTAssertEqualObjects([set documentForKey:_doc2.key], _doc2);
XCTAssertNil([set documentForKey:_doc3.key]);
}
- (void)testFirstAndLastDocument {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[]);
XCTAssertNil([set firstDocument]);
XCTAssertNil([set lastDocument]);
set = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
XCTAssertEqualObjects([set firstDocument], _doc3);
XCTAssertEqualObjects([set lastDocument], _doc2);
}
- (void)testKeepsDocumentsInTheRightOrder {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
XCTAssertEqualObjects([[set documentEnumerator] allObjects], (@[ _doc3, _doc1, _doc2 ]));
}
- (void)testPredecessorDocumentForKey {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
XCTAssertNil([set predecessorDocumentForKey:_doc3.key]);
XCTAssertEqualObjects([set predecessorDocumentForKey:_doc1.key], _doc3);
XCTAssertEqualObjects([set predecessorDocumentForKey:_doc2.key], _doc1);
}
- (void)testDeletes {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
FSTDocumentSet *setWithoutDoc1 = [set documentSetByRemovingKey:_doc1.key];
XCTAssertEqualObjects([[setWithoutDoc1 documentEnumerator] allObjects], (@[ _doc3, _doc2 ]));
XCTAssertEqual([setWithoutDoc1 count], 2);
// Original remains unchanged
XCTAssertEqualObjects([[set documentEnumerator] allObjects], (@[ _doc3, _doc1, _doc2 ]));
FSTDocumentSet *setWithoutDoc3 = [setWithoutDoc1 documentSetByRemovingKey:_doc3.key];
XCTAssertEqualObjects([[setWithoutDoc3 documentEnumerator] allObjects], (@[ _doc2 ]));
XCTAssertEqual([setWithoutDoc3 count], 1);
}
- (void)testUpdates {
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
FSTDocument *doc2Prime = FSTTestDoc(@"docs/2", 0, @{ @"sort" : @9 }, NO);
set = [set documentSetByAddingDocument:doc2Prime];
XCTAssertEqual([set count], 3);
XCTAssertEqualObjects([set documentForKey:doc2Prime.key], doc2Prime);
XCTAssertEqualObjects([[set documentEnumerator] allObjects], (@[ _doc3, _doc1, doc2Prime ]));
}
- (void)testAddsDocsWithEqualComparisonValues {
FSTDocument *doc4 = FSTTestDoc(@"docs/4", 0, @{ @"sort" : @2 }, NO);
FSTDocumentSet *set = FSTTestDocSet(_comp, @[ _doc1, doc4 ]);
XCTAssertEqualObjects([[set documentEnumerator] allObjects], (@[ _doc1, doc4 ]));
}
- (void)testIsEqual {
FSTDocumentSet *set1 = FSTTestDocSet(FSTDocumentComparatorByKey, @[ _doc1, _doc2, _doc3 ]);
FSTDocumentSet *set2 = FSTTestDocSet(FSTDocumentComparatorByKey, @[ _doc1, _doc2, _doc3 ]);
XCTAssertEqualObjects(set1, set1);
XCTAssertEqualObjects(set1, set2);
XCTAssertNotEqualObjects(set1, nil);
FSTDocumentSet *sortedSet1 = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
FSTDocumentSet *sortedSet2 = FSTTestDocSet(_comp, @[ _doc1, _doc2, _doc3 ]);
XCTAssertEqualObjects(sortedSet1, sortedSet1);
XCTAssertEqualObjects(sortedSet1, sortedSet2);
XCTAssertNotEqualObjects(sortedSet1, nil);
FSTDocumentSet *shortSet = FSTTestDocSet(FSTDocumentComparatorByKey, @[ _doc1, _doc2 ]);
XCTAssertNotEqualObjects(set1, shortSet);
XCTAssertNotEqualObjects(set1, sortedSet1);
}
@end
NS_ASSUME_NONNULL_END