1 files changed, 97 insertions, 0 deletions
diff --git a/Swiften/Parser/UnitTest/PayloadParserFactoryCollectionTest.cpp b/Swiften/Parser/UnitTest/PayloadParserFactoryCollectionTest.cpp
new file mode 100644
index 0000000..bfd2c2c
--- /dev/null
+++ b/Swiften/Parser/UnitTest/PayloadParserFactoryCollectionTest.cpp
@@ -0,0 +1,97 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+
+#include "Swiften/Parser/PayloadParserFactoryCollection.h"
+#include "Swiften/Parser/PayloadParserFactory.h"
+
+using namespace Swift;
+
+class PayloadParserFactoryCollectionTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(PayloadParserFactoryCollectionTest);
+		CPPUNIT_TEST(testGetPayloadParserFactory);
+		CPPUNIT_TEST(testGetPayloadParserFactory_NoMatchingFactory);
+		CPPUNIT_TEST(testGetPayloadParserFactory_TwoMatchingFactories);
+		CPPUNIT_TEST(testGetPayloadParserFactory_MatchWithDefaultFactory);
+		CPPUNIT_TEST(testGetPayloadParserFactory_NoMatchWithDefaultFactory);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		PayloadParserFactoryCollectionTest() {}
+
+		void setUp() {
+		}
+
+		void tearDown() {
+		}
+
+		void testGetPayloadParserFactory() {
+			PayloadParserFactoryCollection testling;
+			DummyFactory factory1("foo");
+			testling.addFactory(&factory1);
+			DummyFactory factory2("bar");
+			testling.addFactory(&factory2);
+			DummyFactory factory3("baz");
+			testling.addFactory(&factory3);
+
+			PayloadParserFactory* factory = testling.getPayloadParserFactory("bar", "", AttributeMap());
+
+			CPPUNIT_ASSERT(factory == &factory2);
+		}
+
+		void testGetPayloadParserFactory_NoMatchingFactory() {
+			PayloadParserFactoryCollection testling;
+			DummyFactory factory("foo");
+
+			CPPUNIT_ASSERT(!testling.getPayloadParserFactory("bar", "", AttributeMap()));
+		}
+
+		void testGetPayloadParserFactory_TwoMatchingFactories() {
+			PayloadParserFactoryCollection testling;
+			DummyFactory factory1("foo");
+			testling.addFactory(&factory1);
+			DummyFactory factory2("foo");
+			testling.addFactory(&factory2);
+
+			PayloadParserFactory* factory = testling.getPayloadParserFactory("foo", "", AttributeMap());
+
+			CPPUNIT_ASSERT(factory == &factory2);
+		}
+
+		void testGetPayloadParserFactory_MatchWithDefaultFactory() {
+			PayloadParserFactoryCollection testling;
+			DummyFactory factory1("foo");
+			testling.addFactory(&factory1);
+			DummyFactory factory2;
+			testling.setDefaultFactory(&factory2);
+
+			PayloadParserFactory* factory = testling.getPayloadParserFactory("foo", "", AttributeMap());
+
+			CPPUNIT_ASSERT(factory == &factory1);
+		}
+
+		void testGetPayloadParserFactory_NoMatchWithDefaultFactory() {
+			PayloadParserFactoryCollection testling;
+			DummyFactory factory1("foo");
+			testling.addFactory(&factory1);
+			DummyFactory factory2;
+			testling.setDefaultFactory(&factory2);
+
+			PayloadParserFactory* factory = testling.getPayloadParserFactory("baz", "", AttributeMap());
+
+			CPPUNIT_ASSERT(factory == &factory2);
+		}
+	
+	
+	private:
+		struct DummyFactory : public PayloadParserFactory {
+			DummyFactory(const String& element = "") : element(element) {}
+			virtual bool canParse(const String& e, const String&, const AttributeMap&) const {
+				return element.isEmpty() ? true : element == e;
+			}
+			virtual PayloadParser* createPayloadParser() { return NULL; }
+			String element;
+		};
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(PayloadParserFactoryCollectionTest);