Files
firebase-ios-sdk/Firestore/Example/Tests/Local/FSTPersistenceTestHelpers.mm
T
Gil 6ad473ae04 Merge persistence-master to master (#3829)
* Port FSTMemoryPersistence to C++ (#3754)

* Add local::Persistence

* Add MemoryPersistence

* Add MemoryEagerReferenceDelegate

* Make LruDelegate extend ReferenceDelegate

This enables downcasting to LruDelegate in tests that need to do so.

* Add MemoryLruReferenceDelegate

* Don't double-count document key size.

* Port FSTLevelDB to C++ (#3755)

* Add LevelDbPersistence

Note that Apple-specific initialization has been partitioned off in
anticipation of being able to run this code on non-Apple platforms.

* Add LevelDbLruReferenceDelegate

* Migrate to C++ Persistence (1/2) (#3756)

* Use C++ Persistence classes in FSTLocalStore

* Remove ReferenceDelegateBridge

* Cut src over to C++ persistence types

* Remove Objective-C requirement from some classes

... that no longer require it.

* Migrate to C++ Persistence (2/2) (#3757)

* Remove LruDelegateBridge

* Update test initialization of persistence components

In particular, ensure that there's always an owner for the Persistence
instance, which wasn't always handled explicitly in the Objective-C
version of the tests.

* Largely mechanical changes to tests

* Remove Objective-C persistence classes

* Forward declare util::Status when possible (#3761)

* Add a header dedicated to forward declarations for util::Status

* Use status_fwd.h wherever possible
2019-09-12 14:28:10 -07:00

107 lines
4.0 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 "Firestore/Example/Tests/Local/FSTPersistenceTestHelpers.h"
#include <utility>
#import "Firestore/Source/Local/FSTLocalSerializer.h"
#import "Firestore/Source/Remote/FSTSerializerBeta.h"
#include "Firestore/core/src/firebase/firestore/local/leveldb_persistence.h"
#include "Firestore/core/src/firebase/firestore/local/lru_garbage_collector.h"
#include "Firestore/core/src/firebase/firestore/local/memory_persistence.h"
#include "Firestore/core/src/firebase/firestore/local/proto_sizer.h"
#include "Firestore/core/src/firebase/firestore/model/database_id.h"
#include "Firestore/core/src/firebase/firestore/util/filesystem.h"
#include "Firestore/core/src/firebase/firestore/util/path.h"
#include "Firestore/core/src/firebase/firestore/util/status.h"
#include "Firestore/core/src/firebase/firestore/util/string_apple.h"
namespace util = firebase::firestore::util;
using firebase::firestore::local::LevelDbPersistence;
using firebase::firestore::local::LruParams;
using firebase::firestore::local::MemoryPersistence;
using firebase::firestore::local::ProtoSizer;
using firebase::firestore::model::DatabaseId;
using firebase::firestore::util::Path;
using firebase::firestore::util::Status;
NS_ASSUME_NONNULL_BEGIN
@implementation FSTPersistenceTestHelpers
+ (FSTLocalSerializer *)localSerializer {
auto remoteSerializer = [[FSTSerializerBeta alloc] initWithDatabaseID:DatabaseId("p", "d")];
return [[FSTLocalSerializer alloc] initWithRemoteSerializer:remoteSerializer];
}
+ (Path)levelDBDir {
Path dir = util::TempDir().AppendUtf8("FSTPersistenceTestHelpers");
// Delete the directory first to ensure isolation between runs.
util::Status status = util::RecursivelyDelete(dir);
if (!status.ok()) {
[NSException
raise:NSInternalInconsistencyException
format:@"Failed to clean up leveldb path %s: %s", dir.c_str(), status.ToString().c_str()];
}
return dir;
}
+ (std::unique_ptr<LevelDbPersistence>)levelDBPersistenceWithDir:(Path)dir {
return [self levelDBPersistenceWithDir:dir lruParams:LruParams::Default()];
}
+ (std::unique_ptr<LevelDbPersistence>)levelDBPersistenceWithDir:(Path)dir
lruParams:(LruParams)params {
FSTLocalSerializer *serializer = [self localSerializer];
auto created = LevelDbPersistence::Create(std::move(dir), serializer, params);
if (!created.ok()) {
[NSException raise:NSInternalInconsistencyException
format:@"Failed to open DB: %s", created.status().ToString().c_str()];
}
return std::move(created).ValueOrDie();
}
+ (std::unique_ptr<local::LevelDbPersistence>)levelDBPersistenceWithLruParams:(LruParams)lruParams {
return [self levelDBPersistenceWithDir:[self levelDBDir] lruParams:lruParams];
}
+ (std::unique_ptr<local::LevelDbPersistence>)levelDBPersistence {
return [self levelDBPersistenceWithDir:[self levelDBDir]];
}
+ (std::unique_ptr<local::MemoryPersistence>)eagerGCMemoryPersistence {
return MemoryPersistence::WithEagerGarbageCollector();
}
+ (std::unique_ptr<local::MemoryPersistence>)lruMemoryPersistence {
return [self lruMemoryPersistenceWithLruParams:LruParams::Default()];
}
+ (std::unique_ptr<local::MemoryPersistence>)lruMemoryPersistenceWithLruParams:
(LruParams)lruParams {
FSTLocalSerializer *serializer = [self localSerializer];
auto sizer = absl::make_unique<ProtoSizer>(serializer);
return MemoryPersistence::WithLruGarbageCollector(lruParams, std::move(sizer));
}
@end
NS_ASSUME_NONNULL_END