82 lines
2.2 KiB
C++
82 lines
2.2 KiB
C++
#include "../../../../implementations/ondisk/OnDiskBlock.h"
|
|
#include <messmer/cpp-utils/data/DataFixture.h>
|
|
#include "../../../../utils/FileDoesntExistException.h"
|
|
#include "google/gtest/gtest.h"
|
|
|
|
#include <messmer/cpp-utils/data/Data.h>
|
|
#include <messmer/cpp-utils/tempfile/TempFile.h>
|
|
#include <messmer/cpp-utils/tempfile/TempDir.h>
|
|
#include <messmer/cpp-utils/pointer/unique_ref_boost_optional_gtest_workaround.h>
|
|
#include <fstream>
|
|
|
|
using ::testing::Test;
|
|
using ::testing::WithParamInterface;
|
|
using ::testing::Values;
|
|
|
|
using std::ofstream;
|
|
using std::unique_ptr;
|
|
using std::ios;
|
|
using cpputils::Data;
|
|
using cpputils::DataFixture;
|
|
using cpputils::TempFile;
|
|
using cpputils::TempDir;
|
|
using cpputils::unique_ref;
|
|
|
|
using namespace blockstore;
|
|
using namespace blockstore::ondisk;
|
|
|
|
namespace bf = boost::filesystem;
|
|
|
|
class OnDiskBlockLoadTest: public Test, public WithParamInterface<size_t> {
|
|
public:
|
|
OnDiskBlockLoadTest():
|
|
dir(),
|
|
key(Key::FromString("1491BB4932A389EE14BC7090AC772972")),
|
|
file(dir.path() / key.ToString()) {
|
|
}
|
|
TempDir dir;
|
|
Key key;
|
|
TempFile file;
|
|
|
|
void SetFileSize(size_t size) {
|
|
Data data(size);
|
|
data.StoreToFile(file.path());
|
|
}
|
|
|
|
void StoreData(const Data &data) {
|
|
data.StoreToFile(file.path());
|
|
}
|
|
|
|
unique_ref<OnDiskBlock> LoadBlock() {
|
|
return OnDiskBlock::LoadFromDisk(dir.path(), key).value();
|
|
}
|
|
|
|
void EXPECT_BLOCK_DATA_EQ(const Data &expected, const OnDiskBlock &actual) {
|
|
EXPECT_EQ(expected.size(), actual.size());
|
|
EXPECT_EQ(0, std::memcmp(expected.data(), actual.data(), expected.size()));
|
|
}
|
|
};
|
|
INSTANTIATE_TEST_CASE_P(OnDiskBlockLoadTest, OnDiskBlockLoadTest, Values(0, 1, 5, 1024, 10*1024*1024));
|
|
|
|
TEST_P(OnDiskBlockLoadTest, FileSizeIsCorrect) {
|
|
SetFileSize(GetParam());
|
|
|
|
auto block = LoadBlock();
|
|
|
|
EXPECT_EQ(GetParam(), block->size());
|
|
}
|
|
|
|
TEST_P(OnDiskBlockLoadTest, LoadedDataIsCorrect) {
|
|
Data randomData = DataFixture::generate(GetParam());
|
|
StoreData(randomData);
|
|
|
|
auto block = LoadBlock();
|
|
|
|
EXPECT_BLOCK_DATA_EQ(randomData, *block);
|
|
}
|
|
|
|
TEST_F(OnDiskBlockLoadTest, LoadNotExistingBlock) {
|
|
Key key2 = Key::FromString("272EE5517627CFA147A971A8E6E747E0");
|
|
EXPECT_EQ(boost::none, OnDiskBlock::LoadFromDisk(dir.path(), key2));
|
|
}
|