#include "../../../../implementations/onblocks/datanodestore/DataNodeView.h"
#include <google/gtest/gtest.h>

#include "messmer/blockstore/implementations/testfake/FakeBlockStore.h"
#include "messmer/blockstore/implementations/testfake/FakeBlock.h"
#include "../../../../implementations/onblocks/BlobStoreOnBlocks.h"
#include <messmer/cpp-utils/data/DataFixture.h>

using ::testing::Test;
using ::testing::WithParamInterface;
using ::testing::Values;
using std::string;

using blockstore::BlockStore;
using blockstore::testfake::FakeBlockStore;
using cpputils::Data;
using cpputils::DataFixture;
using cpputils::unique_ref;
using cpputils::make_unique_ref;
using namespace blobstore;
using namespace blobstore::onblocks;
using namespace blobstore::onblocks::datanodestore;

class DataNodeViewTest: public Test {
public:
  static constexpr uint32_t BLOCKSIZE_BYTES = 1024;
  static constexpr uint32_t DATASIZE_BYTES = DataNodeLayout(DataNodeViewTest::BLOCKSIZE_BYTES).datasizeBytes();

  unique_ref<BlockStore> blockStore = make_unique_ref<FakeBlockStore>();
};

class DataNodeViewDepthTest: public DataNodeViewTest, public WithParamInterface<uint8_t> {
};
INSTANTIATE_TEST_CASE_P(DataNodeViewDepthTest, DataNodeViewDepthTest, Values(0, 1, 3, 10, 100));

TEST_P(DataNodeViewDepthTest, DepthIsStored) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  auto key = block->key();
  {
    DataNodeView view(std::move(block));
    view.setDepth(GetParam());
  }
  DataNodeView view(blockStore->load(key));
  EXPECT_EQ(GetParam(), view.Depth());
}

class DataNodeViewSizeTest: public DataNodeViewTest, public WithParamInterface<uint32_t> {
};
INSTANTIATE_TEST_CASE_P(DataNodeViewSizeTest, DataNodeViewSizeTest, Values(0, 50, 64, 1024, 1024*1024*1024));

TEST_P(DataNodeViewSizeTest, SizeIsStored) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  auto key = block->key();
  {
    DataNodeView view(std::move(block));
    view.setSize(GetParam());
  }
  DataNodeView view(blockStore->load(key));
  EXPECT_EQ(GetParam(), view.Size());
}

TEST_F(DataNodeViewTest, DataIsStored) {
  Data randomData = DataFixture::generate(DATASIZE_BYTES);
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  auto key = block->key();
  {
    DataNodeView view(std::move(block));
    view.write(randomData.data(), 0, randomData.size());
  }
  DataNodeView view(blockStore->load(key));
  EXPECT_EQ(0, std::memcmp(view.data(), randomData.data(), randomData.size()));
}

TEST_F(DataNodeViewTest, HeaderAndBodyDontOverlap) {
  Data randomData = DataFixture::generate(DATASIZE_BYTES);
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  auto key = block->key();
  {
    DataNodeView view(std::move(block));
    view.setDepth(3);
    view.setSize(1000000000u);
    view.write(randomData.data(), 0, DATASIZE_BYTES);
  }
  DataNodeView view(blockStore->load(key));
  EXPECT_EQ(3, view.Depth());
  EXPECT_EQ(1000000000u, view.Size());
  EXPECT_EQ(0, std::memcmp(view.data(), randomData.data(), DATASIZE_BYTES));
}

TEST_F(DataNodeViewTest, DataBeginWorksWithOneByteEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  EXPECT_EQ(blockBegin+DataNodeLayout::HEADERSIZE_BYTES, view.DataBegin<uint8_t>());
}

TEST_F(DataNodeViewTest, DataBeginWorksWithEightByteEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  EXPECT_EQ(blockBegin+DataNodeLayout::HEADERSIZE_BYTES, (uint8_t*)view.DataBegin<uint64_t>());
}

TEST_F(DataNodeViewTest, DataEndWorksWithOneByteEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  EXPECT_EQ(blockBegin+BLOCKSIZE_BYTES, view.DataEnd<uint8_t>());
}

TEST_F(DataNodeViewTest, DataEndWorksWithEightByteEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  EXPECT_EQ(blockBegin+BLOCKSIZE_BYTES, (uint8_t*)view.DataEnd<uint64_t>());
}

struct SizedDataEntry {
  uint8_t data[6];
};
BOOST_STATIC_ASSERT_MSG(DataNodeViewTest::DATASIZE_BYTES % sizeof(SizedDataEntry) != 0,
  "This test case only makes sense, if the data entries don't fill up the whole space. "
  "There should be some space left at the end that is not used, because it isn't enough space for a full entry. "
  "If this static assertion fails, please use a different size for SizedDataEntry.");

TEST_F(DataNodeViewTest, DataBeginWorksWithStructEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  EXPECT_EQ(blockBegin+DataNodeLayout::HEADERSIZE_BYTES, (uint8_t*)view.DataBegin<SizedDataEntry>());
}

TEST_F(DataNodeViewTest, DataEndWorksWithStructByteEntries) {
  auto block = blockStore->create(Data(BLOCKSIZE_BYTES));
  uint8_t *blockBegin = (uint8_t*)block->data();
  DataNodeView view(std::move(block));

  unsigned int numFittingEntries = DATASIZE_BYTES / sizeof(SizedDataEntry);

  uint8_t *dataEnd = (uint8_t*)view.DataEnd<SizedDataEntry>();
  EXPECT_EQ(blockBegin+DataNodeLayout::HEADERSIZE_BYTES + numFittingEntries * sizeof(SizedDataEntry), dataEnd);
  EXPECT_LT(dataEnd, blockBegin + BLOCKSIZE_BYTES);
}

//TODO Test that header fields (and data) are also stored over reloads