mirror of https://github.com/facebook/rocksdb.git
82 lines
2.4 KiB
C++
82 lines
2.4 KiB
C++
// Copyright (c) 2013, 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.
|
|
//
|
|
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file. See the AUTHORS file for names of contributors.
|
|
|
|
#include "db/db_impl.h"
|
|
#include "rocksdb/db.h"
|
|
#include "util/testharness.h"
|
|
|
|
#include <algorithm>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
namespace rocksdb {
|
|
|
|
using namespace std;
|
|
|
|
class ColumnFamilyTest {
|
|
public:
|
|
ColumnFamilyTest() {
|
|
dbname_ = test::TmpDir() + "/column_family_test";
|
|
db_options_.create_if_missing = true;
|
|
options_.create_if_missing = true;
|
|
DestroyDB(dbname_, options_);
|
|
}
|
|
|
|
void Close() {
|
|
delete db_;
|
|
db_ = nullptr;
|
|
}
|
|
|
|
Status Open(vector<string> cf) {
|
|
vector<ColumnFamilyDescriptor> column_families;
|
|
for (auto x : cf) {
|
|
column_families.push_back(
|
|
ColumnFamilyDescriptor(x, ColumnFamilyOptions()));
|
|
}
|
|
vector <ColumnFamilyHandle> handles;
|
|
return DB::OpenWithColumnFamilies(db_options_, dbname_, column_families,
|
|
&handles, &db_);
|
|
}
|
|
|
|
Options options_;
|
|
ColumnFamilyOptions column_family_options_;
|
|
DBOptions db_options_;
|
|
string dbname_;
|
|
DB* db_;
|
|
};
|
|
|
|
TEST(ColumnFamilyTest, AddDrop) {
|
|
ASSERT_OK(Open({"default"}));
|
|
ColumnFamilyHandle handles[4];
|
|
ASSERT_OK(
|
|
db_->CreateColumnFamily(column_family_options_, "one", &handles[0]));
|
|
ASSERT_OK(
|
|
db_->CreateColumnFamily(column_family_options_, "two", &handles[1]));
|
|
ASSERT_OK(
|
|
db_->CreateColumnFamily(column_family_options_, "three", &handles[2]));
|
|
ASSERT_OK(db_->DropColumnFamily(handles[1]));
|
|
ASSERT_OK(
|
|
db_->CreateColumnFamily(column_family_options_, "four", &handles[3]));
|
|
Close();
|
|
ASSERT_TRUE(Open({"default"}).IsInvalidArgument());
|
|
ASSERT_OK(Open({"default", "one", "three", "four"}));
|
|
Close();
|
|
|
|
vector<string> families;
|
|
ASSERT_OK(DB::ListColumnFamilies(db_options_, dbname_, &families));
|
|
sort(families.begin(), families.end());
|
|
ASSERT_TRUE(families == vector<string>({"default", "four", "one", "three"}));
|
|
}
|
|
|
|
} // namespace rocksdb
|
|
|
|
int main(int argc, char** argv) {
|
|
return rocksdb::test::RunAllTests();
|
|
}
|