mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-10 22:45:59 +05:45
Greg Soltis
5f5f808258
* Implement schema versions * Style fixes * newlines, copyrights, assumptions * Fix nullability * Raw ptr -> shared_ptr * kVersionTableGlobal -> kVersionGlobalTable * Drop utils, move into static methods * Drop extra include * Add a few more comments * Move version constant into migrations file * formatting? * Fix comment * Split add and update queryData * Work on adding targetCount * More work on count * Using shared_ptr * Implement count for query cache * use quotes * Add cast * Styling * Revert year bump in copyright * Add adversarial key to migration test * Add comment * Fix style
105 lines
3.5 KiB
Plaintext
105 lines
3.5 KiB
Plaintext
/*
|
|
* Copyright 2018 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 <XCTest/XCTest.h>
|
|
#include <leveldb/db.h>
|
|
|
|
#import "Firestore/Protos/objc/firestore/local/Target.pbobjc.h"
|
|
#import "Firestore/Source/Local/FSTLevelDBKey.h"
|
|
#import "Firestore/Source/Local/FSTLevelDBMigrations.h"
|
|
#import "Firestore/Source/Local/FSTLevelDBQueryCache.h"
|
|
#import "Firestore/Source/Local/FSTWriteGroup.h"
|
|
|
|
#include "Firestore/core/src/firebase/firestore/util/ordered_code.h"
|
|
|
|
#import "Firestore/Example/Tests/Local/FSTPersistenceTestHelpers.h"
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
using firebase::firestore::util::OrderedCode;
|
|
using leveldb::DB;
|
|
using leveldb::Options;
|
|
using leveldb::Status;
|
|
|
|
@interface FSTLevelDBMigrationsTests : XCTestCase
|
|
@end
|
|
|
|
@implementation FSTLevelDBMigrationsTests {
|
|
std::shared_ptr<DB> _db;
|
|
}
|
|
|
|
- (void)setUp {
|
|
Options options;
|
|
options.error_if_exists = true;
|
|
options.create_if_missing = true;
|
|
|
|
NSString *dir = [FSTPersistenceTestHelpers levelDBDir];
|
|
DB *db;
|
|
Status status = DB::Open(options, [dir UTF8String], &db);
|
|
XCTAssert(status.ok(), @"Failed to create db: %s", status.ToString().c_str());
|
|
_db.reset(db);
|
|
}
|
|
|
|
- (void)tearDown {
|
|
_db.reset();
|
|
}
|
|
|
|
- (void)testAddsTargetGlobal {
|
|
FSTPBTargetGlobal *metadata = [FSTLevelDBQueryCache readTargetMetadataFromDB:_db];
|
|
XCTAssertNil(metadata, @"Not expecting metadata yet, we should have an empty db");
|
|
[FSTLevelDBMigrations runMigrationsOnDB:_db];
|
|
metadata = [FSTLevelDBQueryCache readTargetMetadataFromDB:_db];
|
|
XCTAssertNotNil(metadata, @"Migrations should have added the metadata");
|
|
}
|
|
|
|
- (void)testSetsVersionNumber {
|
|
FSTLevelDBSchemaVersion initial = [FSTLevelDBMigrations schemaVersionForDB:_db];
|
|
XCTAssertEqual(0, initial, "No version should be equivalent to 0");
|
|
|
|
// Pick an arbitrary high migration number and migrate to it.
|
|
[FSTLevelDBMigrations runMigrationsOnDB:_db];
|
|
FSTLevelDBSchemaVersion actual = [FSTLevelDBMigrations schemaVersionForDB:_db];
|
|
XCTAssertGreaterThan(actual, 0, @"Expected to migrate to a schema version > 0");
|
|
}
|
|
|
|
- (void)testCountsQueries {
|
|
NSUInteger expected = 50;
|
|
FSTWriteGroup *group = [FSTWriteGroup groupWithAction:@"Setup"];
|
|
for (int i = 0; i < expected; i++) {
|
|
std::string key = [FSTLevelDBTargetKey keyWithTargetID:i];
|
|
[group setData:"dummy" forKey:key];
|
|
}
|
|
// Add a dummy entry after the targets to make sure the iteration is correctly bounded.
|
|
// Use a table that would sort logically right after that table 'target'.
|
|
std::string dummyKey;
|
|
// Magic number that indicates a table name follows. Needed to mimic the prefix to the target
|
|
// table.
|
|
OrderedCode::WriteSignedNumIncreasing(&dummyKey, 5);
|
|
OrderedCode::WriteString(&dummyKey, "targetA");
|
|
[group setData:"dummy" forKey:dummyKey];
|
|
|
|
Status status = [group writeToDB:_db];
|
|
XCTAssertTrue(status.ok(), @"Failed to write targets");
|
|
|
|
[FSTLevelDBMigrations runMigrationsOnDB:_db];
|
|
FSTPBTargetGlobal *metadata = [FSTLevelDBQueryCache readTargetMetadataFromDB:_db];
|
|
XCTAssertEqual(expected, metadata.targetCount, @"Failed to count all of the targets we added");
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|