//
//    FILE: unit_test_cyclic_store.cpp
//  AUTHOR: Tomas Hübner
//    DATE: 2021-01-06
// PURPOSE: unit test for the I2C_eeprom_cyclic_store class of the I2C_EEPROM library
//          https://github.com/Arduino-CI/arduino_ci/blob/master/REFERENCE.md
//

// Note: The implementation of the I2C_eeprom_cyclic_store should have been
// made injactable so that a mock or fake for I2C_eeprom could have been used
// instead of having to rely on a test implementation of the Wire singleton.
// 
// This could have been done by changing the class to take two template
// parameters, the first being one for I2C_eeprom or it's mock and the second
// being the data type to store. For simplicity an alias could have been
// introduced so that users wouldn't have to specify I2C_eeprom but that made
// the editor lose track of the intellisense comments. :-/ 
//
// Thus this remains until someone comes up with a brighter idea.

#include <ArduinoUnitTests.h>

#include "Arduino.h"
#include "I2C_eeprom.h"
#include "I2C_eeprom_cyclic_store.h"

#define I2C_EEPROM_ADDR 0x50
#define I2C_EEPROM_SIZE 0x1000 // 4096

struct DummyTestData {
  uint8_t padding;
};

unittest_setup()
{
}

unittest_teardown()
{
}

/**
 * Verify that I2C_eeprom_cyclic_store fails to 
 * initialize when unable to read from eeprom.
 */
unittest(cyclic_store_fails_begin)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(false, CS.begin(EE, 32, 4));
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully 
 * initializes on empty eeprom.
 */
unittest(cyclic_store_empty_begin)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();
  
  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));
  assertEqual(2, mosi->size());
}

/**
 * Verify that I2C_eeprom_cyclic_store calculate
 * number of slots correctly
 */
unittest(cyclic_store_double_page_buffer)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(2, slots);
  assertEqual(0, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully finds
 * the latest version when there is only one in the eeprom.
 */
unittest(cyclic_store_finds_single_stored_version)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);

  uint32_t tmp = 0;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(1, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully finds
 * the last entry before the entire eeprom has been filled.
 */
unittest(cyclic_store_finds_last_half_filled_eeprom)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);

  uint32_t tmp = 1;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 2;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 3;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(4, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully finds the last
 * version when it was written to the last slot of the memory.
 */
unittest(cyclic_store_finds_version_if_last_slot)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);

  uint32_t tmp = 9;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 10;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 11;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 12;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(13, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully finds the last
 * entry after writes have wrapped around the end of the eeprom.
 */
unittest(cyclic_store_finds_last_wrapped_eeprom)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);

  uint32_t tmp = 9;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 10;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 11;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 8;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(12, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store successfully finds the last
 * entry after writes have wrapped around the end of the eeprom
 * when using a buffer that spans two pages. And that it does so
 * even if the number of pages is not evenly divisable by the 
 * number of pages the buffer span.
 */
unittest(cyclic_store_finds_last_wrapped_eeprom_double_page_buffer)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);

  uint32_t tmp = 9;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 10;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 11;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  tmp = 8;
  miso->push_back(((uint8_t*)&tmp)[0]);
  miso->push_back(((uint8_t*)&tmp)[1]);
  miso->push_back(((uint8_t*)&tmp)[2]);
  miso->push_back(((uint8_t*)&tmp)[3]);

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  assertEqual(true, CS.begin(EE, 32, 9));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(12, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store fails to
 * initialize if buffer too large
 */
unittest(cyclic_store_fails_init_on_too_large_buffer)
{
  Wire.resetMocks();

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[400]> CS;
  assertEqual(false, CS.begin(EE, 32, 4));
}

/**
 * Verify that I2C_eeprom_cyclic_store reports
 * the correct metrics on empty eeprom.
 */
unittest(cyclic_store_empty_metrics)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  CS.getMetrics(slots, writes);

  assertEqual(4, slots);
  assertEqual(0, writes);
}

/**
 * Verify that I2C_eeprom_cyclic_store fails to
 * fetch metrics if initialization has failed.
 */
unittest(cyclic_store_fails_metrics_when_failed_initialize)
{
  Wire.resetMocks();

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[400]> CS;
  assertEqual(false, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t writes;

  assertEqual(false, CS.getMetrics(slots, writes));
}

/**
 * Verify that I2C_eeprom_cyclic_store fails to
 * fetch metrics if initialization has failed.
 */
unittest(cyclic_store_fails_metrics_when_not_initialized)
{
  Wire.resetMocks();

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[400]> CS;

  uint16_t slots;
  uint32_t writes;

  assertEqual(false, CS.getMetrics(slots, writes));
}

/**
 * Verify that I2C_eeprom_cyclic_store correctly 
 * increments the number of writes when saving
 * new versions.
 */
unittest(cyclic_store_increments_write_counter)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<DummyTestData> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint16_t slots;
  uint32_t originalWrites;

  CS.getMetrics(slots, originalWrites);

  DummyTestData data;

  CS.write(data);
  CS.write(data);

  uint32_t newWrites;

  CS.getMetrics(slots, newWrites);

  assertEqual(4, slots);
  assertEqual(2, newWrites);
  assertNotEqual(originalWrites, newWrites);
}

/**
 * Check that the format() call of I2C_eeprom_cyclic_store
 * writes to the correct places when buffer size is smaller
 * than page size
 */
unittest(cyclic_store_format_single_page)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[20]> CS;
  CS.begin(EE, 32, 4);

  mosi->clear();

  CS.format();

  // It should write exactly 24 bytes to the eeprom
  assertEqual(24, mosi->size());

  // CHeck that it writes empty marker to 0, 32, 64 and 96
  uint8_t expected[24] = {0,0,0xff,0xff,0xff,0xff,0,32,0xff,0xff,0xff,0xff,0,64,0xff,0xff,0xff,0xff,0,96,0xff,0xff,0xff,0xff};

  for(int i = 0; i < 24; i++)
  {
    assertEqual(expected[i], mosi->front());
    mosi->pop_front();
  }
}

/**
 * Check that the format() call of I2C_eeprom_cyclic_store
 * writes to the correct places when buffer is two pages large.
 */
unittest(cyclic_store_format_double_page)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  CS.begin(EE, 32, 4);

  mosi->clear();

  CS.format();

  // It should write exactly 12 bytes to the eeprom
  assertEqual(12, mosi->size());

  // CHeck that it writes empty marker to 0 and 64
  uint8_t expected[12] = {0,0,0xff,0xff,0xff,0xff,0,64,0xff,0xff,0xff,0xff};

  for(int i = 0; i < 12; i++)
  {
    assertEqual(expected[i], mosi->front());
    mosi->pop_front();
  }
}

/**
 * Check that the format() call of I2C_eeprom_cyclic_store
 * writes to the correct places when the buffer is two pages
 * large and an odd number of pages is available.
 */
unittest(cyclic_store_format_double_page_odd_space)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  CS.begin(EE, 32, 5);

  mosi->clear();

  CS.format();

  // It should write exactly 12 bytes to the eeprom
  assertEqual(12, mosi->size());

  // CHeck that it writes empty marker to 0 and 64
  uint8_t expected[12] = {0,0,0xff,0xff,0xff,0xff,0,64,0xff,0xff,0xff,0xff};

  for(int i = 0; i < 12; i++)
  {
    assertEqual(expected[i], mosi->front());
    mosi->pop_front();
  }
}

/**
 * Check that writing buffers to the store makes it wrap
 * to the start of the memory once it has been filled
 * using a single page buffer.
 */
unittest(cyclic_store_wrapping_single_page)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();
  
  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<uint8_t[20]> CS;
  assertEqual(true, CS.begin(EE, 32, 4));
  assertEqual(2, mosi->size());

  uint8_t dummy[20];

  CS.write(dummy);
  CS.write(dummy);
  CS.write(dummy);
  CS.write(dummy);

  mosi->clear();

  CS.write(dummy);

  // It should write exactly 26 bytes to the eeprom (addr+header+buffer)
  assertEqual(26, mosi->size());

  // check that the address is 0
  assertEqual(0, mosi->front());
  mosi->pop_front();
  assertEqual(0, mosi->front());
  mosi->pop_front();
}

/**
 * Check that writing buffers to the store makes it wrap
 * to the start of the memory once it has been filled
 * using a single page buffer.
 */
unittest(cyclic_store_wrapping_double_page)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();
  
  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  assertEqual(true, CS.begin(EE, 32, 4));

  uint8_t dummy[40];

  CS.write(dummy);
  CS.write(dummy);

  mosi->clear();

  CS.write(dummy);

  // It should write exactly 50 bytes to the eeprom (addr+header+buffer+wait+addr+remainder)
  assertEqual(50, mosi->size());

  // check that the address is 0
  assertEqual(0, mosi->front());
  mosi->pop_front();
  assertEqual(0, mosi->front());
  mosi->pop_front();
}

/**
 * Check that writing buffers to the store makes it wrap
 * to the start of the memory once it has been filled
 * using a single page buffer.
 */
unittest(cyclic_store_wrapping_double_page_odd_space)
{
  Wire.resetMocks();

  auto mosi = Wire.getMosi(I2C_EEPROM_ADDR);

  I2C_eeprom EE(I2C_EEPROM_ADDR, I2C_EEPROM_SIZE);
  EE.begin();
  
  auto miso = Wire.getMiso(I2C_EEPROM_ADDR);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);
  miso->push_back(0xff);

  I2C_eeprom_cyclic_store<uint8_t[40]> CS;
  assertEqual(true, CS.begin(EE, 32, 5));

  uint8_t dummy[40];

  CS.write(dummy);
  CS.write(dummy);

  mosi->clear();

  CS.write(dummy);

  // It should write exactly 50 bytes to the eeprom (addr+header+buffer+wait+addr+remainder)
  assertEqual(50, mosi->size());

  // check that the address is 0
  assertEqual(0, mosi->front());
  mosi->pop_front();
  assertEqual(0, mosi->front());
  mosi->pop_front();
}

unittest_main()

// --------