Gregory Maxwell
12 years ago
6 changed files with 107 additions and 9 deletions
@ -0,0 +1,92 @@
@@ -0,0 +1,92 @@
|
||||
// Copyright (c) 2013 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.
|
||||
|
||||
// Test for issue 178: a manual compaction causes deleted data to reappear.
|
||||
#include <iostream> |
||||
#include <sstream> |
||||
#include <cstdlib> |
||||
|
||||
#include "leveldb/db.h" |
||||
#include "leveldb/write_batch.h" |
||||
#include "util/testharness.h" |
||||
|
||||
namespace { |
||||
|
||||
const int kNumKeys = 1100000; |
||||
|
||||
std::string Key1(int i) { |
||||
char buf[100]; |
||||
snprintf(buf, sizeof(buf), "my_key_%d", i); |
||||
return buf; |
||||
} |
||||
|
||||
std::string Key2(int i) { |
||||
return Key1(i) + "_xxx"; |
||||
} |
||||
|
||||
class Issue178 { }; |
||||
|
||||
TEST(Issue178, Test) { |
||||
// Get rid of any state from an old run.
|
||||
std::string dbpath = leveldb::test::TmpDir() + "/leveldb_cbug_test"; |
||||
DestroyDB(dbpath, leveldb::Options()); |
||||
|
||||
// Open database. Disable compression since it affects the creation
|
||||
// of layers and the code below is trying to test against a very
|
||||
// specific scenario.
|
||||
leveldb::DB* db; |
||||
leveldb::Options db_options; |
||||
db_options.create_if_missing = true; |
||||
db_options.compression = leveldb::kNoCompression; |
||||
ASSERT_OK(leveldb::DB::Open(db_options, dbpath, &db)); |
||||
|
||||
// create first key range
|
||||
leveldb::WriteBatch batch; |
||||
for (size_t i = 0; i < kNumKeys; i++) { |
||||
batch.Put(Key1(i), "value for range 1 key"); |
||||
} |
||||
ASSERT_OK(db->Write(leveldb::WriteOptions(), &batch)); |
||||
|
||||
// create second key range
|
||||
batch.Clear(); |
||||
for (size_t i = 0; i < kNumKeys; i++) { |
||||
batch.Put(Key2(i), "value for range 2 key"); |
||||
} |
||||
ASSERT_OK(db->Write(leveldb::WriteOptions(), &batch)); |
||||
|
||||
// delete second key range
|
||||
batch.Clear(); |
||||
for (size_t i = 0; i < kNumKeys; i++) { |
||||
batch.Delete(Key2(i)); |
||||
} |
||||
ASSERT_OK(db->Write(leveldb::WriteOptions(), &batch)); |
||||
|
||||
// compact database
|
||||
std::string start_key = Key1(0); |
||||
std::string end_key = Key1(kNumKeys - 1); |
||||
leveldb::Slice least(start_key.data(), start_key.size()); |
||||
leveldb::Slice greatest(end_key.data(), end_key.size()); |
||||
|
||||
// commenting out the line below causes the example to work correctly
|
||||
db->CompactRange(&least, &greatest); |
||||
|
||||
// count the keys
|
||||
leveldb::Iterator* iter = db->NewIterator(leveldb::ReadOptions()); |
||||
size_t num_keys = 0; |
||||
for (iter->SeekToFirst(); iter->Valid(); iter->Next()) { |
||||
num_keys++; |
||||
} |
||||
delete iter; |
||||
ASSERT_EQ(kNumKeys, num_keys) << "Bad number of keys"; |
||||
|
||||
// close database
|
||||
delete db; |
||||
DestroyDB(dbpath, leveldb::Options()); |
||||
} |
||||
|
||||
} // anonymous namespace
|
||||
|
||||
int main(int argc, char** argv) { |
||||
return leveldb::test::RunAllTests(); |
||||
} |
Loading…
Reference in new issue