mirror of https://github.com/facebook/rocksdb.git
143 lines
4.1 KiB
C++
143 lines
4.1 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/filename.h"
|
|
|
|
#include "db/dbformat.h"
|
|
#include "port/port.h"
|
|
#include "util/logging.h"
|
|
#include "util/testharness.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
class FileNameTest { };
|
|
|
|
TEST(FileNameTest, Parse) {
|
|
Slice db;
|
|
FileType type;
|
|
uint64_t number;
|
|
|
|
// Successful parses
|
|
static struct {
|
|
const char* fname;
|
|
uint64_t number;
|
|
FileType type;
|
|
} cases[] = {
|
|
{ "100.log", 100, kLogFile },
|
|
{ "0.log", 0, kLogFile },
|
|
{ "0.sst", 0, kTableFile },
|
|
{ "CURRENT", 0, kCurrentFile },
|
|
{ "LOCK", 0, kDBLockFile },
|
|
{ "MANIFEST-2", 2, kDescriptorFile },
|
|
{ "MANIFEST-7", 7, kDescriptorFile },
|
|
{ "METADB-2", 2, kMetaDatabase },
|
|
{ "METADB-7", 7, kMetaDatabase },
|
|
{ "LOG", 0, kInfoLogFile },
|
|
{ "LOG.old", 0, kInfoLogFile },
|
|
{ "18446744073709551615.log", 18446744073709551615ull, kLogFile },
|
|
};
|
|
for (unsigned int i = 0; i < sizeof(cases) / sizeof(cases[0]); i++) {
|
|
std::string f = cases[i].fname;
|
|
ASSERT_TRUE(ParseFileName(f, &number, &type)) << f;
|
|
ASSERT_EQ(cases[i].type, type) << f;
|
|
ASSERT_EQ(cases[i].number, number) << f;
|
|
}
|
|
|
|
// Errors
|
|
static const char* errors[] = {
|
|
"",
|
|
"foo",
|
|
"foo-dx-100.log",
|
|
".log",
|
|
"",
|
|
"manifest",
|
|
"CURREN",
|
|
"CURRENTX",
|
|
"MANIFES",
|
|
"MANIFEST",
|
|
"MANIFEST-",
|
|
"XMANIFEST-3",
|
|
"MANIFEST-3x",
|
|
"META",
|
|
"METADB",
|
|
"METADB-",
|
|
"XMETADB-3",
|
|
"METADB-3x",
|
|
"LOC",
|
|
"LOCKx",
|
|
"LO",
|
|
"LOGx",
|
|
"18446744073709551616.log",
|
|
"184467440737095516150.log",
|
|
"100",
|
|
"100.",
|
|
"100.lop"
|
|
};
|
|
for (unsigned int i = 0; i < sizeof(errors) / sizeof(errors[0]); i++) {
|
|
std::string f = errors[i];
|
|
ASSERT_TRUE(!ParseFileName(f, &number, &type)) << f;
|
|
};
|
|
}
|
|
|
|
TEST(FileNameTest, Construction) {
|
|
uint64_t number;
|
|
FileType type;
|
|
std::string fname;
|
|
|
|
fname = CurrentFileName("foo");
|
|
ASSERT_EQ("foo/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(0U, number);
|
|
ASSERT_EQ(kCurrentFile, type);
|
|
|
|
fname = LockFileName("foo");
|
|
ASSERT_EQ("foo/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(0U, number);
|
|
ASSERT_EQ(kDBLockFile, type);
|
|
|
|
fname = LogFileName("foo", 192);
|
|
ASSERT_EQ("foo/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(192U, number);
|
|
ASSERT_EQ(kLogFile, type);
|
|
|
|
fname = TableFileName({"bar"}, 200, 0);
|
|
std::string fname1 = TableFileName({"foo", "bar"}, 200, 1);
|
|
ASSERT_EQ(fname, fname1);
|
|
ASSERT_EQ("bar/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(200U, number);
|
|
ASSERT_EQ(kTableFile, type);
|
|
|
|
fname = DescriptorFileName("bar", 100);
|
|
ASSERT_EQ("bar/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(100U, number);
|
|
ASSERT_EQ(kDescriptorFile, type);
|
|
|
|
fname = TempFileName("tmp", 999);
|
|
ASSERT_EQ("tmp/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(999U, number);
|
|
ASSERT_EQ(kTempFile, type);
|
|
|
|
fname = MetaDatabaseName("met", 100);
|
|
ASSERT_EQ("met/", std::string(fname.data(), 4));
|
|
ASSERT_TRUE(ParseFileName(fname.c_str() + 4, &number, &type));
|
|
ASSERT_EQ(100U, number);
|
|
ASSERT_EQ(kMetaDatabase, type);
|
|
}
|
|
|
|
} // namespace rocksdb
|
|
|
|
int main(int argc, char** argv) {
|
|
return rocksdb::test::RunAllTests();
|
|
}
|