// Unit Test
//
// Douglas Thrift
//
// $Id$

#include <string>
#include <sstream>
#include <cassert>
#include "DoublyLinkedList.hpp"

using namespace std;

template<typename Type>
inline bool contains(const Type& value, DoublyLinkedList<Type>& list)
{
	for (Iterator<Type> itor(list.iterator()); !itor.end(); itor++)
	{
		if (*itor == value) return true;
	}

	return false;
}

class VowelVisitor : public Visitor<string>
{
private:
	ostream* os;
public:
	VowelVisitor(ostream& os) { this->os = &os; }
	void visit(const string& value) { if (value.find_first_of("aAeEiIoOuU") == 0) *os << value << '\n'; }
};

int main(int argc, char* argv[])
{
	DoublyLinkedList<int> list;

	list.addFront(1);
	list.addBack(2);
	list.addFront(4);
	list.addBack(5);

	// 4 1 2 5

	assert(list.contains(1) == true);
	assert(list.contains(2) == true);
	assert(list.contains(3) == false);
	assert(list.contains(4) == true);
	assert(list.contains(5) == true);

	list.removeFirst(2);

	// 4 1 5

	assert(list.contains(2) == false);
	assert(list.contains(5) == true);

	ostringstream sout;

	for (Iterator<int> itor(list.iterator()); !itor.end(); itor++)
	{
		sout << *itor << '\n';
	}

	assert(sout.str() == "4\n1\n5\n");
	sout.str("");
	assert(contains(3, list) == false);
	assert(contains(4, list) == true);

	DoublyLinkedList<string> words;

	words.addFront("Douglas");
	words.addBack("Allen");
	words.addFront("Edward");
	words.addBack("attack");
	words.addFront("Thrift");

	for (Iterator<string> itor(words.iterator()); !itor.end(); itor++)
	{
		if (itor->find_first_of("aAeEiIoOuU") == 0) sout << *itor << '\n';
	}

	assert(sout.str() == "Edward\nAllen\nattack\n");
	sout.str("");

	VowelVisitor vowels(sout);

	words.accept(dynamic_cast<Visitor<string>*>(&vowels));

	assert(sout.str() == "Edward\nAllen\nattack\n");

	return 0;
}