/*-
 * Copyright 2012-2018 Matthew Endsley
 * All rights reserved
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted providing that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <tinystl/vector.h>
#include <UnitTest++.h>
#include <algorithm>
#include <string.h>
#include <stdlib.h>

struct complex {
	complex() {data = 0;}
	complex(const char* s) { data = strdup(s); }
	~complex() { free(data); }
	complex(const complex& other) { data = 0; if (other.data) data = strdup(other.data); }
	complex& operator=(const complex& other) { complex(other).swap(*this); return *this; }
	void swap(complex& other) { std::swap(data, other.data); }

	char* data;
};

static inline bool operator==(const complex& lhs, const complex& rhs) {
	if (lhs.data == 0 && rhs.data == 0)
		return true;
	if (lhs.data != 0 && rhs.data != 0)
		return 0 == strcmp(lhs.data, rhs.data);
	return false;
}

TEST(vector_complex_constructor) {
	typedef tinystl::vector<complex> vector;

	{
		vector v;
		CHECK( v.empty() );
		CHECK( v.size() == 0 );
	}
	{
		const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
		vector v(array, array + 10);

		CHECK( v.size() == 10 );
		CHECK( std::equal(v.begin(), v.end(), array) );
	}
	{
		const complex value = "127";
		const size_t count = 24;
		vector v(count, value);

		CHECK( v.size() == count );

		vector::iterator it = v.begin(), end = v.end();
		for (; it != end; ++it)
			CHECK(*it == value);
	}
	{
		const size_t count = 24;
		vector v(count);

		CHECK(v.size() == count);
		vector::iterator it = v.begin(), end = v.end();
		for (; it != end; ++it)
			CHECK(*it == complex());
	}
	{
		const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
		vector other(array, array + 10);
		vector v = other;

		CHECK( v.size() == other.size() );
		CHECK( std::equal(v.begin(), v.end(), other.begin()) );
	}
}

TEST(vector_complex_assignment) {
	typedef tinystl::vector<complex> vector;

	{
		const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
		vector other(array, array + 10);

		vector v;
		v = other;

		CHECK( v.size() == 10 );
		CHECK( std::equal(v.begin(), v.end(), array) );
		CHECK( other.size() == 10 );
		CHECK( std::equal(v.begin(), v.end(), other.begin()) );
	}
}

TEST(vector_complex_pushback) {
	tinystl::vector<complex> v;
	v.push_back("42");

	CHECK(v.size() == 1);
	CHECK(v[0] == "42");
}

TEST(vector_complex_vector) {
	tinystl::vector< tinystl::vector<complex> > v(10, tinystl::vector<complex>());

	tinystl::vector< tinystl::vector<complex> >::iterator it = v.begin(), end = v.end();
	for (; it != end; ++it) {
		CHECK( (*it).empty() );
		CHECK( (*it).size() == 0 );
		CHECK( (*it).begin() == (*it).end() );
	}
}

TEST(vector_complex_swap) {
	tinystl::vector<complex> v1;
	v1.push_back("12");
	v1.push_back("20");

	tinystl::vector<complex> v2;
	v2.push_back("54");

	v1.swap(v2);

	CHECK(v1.size() == 1);
	CHECK(v2.size() == 2);
	CHECK(v1[0] == "54");
	CHECK(v2[0] == "12");
	CHECK(v2[1] == "20");
}

TEST(vector_complex_popback) {
	tinystl::vector<complex> v;
	v.push_back("12");
	v.push_back("24");

	CHECK(v.back() == "24");
	
	v.pop_back();

	CHECK(v.back() == "12");
	CHECK(v.size() == 1);
}

TEST(vector_complex_assign) {
	tinystl::vector<complex> v;

	CHECK(v.size() == 0);

	const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
	v.assign(array, array + 10);
	CHECK(v.size() == 10);
	CHECK( std::equal(v.begin(), v.end(), array) );
}

TEST(vector_complex_erase) {
	const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
	tinystl::vector<complex> v(array, array + 10);

	tinystl::vector<complex>::iterator it = v.erase(v.begin());
	CHECK(*it == "2");
	CHECK(v.size() == 9);
	CHECK( std::equal(v.begin(), v.end(), array + 1) );

	it = v.erase(v.end() - 1);
	CHECK(it == v.end());
	CHECK(v.size() == 8);
	CHECK( std::equal(v.begin(), v.end(), array + 1) );

	v.erase(v.begin() + 1, v.end() - 1);
	CHECK(v.size() == 2);
	CHECK(v[0] == "2");
	CHECK(v[1] == "9");
}

TEST(vector_complex_erase_unordered) {
	const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
	typedef tinystl::vector<complex> vector;
	vector v(array, array + 10);

	complex first = *(v.begin());
	vector::iterator it = v.erase_unordered(v.begin());
	CHECK(it == v.begin());
	CHECK(v.size() == 9);
	CHECK( std::count(v.begin(), v.end(), first) == 0 );
	for (it = v.begin(); it != v.end(); ++it) {
		CHECK( std::count(v.begin(), v.end(), *it) == 1 );
	}

	complex last = *(v.end() - 1);
	it = v.erase_unordered(v.end() - 1);
	CHECK(it == v.end());
	CHECK(v.size() == 8);
	CHECK( std::count(v.begin(), v.end(), last) == 0 );
	for (it = v.begin(); it != v.end(); ++it) {
		CHECK( std::count(v.begin(), v.end(), *it) == 1 );
	}

	first = *(v.begin());
	last = *(v.end() - 1);
	v.erase_unordered(v.begin() + 1, v.end() - 1);
	CHECK(v.size() == 2);
	CHECK( std::count(v.begin(), v.end(), first) == 1 );
	CHECK( std::count(v.begin(), v.end(), last) == 1 );
}

TEST(vector_complex_insert) {
	const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};
	tinystl::vector<complex> v(array, array + 10);

	v.insert(v.begin(), "0");
	CHECK(v.size() == 11);
	CHECK(v[0] == "0");
	CHECK( std::equal(v.begin() + 1, v.end(), array) );

	v.insert(v.end(), "11");
	CHECK(v.size() == 12);
	CHECK(v[0] == "0");
	CHECK( std::equal(array, array + 10, v.begin() + 1) );
	CHECK(v.back() == "11");

	const complex array2[3] = {"11", "12", "13"};
	const complex finalarray[] = {"0", "1", "2", "3", "11", "12", "13", "4", "5", "6", "7", "8", "9", "10", "11"};
	v.insert(v.begin() + 4, array2, array2 + 3);
	CHECK( v.size() == 15 );
	CHECK( std::equal(v.begin(), v.end(), finalarray) );
}

TEST(vector_complex_iterator) {
	const complex array[10] = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "10"};

	tinystl::vector<complex> v(array, array + 10);
	const tinystl::vector<complex>& cv = v;

	CHECK(v.data() == &*v.begin());
	CHECK(v.data() == &v[0]);
	CHECK(v.data() + v.size() == &*v.end());
	CHECK(v.begin() == cv.begin());
	CHECK(v.end() == cv.end());
	CHECK(v.data() == cv.data());

	tinystl::vector<complex> w = v;
	CHECK(v.begin() != w.begin());
	CHECK(v.end() != w.end());
}
