1 files changed, 186 insertions, 0 deletions
diff --git a/Swiften/Parser/UnitTest/XMPPParserTest.cpp b/Swiften/Parser/UnitTest/XMPPParserTest.cpp
new file mode 100644
index 0000000..ecdd565
--- /dev/null
+++ b/Swiften/Parser/UnitTest/XMPPParserTest.cpp
@@ -0,0 +1,186 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <vector>
+
+#include "Swiften/Elements/ProtocolHeader.h"
+#include "Swiften/Base/String.h"
+#include "Swiften/Parser/XMPPParser.h"
+#include "Swiften/Parser/ElementParser.h"
+#include "Swiften/Parser/XMPPParserClient.h"
+#include "Swiften/Parser/PayloadParserFactoryCollection.h"
+#include "Swiften/Elements/Presence.h"
+#include "Swiften/Elements/IQ.h"
+#include "Swiften/Elements/Message.h"
+#include "Swiften/Elements/StreamFeatures.h"
+#include "Swiften/Elements/UnknownElement.h"
+
+using namespace Swift;
+
+class XMPPParserTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(XMPPParserTest);
+		CPPUNIT_TEST(testParse_SimpleSession);
+		CPPUNIT_TEST(testParse_SimpleClientFromServerSession);
+		CPPUNIT_TEST(testParse_Presence);
+		CPPUNIT_TEST(testParse_IQ);
+		CPPUNIT_TEST(testParse_Message);
+		CPPUNIT_TEST(testParse_StreamFeatures);
+		CPPUNIT_TEST(testParse_UnknownElement);
+		CPPUNIT_TEST(testParse_StrayCharacterData);
+		CPPUNIT_TEST(testParse_InvalidStreamStart);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		XMPPParserTest() {}
+
+		void testParse_SimpleSession() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<?xml version='1.0'?>"));
+			CPPUNIT_ASSERT(testling.parse("<stream:stream to='example.com' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' >"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+			CPPUNIT_ASSERT(testling.parse("<iq/>"));
+			CPPUNIT_ASSERT(testling.parse("</stream:stream>"));
+
+			CPPUNIT_ASSERT_EQUAL(5, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::StreamStart, client_.events[0].type);
+			CPPUNIT_ASSERT_EQUAL(String("example.com"), client_.events[0].to); 
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[1].type);
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[2].type);
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[3].type);
+			CPPUNIT_ASSERT_EQUAL(Client::StreamEnd, client_.events[4].type);
+		}
+
+		void testParse_SimpleClientFromServerSession() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<?xml version='1.0'?>"));
+			CPPUNIT_ASSERT(testling.parse("<stream:stream from='example.com' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' id='aeab'>"));
+
+			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::StreamStart, client_.events[0].type);
+			CPPUNIT_ASSERT_EQUAL(String("example.com"), client_.events[0].from); 
+			CPPUNIT_ASSERT_EQUAL(String("aeab"), client_.events[0].id); 
+		}
+
+
+		void testParse_Presence() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[1].type);
+			CPPUNIT_ASSERT(dynamic_cast<Presence*>(client_.events[1].element.get()));
+		}
+
+		void testParse_IQ() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<iq/>"));
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[1].type);
+			CPPUNIT_ASSERT(dynamic_cast<IQ*>(client_.events[1].element.get()));
+		}
+
+		void testParse_Message() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<message/>"));
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[1].type);
+			CPPUNIT_ASSERT(dynamic_cast<Message*>(client_.events[1].element.get()));
+		}
+
+		void testParse_StreamFeatures() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<stream:features/>"));
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[1].type);
+			CPPUNIT_ASSERT(dynamic_cast<StreamFeatures*>(client_.events[1].element.get()));
+		}
+
+		void testParse_UnknownElement() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+			CPPUNIT_ASSERT(testling.parse("<foo/>"));
+			CPPUNIT_ASSERT(testling.parse("<bar/>"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+
+			CPPUNIT_ASSERT_EQUAL(5, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[2].type);
+			CPPUNIT_ASSERT(dynamic_cast<UnknownElement*>(client_.events[2].element.get()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[3].type);
+			CPPUNIT_ASSERT(dynamic_cast<UnknownElement*>(client_.events[3].element.get()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[4].type);
+			CPPUNIT_ASSERT(dynamic_cast<Presence*>(client_.events[4].element.get()));
+		}
+
+		void testParse_StrayCharacterData() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(testling.parse("<stream:stream xmlns:stream='http://etherx.jabber.org/streams'>"));
+			CPPUNIT_ASSERT(testling.parse("<presence/>"));
+			CPPUNIT_ASSERT(testling.parse("bla"));
+			CPPUNIT_ASSERT(testling.parse("<iq/>"));
+
+			CPPUNIT_ASSERT_EQUAL(3, static_cast<int>(client_.events.size()));
+			CPPUNIT_ASSERT_EQUAL(Client::ElementEvent, client_.events[2].type);
+			CPPUNIT_ASSERT(dynamic_cast<IQ*>(client_.events[2].element.get()));
+		}
+
+		void testParse_InvalidStreamStart() {
+			XMPPParser testling(&client_, &factories_);
+
+			CPPUNIT_ASSERT(!testling.parse("<tream>"));
+		}
+
+	private:
+		class Client : public XMPPParserClient {
+			public:
+				enum Type { StreamStart, ElementEvent, StreamEnd };
+				struct Event {
+					Event(Type type, boost::shared_ptr<Element> element) 
+						: type(type), element(element) {}
+					Event(Type type, const String& from, const String& to, const String& id) : type(type), from(from), to(to), id(id) {}
+
+					Event(Type type) : type(type) {}
+
+					Type type;
+					String from;
+					String to;
+					String id;
+					boost::shared_ptr<Element> element;
+				};
+
+				Client() {}
+
+				void handleStreamStart(const ProtocolHeader& header) {
+					events.push_back(Event(StreamStart, header.getFrom(), header.getTo(), header.getID()));
+				}
+
+				void handleElement(boost::shared_ptr<Element> element) {
+					events.push_back(Event(ElementEvent, element));
+				}
+
+				void handleStreamEnd() {
+					events.push_back(Event(StreamEnd));
+				}
+
+				std::vector<Event> events;
+		} client_;
+		PayloadParserFactoryCollection factories_;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(XMPPParserTest);