/*
 * TestPack.cpp -- test the pack serializer
 *
 * Copyright (c) 2013, 2014 David Demelier <markand@malikania.fr>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sstream>
#include <vector>

#include <gtest/gtest.h>

#include <Pack.h>

struct Point
{
	uint32_t width{};
	uint32_t height{};

	Point() = default;

	Point(uint32_t width, uint32_t height)
		: width(width)
		, height(height)
	{
	}

	inline bool operator==(const Point &other) const
	{
		return width == other.width && height == other.height;
	}
};

template <>
struct Pack::TypeInfo<Point> : public Pack::Serializable
{
	static void serialize(PackWriter &writer, const Point &point)
	{
		writer << point.width << point.height;
	}

	static void unserialize(PackReader &reader, Point &point)
	{
		reader >> point.width >> point.height;
	}
};

TEST(File, simpleLittleEndian)
{
	uint8_t u8(1), r8;
	uint16_t u16(2), r16;
	uint32_t u32(3), r32;
	uint64_t u64(4), r64;

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Little};
			writer << u8 << u16 << u32 << u64;
		}

		PackFileReader reader{"output.bin", Pack::Little};
		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(File, simpleBigEndian)
{
	uint8_t u8(1), r8;
	uint16_t u16(2), r16;
	uint32_t u32(3), r32;
	uint64_t u64(4), r64;

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Big};
			writer << u8 << u16 << u32 << u64;
		}

		PackFileReader reader{"output.bin", Pack::Big};
		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(File, arrayLittleEndian)
{
	std::vector<uint8_t> u8 { 1, 2, 3, 4, 5, 6, 7, 8 };
	std::vector<uint16_t> u16 { 10, 20, 30, 40, 50, 60, 70, 80 };
	std::vector<uint32_t> u32 { 100, 200, 300, 400, 500, 600, 700, 800 };
	std::vector<uint64_t> u64 { 1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000 };

	std::vector<uint8_t> r8(8);
	std::vector<uint16_t> r16(8);
	std::vector<uint32_t> r32(8);
	std::vector<uint64_t> r64(8);

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Little};

			writer << u8 << u16 << u32 << u64;
		}

		PackFileReader reader{"output.bin", Pack::Little};

		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(File, arrayBigEndian)
{
	std::vector<uint8_t> u8 { 1, 2, 3, 4, 5, 6, 7, 8 };
	std::vector<uint16_t> u16 { 10, 20, 30, 40, 50, 60, 70, 80 };
	std::vector<uint32_t> u32 { 100, 200, 300, 400, 500, 600, 700, 800 };
	std::vector<uint64_t> u64 { 1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000 };

	std::vector<uint8_t> r8(8);
	std::vector<uint16_t> r16(8);
	std::vector<uint32_t> r32(8);
	std::vector<uint64_t> r64(8);

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Big};

			writer << u8 << u16 << u32 << u64;
		}

		PackFileReader reader{"output.bin", Pack::Big};

		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(File, serializeSimpleLittleEndian)
{
	Point point{200, 400};
	Point result;

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Little};

			writer << point;
		}

		PackFileReader reader{"output.bin", Pack::Little};

		reader >> result;

		ASSERT_EQ(point, result);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(File, serializeSimpleBigEndian)
{
	Point point{200, 400};
	Point result;

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Big};

			writer << point;
		}

		PackFileReader reader{"output.bin", Pack::Big};

		reader >> result;

		ASSERT_EQ(point, result);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(File, serializeArrayLittleEndian)
{
	std::vector<Point> points{{10, 20}, {30, 40}};
	std::vector<Point> rpoints(2);

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Little};

			writer << points;
		}

		PackFileReader reader{"output.bin", Pack::Little};

		reader >> rpoints;

		ASSERT_EQ(points, rpoints);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(File, serializeArrayBigEndian)
{
	std::vector<Point> points{{10, 20}, {30, 40}};
	std::vector<Point> rpoints(2);

	remove("output.bin");

	try {
		{
			PackFileWriter writer{"output.bin", Pack::Big};

			writer << points;
		}

		PackFileReader reader{"output.bin", Pack::Big};

		reader >> rpoints;

		ASSERT_EQ(points, rpoints);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(String, simpleLittleEndian)
{
	uint8_t u8(1), r8;
	uint16_t u16(2), r16;
	uint32_t u32(3), r32;
	uint64_t u64(4), r64;

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Little};
			writer << u8 << u16 << u32 << u64;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Little};
		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(String, simpleBigEndian)
{
	uint8_t u8(1), r8;
	uint16_t u16(2), r16;
	uint32_t u32(3), r32;
	uint64_t u64(4), r64;

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Big};
			writer << u8 << u16 << u32 << u64;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Big};
		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(String, arrayLittleEndian)
{
	std::vector<uint8_t> u8 { 1, 2, 3, 4, 5, 6, 7, 8 };
	std::vector<uint16_t> u16 { 10, 20, 30, 40, 50, 60, 70, 80 };
	std::vector<uint32_t> u32 { 100, 200, 300, 400, 500, 600, 700, 800 };
	std::vector<uint64_t> u64 { 1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000 };

	std::vector<uint8_t> r8(8);
	std::vector<uint16_t> r16(8);
	std::vector<uint32_t> r32(8);
	std::vector<uint64_t> r64(8);

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Little};

			writer << u8 << u16 << u32 << u64;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Little};

		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(String, arrayBigEndian)
{
	std::vector<uint8_t> u8 { 1, 2, 3, 4, 5, 6, 7, 8 };
	std::vector<uint16_t> u16 { 10, 20, 30, 40, 50, 60, 70, 80 };
	std::vector<uint32_t> u32 { 100, 200, 300, 400, 500, 600, 700, 800 };
	std::vector<uint64_t> u64 { 1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000 };

	std::vector<uint8_t> r8(8);
	std::vector<uint16_t> r16(8);
	std::vector<uint32_t> r32(8);
	std::vector<uint64_t> r64(8);

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Big};

			writer << u8 << u16 << u32 << u64;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Big};

		reader >> r8 >> r16 >> r32 >> r64;

		ASSERT_EQ(u8, r8);
		ASSERT_EQ(u16, r16);
		ASSERT_EQ(u32, r32);
		ASSERT_EQ(u64, r64);
	} catch (const std::exception &error) {
		FAIL() << error.what();
	}
}

TEST(String, serializeSimpleLittleEndian)
{
	Point point{200, 400};
	Point result;

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Little};

			writer << point;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Little};

		reader >> result;

		ASSERT_EQ(point, result);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(String, serializeSimpleBigEndian)
{
	Point point{200, 400};
	Point result;

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Big};

			writer << point;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Big};

		reader >> result;

		ASSERT_EQ(point, result);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(String, serializeArrayLittleEndian)
{
	std::vector<Point> points{{10, 20}, {30, 40}};
	std::vector<Point> rpoints(2);

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Little};

			writer << points;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Little};

		reader >> rpoints;

		ASSERT_EQ(points, rpoints);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

TEST(String, serializeArrayBigEndian)
{
	std::vector<Point> points{{10, 20}, {30, 40}};
	std::vector<Point> rpoints(2);

	try {
		std::string input;

		{
			PackStringWriter writer{Pack::Big};

			writer << points;
			input = writer.buffer();
		}

		PackStringReader reader{std::move(input), Pack::Big};

		reader >> rpoints;

		ASSERT_EQ(points, rpoints);
	} catch (const std::exception &ex) {
		std::cerr << "warning: " << ex.what() << std::endl;
	}
}

int main(int argc, char **argv)
{
	testing::InitGoogleTest(&argc, argv);

	return RUN_ALL_TESTS();
}