68 lines
2.1 KiB
C++
68 lines
2.1 KiB
C++
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
#include "blockstore/implementations/inmemory/InMemoryBlockStore.h"
|
||
|
|
#include "blockstore/implementations/inmemory/InMemoryBlock.h"
|
||
|
|
#include "blobstore/implementations/onblocks/BlobStoreOnBlocks.h"
|
||
|
|
#include "blobstore/implementations/onblocks/impl/DataLeafNode.h"
|
||
|
|
#include "test/testutils/DataBlockFixture.h"
|
||
|
|
|
||
|
|
using ::testing::Test;
|
||
|
|
using ::testing::WithParamInterface;
|
||
|
|
using ::testing::Values;
|
||
|
|
using std::unique_ptr;
|
||
|
|
using std::make_unique;
|
||
|
|
using std::string;
|
||
|
|
|
||
|
|
using blockstore::BlockStore;
|
||
|
|
using blockstore::Data;
|
||
|
|
using blockstore::inmemory::InMemoryBlockStore;
|
||
|
|
using namespace blobstore;
|
||
|
|
using namespace blobstore::onblocks;
|
||
|
|
|
||
|
|
class DataLeafNodeTest: public Test {
|
||
|
|
public:
|
||
|
|
unique_ptr<BlockStore> blockStore = make_unique<InMemoryBlockStore>();
|
||
|
|
|
||
|
|
DataLeafNodeTest(): randomData(DataNodeView::DATASIZE_BYTES) {
|
||
|
|
DataBlockFixture dataFixture(DataNodeView::DATASIZE_BYTES);
|
||
|
|
|
||
|
|
std::memcpy(randomData.data(), dataFixture.data(), randomData.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
Key WriteDataToNewLeafBlockAndReturnKey() {
|
||
|
|
auto block = blockStore->create(BlobStoreOnBlocks::BLOCKSIZE);
|
||
|
|
DataLeafNode leaf(DataNodeView(std::move(block.block)));
|
||
|
|
leaf.write(0, randomData.size(), randomData);
|
||
|
|
return block.key;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ReadDataFromLeafBlock(Key key, Data *data) {
|
||
|
|
DataLeafNode leaf(DataNodeView(blockStore->load(key)));
|
||
|
|
leaf.read(0, data->size(), data);
|
||
|
|
}
|
||
|
|
|
||
|
|
Data randomData;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(DataLeafNodeTest, ReadWrittenDataImmediately) {
|
||
|
|
auto block = blockStore->create(BlobStoreOnBlocks::BLOCKSIZE);
|
||
|
|
DataLeafNode leaf(DataNodeView(std::move(block.block)));
|
||
|
|
leaf.write(0, randomData.size(), randomData);
|
||
|
|
|
||
|
|
Data read(DataNodeView::DATASIZE_BYTES);
|
||
|
|
leaf.read(0, read.size(), &read);
|
||
|
|
EXPECT_EQ(0, std::memcmp(randomData.data(), read.data(), randomData.size()));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(DataLeafNodeTest, ReadWrittenDataAfterReloadingBLock) {
|
||
|
|
Key key = WriteDataToNewLeafBlockAndReturnKey();
|
||
|
|
|
||
|
|
Data data(DataNodeView::DATASIZE_BYTES);
|
||
|
|
ReadDataFromLeafBlock(key, &data);
|
||
|
|
|
||
|
|
EXPECT_EQ(0, std::memcmp(randomData.data(), data.data(), randomData.size()));
|
||
|
|
}
|
||
|
|
|
||
|
|
//TODO Write tests that only read part of the data
|
||
|
|
//TODO Test numBytesInThisNode()
|