af0c9ac01d
Summary: This enables configurable Envs without recompiling. For example, my next diff will make env_test test an Env created by NewEnvFromUri(). Then, users can determine which Env is tested simply by providing the URI for NewEnvFromUri() (e.g., through a CLI argument or environment variable). The registration process allows us to register any Env that is linked with the RocksDB library, so we can register our internal Envs as well. The registration code is inspired by our internal InitRegistry. Test Plan: new unit test Reviewers: IslamAbdelRahman, lightmark, ldemailly, sdong Reviewed By: sdong Subscribers: leveldb, dhruba, andrewkr Differential Revision: https://reviews.facebook.net/D58449
73 lines
2.0 KiB
C++
73 lines
2.0 KiB
C++
// Copyright (c) 2016-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
#include "rocksdb/utilities/env_registry.h"
|
|
#include "util/testharness.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
class EnvRegistryTest : public testing::Test {
|
|
public:
|
|
static int num_a, num_b;
|
|
};
|
|
|
|
int EnvRegistryTest::num_a = 0;
|
|
int EnvRegistryTest::num_b = 0;
|
|
|
|
static EnvRegistrar test_reg_a("a://", [](const std::string& uri,
|
|
std::unique_ptr<Env>* env_guard) {
|
|
++EnvRegistryTest::num_a;
|
|
return Env::Default();
|
|
});
|
|
|
|
static EnvRegistrar test_reg_b("b://", [](const std::string& uri,
|
|
std::unique_ptr<Env>* env_guard) {
|
|
++EnvRegistryTest::num_b;
|
|
// Env::Default() is a singleton so we can't grant ownership directly to the
|
|
// caller - we must wrap it first.
|
|
env_guard->reset(new EnvWrapper(Env::Default()));
|
|
return env_guard->get();
|
|
});
|
|
|
|
TEST_F(EnvRegistryTest, Basics) {
|
|
std::unique_ptr<Env> env_guard;
|
|
auto res = NewEnvFromUri("a://test", &env_guard);
|
|
ASSERT_NE(res, nullptr);
|
|
ASSERT_EQ(env_guard, nullptr);
|
|
ASSERT_EQ(1, num_a);
|
|
ASSERT_EQ(0, num_b);
|
|
|
|
res = NewEnvFromUri("b://test", &env_guard);
|
|
ASSERT_NE(res, nullptr);
|
|
ASSERT_NE(env_guard, nullptr);
|
|
ASSERT_EQ(1, num_a);
|
|
ASSERT_EQ(1, num_b);
|
|
|
|
res = NewEnvFromUri("c://test", &env_guard);
|
|
ASSERT_EQ(res, nullptr);
|
|
ASSERT_EQ(env_guard, nullptr);
|
|
ASSERT_EQ(1, num_a);
|
|
ASSERT_EQ(1, num_b);
|
|
}
|
|
|
|
} // namespace rocksdb
|
|
|
|
int main(int argc, char** argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|
|
|
|
#else // ROCKSDB_LITE
|
|
#include <stdio.h>
|
|
|
|
int main(int argc, char** argv) {
|
|
fprintf(stderr, "SKIPPED as EnvRegistry is not supported in ROCKSDB_LITE\n");
|
|
return 0;
|
|
}
|
|
|
|
#endif // ROCKSDB_LITE
|