116 lines
3.6 KiB
C++
116 lines
3.6 KiB
C++
#include "../../../../implementations/ondisk/OnDiskBlock.h"
|
|
#include <messmer/cpp-utils/data/DataFixture.h>
|
|
#include "google/gtest/gtest.h"
|
|
|
|
#include <messmer/cpp-utils/tempfile/TempFile.h>
|
|
#include <messmer/cpp-utils/tempfile/TempDir.h>
|
|
|
|
using ::testing::Test;
|
|
using ::testing::WithParamInterface;
|
|
using ::testing::Values;
|
|
|
|
using std::unique_ptr;
|
|
using cpputils::Data;
|
|
using cpputils::DataFixture;
|
|
using cpputils::TempFile;
|
|
using cpputils::TempDir;
|
|
|
|
using namespace blockstore;
|
|
using namespace blockstore::ondisk;
|
|
|
|
namespace bf = boost::filesystem;
|
|
|
|
class OnDiskBlockFlushTest: public Test, public WithParamInterface<size_t> {
|
|
public:
|
|
OnDiskBlockFlushTest()
|
|
// Don't create the temp file yet (therefore pass false to the TempFile constructor)
|
|
: dir(),
|
|
key(Key::FromString("1491BB4932A389EE14BC7090AC772972")),
|
|
file(dir.path() / key.ToString(), false),
|
|
randomData(DataFixture::generate(GetParam())) {
|
|
}
|
|
TempDir dir;
|
|
Key key;
|
|
TempFile file;
|
|
|
|
Data randomData;
|
|
|
|
unique_ptr<OnDiskBlock> CreateBlockAndLoadItFromDisk() {
|
|
{
|
|
OnDiskBlock::CreateOnDisk(dir.path(), key, randomData.copy());
|
|
}
|
|
return OnDiskBlock::LoadFromDisk(dir.path(), key);
|
|
}
|
|
|
|
unique_ptr<OnDiskBlock> CreateBlock() {
|
|
return OnDiskBlock::CreateOnDisk(dir.path(), key, randomData.copy());
|
|
}
|
|
|
|
void WriteDataToBlock(const unique_ptr<OnDiskBlock> &block) {
|
|
block->write(randomData.data(), 0, randomData.size());
|
|
}
|
|
|
|
void EXPECT_BLOCK_DATA_CORRECT(const unique_ptr<OnDiskBlock> &block) {
|
|
EXPECT_EQ(randomData.size(), block->size());
|
|
EXPECT_EQ(0, std::memcmp(randomData.data(), block->data(), randomData.size()));
|
|
}
|
|
|
|
void EXPECT_STORED_FILE_DATA_CORRECT() {
|
|
Data actual = Data::LoadFromFile(file.path()).value();
|
|
EXPECT_EQ(randomData, actual);
|
|
}
|
|
};
|
|
INSTANTIATE_TEST_CASE_P(OnDiskBlockFlushTest, OnDiskBlockFlushTest, Values((size_t)0, (size_t)1, (size_t)1024, (size_t)4096, (size_t)10*1024*1024));
|
|
|
|
// This test is also tested by OnDiskBlockStoreTest, but there the block is created using the BlockStore interface.
|
|
// Here, we create it using OnDiskBlock::CreateOnDisk()
|
|
TEST_P(OnDiskBlockFlushTest, AfterCreate_FlushingDoesntChangeBlock) {
|
|
auto block = CreateBlock();
|
|
WriteDataToBlock(block);
|
|
|
|
EXPECT_BLOCK_DATA_CORRECT(block);
|
|
}
|
|
|
|
// This test is also tested by OnDiskBlockStoreTest, but there the block is created using the BlockStore interface.
|
|
// Here, we create it using OnDiskBlock::CreateOnDisk() / OnDiskBlock::LoadFromDisk()
|
|
TEST_P(OnDiskBlockFlushTest, AfterLoad_FlushingDoesntChangeBlock) {
|
|
auto block = CreateBlockAndLoadItFromDisk();
|
|
WriteDataToBlock(block);
|
|
|
|
EXPECT_BLOCK_DATA_CORRECT(block);
|
|
}
|
|
|
|
TEST_P(OnDiskBlockFlushTest, AfterCreate_FlushingWritesCorrectData) {
|
|
auto block = CreateBlock();
|
|
WriteDataToBlock(block);
|
|
|
|
EXPECT_STORED_FILE_DATA_CORRECT();
|
|
}
|
|
|
|
TEST_P(OnDiskBlockFlushTest, AfterLoad_FlushingWritesCorrectData) {
|
|
auto block = CreateBlockAndLoadItFromDisk();
|
|
WriteDataToBlock(block);
|
|
|
|
EXPECT_STORED_FILE_DATA_CORRECT();
|
|
}
|
|
|
|
// This test is also tested by OnDiskBlockStoreTest, but there it can only checks block content by loading it again.
|
|
// Here, we check the content on disk.
|
|
TEST_P(OnDiskBlockFlushTest, AfterCreate_FlushesWhenDestructed) {
|
|
{
|
|
auto block = CreateBlock();
|
|
WriteDataToBlock(block);
|
|
}
|
|
EXPECT_STORED_FILE_DATA_CORRECT();
|
|
}
|
|
|
|
// This test is also tested by OnDiskBlockStoreTest, but there it can only checks block content by loading it again.
|
|
// Here, we check the content on disk.
|
|
TEST_P(OnDiskBlockFlushTest, AfterLoad_FlushesWhenDestructed) {
|
|
{
|
|
auto block = CreateBlockAndLoadItFromDisk();
|
|
WriteDataToBlock(block);
|
|
}
|
|
EXPECT_STORED_FILE_DATA_CORRECT();
|
|
}
|