1 files changed, 132 insertions, 0 deletions
diff --git a/Swiften/Queries/UnitTest/ResponderTest.cpp b/Swiften/Queries/UnitTest/ResponderTest.cpp
new file mode 100644
index 0000000..5c758e4
--- /dev/null
+++ b/Swiften/Queries/UnitTest/ResponderTest.cpp
@@ -0,0 +1,132 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <boost/shared_ptr.hpp>
+#include <boost/bind.hpp>
+
+#include "Swiften/Queries/Responder.h"
+#include "Swiften/Queries/IQRouter.h"
+#include "Swiften/Queries/DummyIQChannel.h"
+#include "Swiften/Elements/SoftwareVersion.h"
+
+using namespace Swift;
+
+class ResponderTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(ResponderTest);
+		CPPUNIT_TEST(testConstructor);
+		CPPUNIT_TEST(testHandleIQ_Set);
+		CPPUNIT_TEST(testHandleIQ_Get);
+		CPPUNIT_TEST(testHandleIQ_Error);
+		CPPUNIT_TEST(testHandleIQ_Result);
+		CPPUNIT_TEST(testHandleIQ_NoPayload);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		ResponderTest() {}
+
+		void setUp() {
+			channel_ = new DummyIQChannel();
+			router_ = new IQRouter(channel_);
+			payload_ = boost::shared_ptr<SoftwareVersion>(new SoftwareVersion("foo"));
+		}
+
+		void tearDown() {
+			delete router_;
+			delete channel_;
+		}
+
+		void testConstructor() {
+			MyResponder testling(router_);
+
+			channel_->onIQReceived(createRequest(IQ::Set));
+
+			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(testling.setPayloads_.size()));
+		}
+
+		void testHandleIQ_Set() {
+			MyResponder testling(router_);
+
+			CPPUNIT_ASSERT(dynamic_cast<IQHandler*>(&testling)->handleIQ(createRequest(IQ::Set)));
+
+			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(testling.setPayloads_.size()));
+			CPPUNIT_ASSERT(payload_ == testling.setPayloads_[0]);
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.getPayloads_.size()));
+		}
+
+		void testHandleIQ_Get() {
+			MyResponder testling(router_);
+
+			CPPUNIT_ASSERT(dynamic_cast<IQHandler*>(&testling)->handleIQ(createRequest(IQ::Get)));
+
+			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(testling.getPayloads_.size()));
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.setPayloads_.size()));
+			CPPUNIT_ASSERT(payload_ == testling.getPayloads_[0]);
+		}
+
+		void testHandleIQ_Error() {
+			MyResponder testling(router_);
+
+			CPPUNIT_ASSERT(!dynamic_cast<IQHandler*>(&testling)->handleIQ(createRequest(IQ::Error)));
+
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.getPayloads_.size()));
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.setPayloads_.size()));
+		}
+
+		void testHandleIQ_Result() {
+			MyResponder testling(router_);
+
+			CPPUNIT_ASSERT(!dynamic_cast<IQHandler*>(&testling)->handleIQ(createRequest(IQ::Result)));
+
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.getPayloads_.size()));
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.setPayloads_.size()));
+		}
+
+		void testHandleIQ_NoPayload() {
+			MyResponder testling(router_);
+
+			CPPUNIT_ASSERT(!dynamic_cast<IQHandler*>(&testling)->handleIQ(boost::shared_ptr<IQ>(new IQ(IQ::Get))));
+
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.getPayloads_.size()));
+			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(testling.setPayloads_.size()));
+		}
+
+	private:
+		boost::shared_ptr<IQ> createRequest(IQ::Type type) {
+			boost::shared_ptr<IQ> iq(new IQ(type));
+			iq->addPayload(payload_);
+			iq->setID("myid");
+			iq->setFrom(JID("foo@bar.com/baz"));
+			return iq;
+		}
+
+	private:
+		class MyResponder : public Responder<SoftwareVersion> {
+			public:
+				MyResponder(IQRouter* router) : Responder<SoftwareVersion>(router), getRequestResponse_(true), setRequestResponse_(true) {}
+
+				virtual bool handleGetRequest(const JID& from, const String& id, boost::shared_ptr<SoftwareVersion> payload) {
+					CPPUNIT_ASSERT_EQUAL(JID("foo@bar.com/baz"), from);
+					CPPUNIT_ASSERT_EQUAL(String("myid"), id);
+					getPayloads_.push_back(payload);
+					return getRequestResponse_;
+				}
+				virtual bool handleSetRequest(const JID& from, const String& id, boost::shared_ptr<SoftwareVersion> payload) {
+					CPPUNIT_ASSERT_EQUAL(JID("foo@bar.com/baz"), from);
+					CPPUNIT_ASSERT_EQUAL(String("myid"), id);
+					setPayloads_.push_back(payload);
+					return setRequestResponse_;
+				}
+
+				bool getRequestResponse_;
+				bool setRequestResponse_;
+				std::vector<boost::shared_ptr<SoftwareVersion> > getPayloads_;
+				std::vector<boost::shared_ptr<SoftwareVersion> > setPayloads_;
+		};
+
+	private:
+		IQRouter* router_;
+		DummyIQChannel* channel_;
+		boost::shared_ptr<SoftwareVersion> payload_;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(ResponderTest);