#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>

#include "Swiften/Parser/StanzaParser.h"
#include "Swiften/Parser/GenericPayloadParser.h"
#include "Swiften/Parser/PayloadParserFactory.h"
#include "Swiften/Parser/PayloadParserFactoryCollection.h"
#include "Swiften/Elements/Stanza.h"
#include "Swiften/Elements/Payload.h"

using namespace Swift;

class StanzaParserTest : public CppUnit::TestFixture
{
		CPPUNIT_TEST_SUITE(StanzaParserTest);
		CPPUNIT_TEST(testHandleEndElement_OnePayload);
		CPPUNIT_TEST(testHandleEndElement_MultiplePayloads);
		CPPUNIT_TEST(testHandleEndElement_StrayCharacterData);
		CPPUNIT_TEST(testHandleEndElement_UnknownPayload);
		CPPUNIT_TEST(testHandleParse_BasicAttributes);
		CPPUNIT_TEST_SUITE_END();

	public:
		StanzaParserTest() {}

		void setUp() {
			factoryCollection_ = new PayloadParserFactoryCollection();
			factoryCollection_->addFactory(&factory1_);
			factoryCollection_->addFactory(&factory2_);
		}

		void tearDown() {
			delete factoryCollection_;
		}

		void testHandleEndElement_OnePayload() {
			MyStanzaParser testling(factoryCollection_);

			AttributeMap attributes;
			attributes["foo"] = "fum";
			attributes["bar"] = "baz";
			testling.handleStartElement("mystanza", "", attributes);
			testling.handleStartElement("mypayload1", "", attributes);
			testling.handleStartElement("child", "", attributes);
			testling.handleEndElement("child", "");
			testling.handleEndElement("mypayload1", "");
			testling.handleEndElement("mystanza", "");

			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload1>());
			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload1>()->hasChild);
		}

		void testHandleEndElement_MultiplePayloads() {
			MyStanzaParser testling(factoryCollection_);

			AttributeMap attributes;
			testling.handleStartElement("mystanza", "", attributes);
			testling.handleStartElement("mypayload1", "", attributes);
			testling.handleEndElement("mypayload1", "");
			testling.handleStartElement("mypayload2", "", attributes);
			testling.handleEndElement("mypayload2", "");
			testling.handleEndElement("mystanza", "");

			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload1>());
			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload2>());
		}

		void testHandleEndElement_StrayCharacterData() {
			MyStanzaParser testling(factoryCollection_);

			AttributeMap attributes;
			testling.handleStartElement("mystanza", "", attributes);
			testling.handleStartElement("mypayload1", "", attributes);
			testling.handleEndElement("mypayload1", "");
			testling.handleCharacterData("bla");
			testling.handleStartElement("mypayload2", "", attributes);
			testling.handleEndElement("mypayload2", "");
			testling.handleEndElement("mystanza", "");

			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload1>());
			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload2>());
		}

		void testHandleEndElement_UnknownPayload() {
			MyStanzaParser testling(factoryCollection_);

			AttributeMap attributes;
			testling.handleStartElement("mystanza", "", attributes);
			testling.handleStartElement("mypayload1", "", attributes);
			testling.handleEndElement("mypayload1", "");
			testling.handleStartElement("unknown-payload", "", attributes);
			testling.handleStartElement("unknown-payload-child", "", attributes);
			testling.handleEndElement("unknown-payload-child", "");
			testling.handleEndElement("unknown-payload", "");
			testling.handleStartElement("mypayload2", "", attributes);
			testling.handleEndElement("mypayload2", "");
			testling.handleEndElement("mystanza", "");

			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload1>());
			CPPUNIT_ASSERT(testling.getStanza()->getPayload<MyPayload2>());
		}

		void testHandleParse_BasicAttributes() {
			MyStanzaParser testling(factoryCollection_);

			AttributeMap attributes;
			attributes["to"] = "foo@example.com/blo";
			attributes["from"] = "bar@example.com/baz";
			attributes["id"] = "id-123";
			testling.handleStartElement("mystanza", "", attributes);
			testling.handleEndElement("mypayload1", "");

			CPPUNIT_ASSERT_EQUAL(JID("foo@example.com/blo"), testling.getStanza()->getTo());
			CPPUNIT_ASSERT_EQUAL(JID("bar@example.com/baz"), testling.getStanza()->getFrom());
			CPPUNIT_ASSERT_EQUAL(String("id-123"), testling.getStanza()->getID());
		}

	private:
		class MyPayload1 : public Payload
		{
			public:
				MyPayload1() : hasChild(false) {}

				bool hasChild;
		};

		class MyPayload1Parser : public GenericPayloadParser<MyPayload1>
		{
			public:	
				MyPayload1Parser() {}

			virtual void handleStartElement(const String& element, const String&, const AttributeMap&) { 
				if (element != "mypayload1") {
					getPayloadInternal()->hasChild = true;
				}
			}

			virtual void handleEndElement(const String&, const String&) {}
			virtual void handleCharacterData(const String&) {}
		};

		class MyPayload1ParserFactory : public PayloadParserFactory
		{
			public:
				MyPayload1ParserFactory() {}

				PayloadParser* createPayloadParser() { return new MyPayload1Parser(); }

				bool canParse(const String& element, const String&, const AttributeMap&) const {
					return element == "mypayload1";
				}
		};

		class MyPayload2 : public Payload
		{
			public:
				MyPayload2() {}
		};

		class MyPayload2Parser : public GenericPayloadParser<MyPayload2>
		{
			public:	
				MyPayload2Parser() {}

			virtual void handleStartElement(const String&, const String&, const AttributeMap&) {}
			virtual void handleEndElement(const String&, const String&) {}
			virtual void handleCharacterData(const String&) {}
		};


		class MyPayload2ParserFactory : public PayloadParserFactory
		{
			public:
				MyPayload2ParserFactory() {}

				PayloadParser* createPayloadParser() { return new MyPayload2Parser(); }
				bool canParse(const String& element, const String&, const AttributeMap&) const {
					return element == "mypayload2";
				}
		};

		class MyStanza : public Stanza
		{
			public:
				MyStanza() {}
		};

		class MyStanzaParser : public StanzaParser
		{
			public:
				MyStanzaParser(PayloadParserFactoryCollection* collection) : StanzaParser(collection)
				{
					stanza_ = boost::shared_ptr<MyStanza>(new MyStanza());
				}
			
				virtual boost::shared_ptr<Element> getElement() const {
					return stanza_;
				}

			private:
				boost::shared_ptr<MyStanza> stanza_;
		};
	
		MyPayload1ParserFactory factory1_;
		MyPayload2ParserFactory factory2_;
		PayloadParserFactoryCollection* factoryCollection_;
};

CPPUNIT_TEST_SUITE_REGISTRATION(StanzaParserTest);