您最多选择25个主题
主题必须以字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
82 行
2.2 KiB
82 行
2.2 KiB
#include <chrono>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "leveldb/env.h"
|
|
#include "leveldb/db.h"
|
|
|
|
using namespace leveldb;
|
|
|
|
constexpr int short_value_size = 4;
|
|
constexpr int long_value_size = 64;
|
|
constexpr int data_size = 512;
|
|
|
|
Status OpenDB(std::string dbName, DB **db) {
|
|
Options options;
|
|
options.create_if_missing = true;
|
|
return DB::Open(options, dbName, db);
|
|
}
|
|
|
|
void InsertData(DB *db, int value_size) {
|
|
WriteOptions writeOptions;
|
|
int key_num = data_size / value_size;
|
|
srand(42);
|
|
|
|
for (int i = 0; i < key_num; i++) {
|
|
int key_ = i;
|
|
std::string key = std::to_string(key_);
|
|
std::string value(value_size, 'a');
|
|
db->Put(writeOptions, key, value);
|
|
}
|
|
|
|
}
|
|
|
|
TEST(TestKV, GetValue) {
|
|
DB *db;
|
|
if(OpenDB("testdb_TestKV_short_value", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
InsertData(db, short_value_size);
|
|
|
|
ReadOptions readOptions;
|
|
Status status;
|
|
int key_num = data_size / short_value_size;
|
|
srand(42);
|
|
for (int i = 0; i < key_num; i++) {
|
|
// int key_ = rand() % key_num+1;
|
|
std::string key = std::to_string(i);
|
|
std::string value;
|
|
std::string expected_value(short_value_size, 'a');
|
|
status = db->Get(readOptions, key, &value);
|
|
// std::cout << key << std::endl;
|
|
ASSERT_TRUE(status.ok());
|
|
EXPECT_EQ(expected_value, value);
|
|
}
|
|
}
|
|
|
|
TEST(TestKV, GetLongValue) {
|
|
DB *db;
|
|
if(OpenDB("testdb_TestKV_long_value", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
InsertData(db, long_value_size);
|
|
|
|
ReadOptions readOptions;
|
|
Status status;
|
|
int key_num = data_size / long_value_size;
|
|
for (int i = 0; i < key_num; i++) {
|
|
std::string key = std::to_string(i);
|
|
std::string value;
|
|
std::string expected_value(long_value_size, 'a');
|
|
status = db->Get(readOptions, key, &value);
|
|
ASSERT_TRUE(status.ok());
|
|
EXPECT_EQ(expected_value, value);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
// All tests currently run with the same read-only file limits.
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|