Move global static functions in db_test_util to DBTestBase

Summary:
Move global static functions in db_test_util to DBTestBase.
This is to prevent unused function warning when decoupling
db_test.cc into multiple files.

Test Plan: db_test

Reviewers: igor, sdong, anthony, IslamAbdelRahman, kradhakrishnan

Reviewed By: kradhakrishnan

Subscribers: dhruba

Differential Revision: https://reviews.facebook.net/D42009
main
Yueh-Hsuan Chiang 10 years ago
parent 625467a08a
commit 49f42ad032
  1. 12
      db/db_test.cc
  2. 24
      util/db_test_util.h

@ -4383,8 +4383,8 @@ void MinLevelHelper(DBTest* self, Options& options) {
std::vector<std::string> values; std::vector<std::string> values;
// Write 120KB (12 values, each 10K) // Write 120KB (12 values, each 10K)
for (int i = 0; i < 12; i++) { for (int i = 0; i < 12; i++) {
values.push_back(RandomString(&rnd, 10000)); values.push_back(DBTestBase::RandomString(&rnd, 10000));
ASSERT_OK(self->Put(Key(i), values[i])); ASSERT_OK(self->Put(DBTestBase::Key(i), values[i]));
} }
self->dbfull()->TEST_WaitForFlushMemTable(); self->dbfull()->TEST_WaitForFlushMemTable();
ASSERT_EQ(self->NumTableFilesAtLevel(0), num + 1); ASSERT_EQ(self->NumTableFilesAtLevel(0), num + 1);
@ -4393,8 +4393,8 @@ void MinLevelHelper(DBTest* self, Options& options) {
//generate one more file in level-0, and should trigger level-0 compaction //generate one more file in level-0, and should trigger level-0 compaction
std::vector<std::string> values; std::vector<std::string> values;
for (int i = 0; i < 12; i++) { for (int i = 0; i < 12; i++) {
values.push_back(RandomString(&rnd, 10000)); values.push_back(DBTestBase::RandomString(&rnd, 10000));
ASSERT_OK(self->Put(Key(i), values[i])); ASSERT_OK(self->Put(DBTestBase::Key(i), values[i]));
} }
self->dbfull()->TEST_WaitForCompact(); self->dbfull()->TEST_WaitForCompact();
@ -9211,7 +9211,7 @@ static void RandomTimeoutWriter(void* arg) {
for (int k = 0; k < num_keys; ++k) { for (int k = 0; k < num_keys; ++k) {
int key = k + thread_id * num_keys; int key = k + thread_id * num_keys;
std::string value = RandomString(&rnd, kValueSize); std::string value = DBTestBase::RandomString(&rnd, kValueSize);
// only the second-half is randomized // only the second-half is randomized
if (k > num_keys / 2) { if (k > num_keys / 2) {
switch (rnd.Next() % 5) { switch (rnd.Next() % 5) {
@ -9231,7 +9231,7 @@ static void RandomTimeoutWriter(void* arg) {
} }
uint64_t time_before_put = db->GetEnv()->NowMicros(); uint64_t time_before_put = db->GetEnv()->NowMicros();
Status s = db->Put(write_opt, Key(key), value); Status s = db->Put(write_opt, DBTestBase::Key(key), value);
uint64_t put_duration = db->GetEnv()->NowMicros() - time_before_put; uint64_t put_duration = db->GetEnv()->NowMicros() - time_before_put;
if (write_opt.timeout_hint_us == 0 || if (write_opt.timeout_hint_us == 0 ||
put_duration + kTimerBias < write_opt.timeout_hint_us) { put_duration + kTimerBias < write_opt.timeout_hint_us) {

@ -55,12 +55,6 @@
namespace rocksdb { namespace rocksdb {
static std::string RandomString(Random* rnd, int len) {
std::string r;
test::RandomString(rnd, len, &r);
return r;
}
namespace anon { namespace anon {
class AtomicCounter { class AtomicCounter {
public: public:
@ -118,12 +112,6 @@ struct OptionsOverride {
} // namespace anon } // namespace anon
static std::string Key(int i) {
char buf[100];
snprintf(buf, sizeof(buf), "key%06d", i);
return std::string(buf);
}
// Special Env used to delay background operations // Special Env used to delay background operations
class SpecialEnv : public EnvWrapper { class SpecialEnv : public EnvWrapper {
public: public:
@ -456,6 +444,18 @@ class DBTestBase : public testing::Test {
~DBTestBase(); ~DBTestBase();
static std::string RandomString(Random* rnd, int len) {
std::string r;
test::RandomString(rnd, len, &r);
return r;
}
static std::string Key(int i) {
char buf[100];
snprintf(buf, sizeof(buf), "key%06d", i);
return std::string(buf);
}
// Switch to a fresh database with the next option configuration to // Switch to a fresh database with the next option configuration to
// test. Return false if there are no more configurations to test. // test. Return false if there are no more configurations to test.
bool ChangeOptions(int skip_mask = kNoSkip); bool ChangeOptions(int skip_mask = kNoSkip);

Loading…
Cancel
Save