mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 02:44:18 +00:00
36a5f8ed7f
- Replace raw slice comparison with a call to user comparator.
Added test for custom comparators.
- Fix end of namespace comments.
- Fixed bug in picking inputs for a level-0 compaction.
When finding overlapping files, the covered range may expand
as files are added to the input set. We now correctly expand
the range when this happens instead of continuing to use the
old range. For example, suppose L0 contains files with the
following ranges:
F1: a .. d
F2: c .. g
F3: f .. j
and the initial compaction target is F3. We used to search
for range f..j which yielded {F2,F3}. However we now expand
the range as soon as another file is added. In this case,
when F2 is added, we expand the range to c..j and restart the
search. That picks up file F1 as well.
This change fixes a bug related to deleted keys showing up
incorrectly after a compaction as described in Issue 44.
(Sync with upstream @25072954)
139 lines
4.6 KiB
C++
139 lines
4.6 KiB
C++
// 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.
|
|
|
|
#ifndef STORAGE_LEVELDB_UTIL_TESTHARNESS_H_
|
|
#define STORAGE_LEVELDB_UTIL_TESTHARNESS_H_
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sstream>
|
|
#include "leveldb/env.h"
|
|
#include "leveldb/slice.h"
|
|
#include "util/random.h"
|
|
|
|
namespace leveldb {
|
|
namespace test {
|
|
|
|
// Run some of the tests registered by the TEST() macro. If the
|
|
// environment variable "LEVELDB_TESTS" is not set, runs all tests.
|
|
// Otherwise, runs only the tests whose name contains the value of
|
|
// "LEVELDB_TESTS" as a substring. E.g., suppose the tests are:
|
|
// TEST(Foo, Hello) { ... }
|
|
// TEST(Foo, World) { ... }
|
|
// LEVELDB_TESTS=Hello will run the first test
|
|
// LEVELDB_TESTS=o will run both tests
|
|
// LEVELDB_TESTS=Junk will run no tests
|
|
//
|
|
// Returns 0 if all tests pass.
|
|
// Dies or returns a non-zero value if some test fails.
|
|
extern int RunAllTests();
|
|
|
|
// Return the directory to use for temporary storage.
|
|
extern std::string TmpDir();
|
|
|
|
// Return a randomization seed for this run. Typically returns the
|
|
// same number on repeated invocations of this binary, but automated
|
|
// runs may be able to vary the seed.
|
|
extern int RandomSeed();
|
|
|
|
// An instance of Tester is allocated to hold temporary state during
|
|
// the execution of an assertion.
|
|
class Tester {
|
|
private:
|
|
bool ok_;
|
|
const char* fname_;
|
|
int line_;
|
|
std::stringstream ss_;
|
|
|
|
public:
|
|
Tester(const char* f, int l)
|
|
: ok_(true), fname_(f), line_(l) {
|
|
}
|
|
|
|
~Tester() {
|
|
if (!ok_) {
|
|
fprintf(stderr, "%s:%d:%s\n", fname_, line_, ss_.str().c_str());
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
Tester& Is(bool b, const char* msg) {
|
|
if (!b) {
|
|
ss_ << " Assertion failure " << msg;
|
|
ok_ = false;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
Tester& IsOk(const Status& s) {
|
|
if (!s.ok()) {
|
|
ss_ << " " << s.ToString();
|
|
ok_ = false;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
#define BINARY_OP(name,op) \
|
|
template <class X, class Y> \
|
|
Tester& name(const X& x, const Y& y) { \
|
|
if (! (x op y)) { \
|
|
ss_ << " failed: " << x << (" " #op " ") << y; \
|
|
ok_ = false; \
|
|
} \
|
|
return *this; \
|
|
}
|
|
|
|
BINARY_OP(IsEq, ==)
|
|
BINARY_OP(IsNe, !=)
|
|
BINARY_OP(IsGe, >=)
|
|
BINARY_OP(IsGt, >)
|
|
BINARY_OP(IsLe, <=)
|
|
BINARY_OP(IsLt, <)
|
|
#undef BINARY_OP
|
|
|
|
// Attach the specified value to the error message if an error has occurred
|
|
template <class V>
|
|
Tester& operator<<(const V& value) {
|
|
if (!ok_) {
|
|
ss_ << " " << value;
|
|
}
|
|
return *this;
|
|
}
|
|
};
|
|
|
|
#define ASSERT_TRUE(c) ::leveldb::test::Tester(__FILE__, __LINE__).Is((c), #c)
|
|
#define ASSERT_OK(s) ::leveldb::test::Tester(__FILE__, __LINE__).IsOk((s))
|
|
#define ASSERT_EQ(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsEq((a),(b))
|
|
#define ASSERT_NE(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsNe((a),(b))
|
|
#define ASSERT_GE(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsGe((a),(b))
|
|
#define ASSERT_GT(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsGt((a),(b))
|
|
#define ASSERT_LE(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsLe((a),(b))
|
|
#define ASSERT_LT(a,b) ::leveldb::test::Tester(__FILE__, __LINE__).IsLt((a),(b))
|
|
|
|
#define TCONCAT(a,b) TCONCAT1(a,b)
|
|
#define TCONCAT1(a,b) a##b
|
|
|
|
#define TEST(base,name) \
|
|
class TCONCAT(_Test_,name) : public base { \
|
|
public: \
|
|
void _Run(); \
|
|
static void _RunIt() { \
|
|
TCONCAT(_Test_,name) t; \
|
|
t._Run(); \
|
|
} \
|
|
}; \
|
|
bool TCONCAT(_Test_ignored_,name) = \
|
|
::leveldb::test::RegisterTest(#base, #name, &TCONCAT(_Test_,name)::_RunIt); \
|
|
void TCONCAT(_Test_,name)::_Run()
|
|
|
|
// Register the specified test. Typically not used directly, but
|
|
// invoked via the macro expansion of TEST.
|
|
extern bool RegisterTest(const char* base, const char* name, void (*func)());
|
|
|
|
|
|
} // namespace test
|
|
} // namespace leveldb
|
|
|
|
#endif // STORAGE_LEVELDB_UTIL_TESTHARNESS_H_
|