You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 rivejä
1.6 KiB
75 rivejä
1.6 KiB
#include "gtest/gtest.h"
|
|
#include "leveldb/env.h"
|
|
#include "leveldb/db.h"
|
|
using namespace leveldb;
|
|
|
|
constexpr int value_size = 2048;
|
|
constexpr int data_size = 128 << 20;
|
|
|
|
std::map<std::string, std::string> data;
|
|
|
|
Status OpenDB(std::string dbName, DB **db) {
|
|
Options options;
|
|
options.create_if_missing = true;
|
|
return DB::Open(options, dbName, db);
|
|
}
|
|
|
|
void InsertData(DB *db) {
|
|
WriteOptions writeOptions;
|
|
int key_num = data_size / value_size;
|
|
srand(0);
|
|
|
|
for (int i = 0; i < key_num; i++) {
|
|
int key_ = rand() % key_num+1;
|
|
std::string key = std::to_string(key_);
|
|
std::string value(value_size, 'a');
|
|
db->Put(writeOptions, key, value);
|
|
data[key] = value;
|
|
}
|
|
}
|
|
|
|
TEST(TestBasicIO, PointGet) {
|
|
DB *db;
|
|
if(OpenDB("testdb", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
|
|
data.clear();
|
|
InsertData(db);
|
|
|
|
ReadOptions readOptions;
|
|
|
|
for (auto iter = data.begin(); iter != data.end(); ++iter) {
|
|
std::string value;
|
|
db->Get(readOptions, iter->first, &value);
|
|
ASSERT_TRUE(value == iter->second);
|
|
}
|
|
|
|
delete db;
|
|
}
|
|
|
|
TEST(TestBasicIO, RangeQuery) {
|
|
DB *db;
|
|
if(OpenDB("testdb", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
|
|
data.clear();
|
|
InsertData(db);
|
|
|
|
ReadOptions readOptions;
|
|
|
|
auto iter = db->NewIterator(readOptions);
|
|
iter->SeekToFirst();
|
|
while (iter->Valid()) {
|
|
ASSERT_TRUE(data[iter->key().ToString()] == iter->fields()["1"]);
|
|
iter->Next();
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|