#include "IntTestSerialization.h"
#include <fsfw/unittest/internal/UnittDefinitions.h>

#include <fsfw/serialize/SerializeElement.h>
#include <fsfw/serialize/SerialBufferAdapter.h>
#include <fsfw/serialize/SerializeIF.h>

#include <array>

using retval = HasReturnvaluesIF;
std::array<uint8_t, 512> testserialize::test_array = { 0 };

ReturnValue_t testserialize::test_serialization() {
	// Here, we test all serialization tools. First test basic cases.
	ReturnValue_t result = test_endianness_tools();
	if(result != retval::RETURN_OK) {
		return result;
	}
	result = test_autoserialization();
	if(result != retval::RETURN_OK) {
		return result;
	}
	result = test_serial_buffer_adapter();
	if(result != retval::RETURN_OK) {
		return result;
	}
	return retval::RETURN_OK;
}

ReturnValue_t testserialize::test_endianness_tools() {
	std::string id = "[test_endianness_tools]";
	test_array[0] = 0;
	test_array[1] = 0;
	uint16_t two_byte_value = 1;
	size_t size = 0;
	uint8_t* p_array = test_array.data();
	SerializeAdapter::serialize(&two_byte_value, &p_array, &size, 2,
	        SerializeIF::Endianness::MACHINE);
	// Little endian: Value one on first byte
	if(test_array[0] != 1 and test_array[1] != 0) {
		return unitt::put_error(id);

	}

	p_array = test_array.data();
	size = 0;
	SerializeAdapter::serialize(&two_byte_value, &p_array, &size, 2,
	        SerializeIF::Endianness::BIG);
	// Big endian: Value one on second byte
	if(test_array[0] != 0 and test_array[1] != 1) {
		return unitt::put_error(id);
	}
	return retval::RETURN_OK;
}

ReturnValue_t testserialize::test_autoserialization() {
	std::string id = "[test_autoserialization]";
	// Unit Test getSerializedSize
	if(SerializeAdapter::
			getSerializedSize(&tv::tv_bool) != sizeof(tv::tv_bool) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_uint8) != sizeof(tv::tv_uint8) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_uint16) != sizeof(tv::tv_uint16) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_uint32) != sizeof(tv::tv_uint32) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_uint64) != sizeof(tv::tv_uint64) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_int8) != sizeof(tv::tv_int8) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_double) != sizeof(tv::tv_double) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_int16) != sizeof(tv::tv_int16) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_int32) != sizeof(tv::tv_int32) or
			SerializeAdapter::
			getSerializedSize(&tv::tv_float) != sizeof(tv::tv_float))
	{
		return unitt::put_error(id);
	}

	size_t serialized_size = 0;
	uint8_t * p_array = test_array.data();

	SerializeAdapter::serialize(&tv::tv_bool, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_uint8, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_uint16, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_uint32, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_int8, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_int16, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_int32, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_uint64, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_float, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_double, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_sfloat, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv::tv_sdouble, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	// expected size is 1 + 1 + 2 + 4 + 1 + 2 + 4 + 8 + 4 + 8 + 4 + 8
	if(serialized_size != 47) {
		return unitt::put_error(id);
	}

	p_array = test_array.data();
	size_t remaining_size = serialized_size;
	bool tv_bool;
	uint8_t tv_uint8;
	uint16_t tv_uint16;
	uint32_t tv_uint32;
	int8_t tv_int8;
	int16_t tv_int16;
	int32_t tv_int32;
	uint64_t tv_uint64;
	float tv_float;
	double tv_double;
	float tv_sfloat;
	double tv_sdouble;

	SerializeAdapter::deSerialize(&tv_bool,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_uint8,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_uint16,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_uint32,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_int8,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_int16,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_int32,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_uint64,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_float,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_double,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_sfloat,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);
	SerializeAdapter::deSerialize(&tv_sdouble,
			const_cast<const uint8_t**>(&p_array), &remaining_size, SerializeIF::Endianness::MACHINE);

	if(tv_bool != tv::tv_bool or tv_uint8 != tv::tv_uint8 or
			tv_uint16 != tv::tv_uint16 or tv_uint32 != tv::tv_uint32 or
			tv_uint64 != tv::tv_uint64 or tv_int8 != tv::tv_int8 or
			tv_int16 != tv::tv_int16 or tv_int32 != tv::tv_int32)
	{
		return unitt::put_error(id);
	}

	// These epsilon values were just guessed.. It appears to work though.
	if(abs(tv_float - tv::tv_float) > 0.0001 or
			abs(tv_double - tv::tv_double) > 0.01 or
			abs(tv_sfloat - tv::tv_sfloat) > 0.0001 or
			abs(tv_sdouble - tv::tv_sdouble) > 0.01) {
		return unitt::put_error(id);
	}

	// Check overflow
	return retval::RETURN_OK;
}

// TODO: Also test for constant buffers.
ReturnValue_t testserialize::test_serial_buffer_adapter() {
	std::string id = "[test_serial_buffer_adapter]";

	// I will skip endian swapper testing, its going to be changed anyway..
	// uint8_t tv::tv_uint8_swapped = EndianSwapper::swap(tv::tv_uint8);

	size_t serialized_size = 0;
	uint8_t * p_array = test_array.data();
	std::array<uint8_t, 5> test_serial_buffer {5, 4, 3, 2, 1};
	SerialBufferAdapter<uint8_t> tv_serial_buffer_adapter =
			SerialBufferAdapter<uint8_t>(test_serial_buffer.data(),
					test_serial_buffer.size(), false);
	uint16_t testUint16 = 16;

	SerializeAdapter::serialize(&tv::tv_bool, &p_array,&serialized_size,
			test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv_serial_buffer_adapter, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&testUint16, &p_array, &serialized_size,
			test_array.size(), SerializeIF::Endianness::MACHINE);

	if(serialized_size != 8 or test_array[0] != true or test_array[1] != 5
			or test_array[2] != 4 or test_array[3] != 3 or test_array[4] != 2
			or test_array[5] != 1)
	{
		return unitt::put_error(id);
	}
	memcpy(&testUint16, test_array.data() + 6, sizeof(testUint16));
	if(testUint16 != 16) {
		return unitt::put_error(id);
	}

	// Serialize with size field
	SerialBufferAdapter<uint8_t> tv_serial_buffer_adapter2 =
			SerialBufferAdapter<uint8_t>(test_serial_buffer.data(),
					test_serial_buffer.size(), true);
	serialized_size = 0;
	p_array = test_array.data();
	SerializeAdapter::serialize(&tv::tv_bool, &p_array,&serialized_size,
			test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&tv_serial_buffer_adapter2, &p_array,
			&serialized_size, test_array.size(), SerializeIF::Endianness::MACHINE);
	SerializeAdapter::serialize(&testUint16, &p_array, &serialized_size,
			test_array.size(), SerializeIF::Endianness::MACHINE);

	if(serialized_size != 9 or test_array[0] != true or test_array[1] != 5
			or test_array[2] != 5 or test_array[3] != 4 or test_array[4] != 3
			or test_array[5] != 2 or test_array[6] != 1)
	{
		return unitt::put_error(id);
	}
	memcpy(&testUint16, test_array.data() + 7, sizeof(testUint16));
	if(testUint16 != 16) {
		return unitt::put_error(id);
	}
	return retval::RETURN_OK;
}