/*
 * Copyright (c) 2010 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <boost/shared_ptr.hpp>
#include <boost/bind.hpp>

#include <Swiften/Queries/Requests/GetPrivateStorageRequest.h>
#include <Swiften/Queries/IQRouter.h>
#include <Swiften/Queries/DummyIQChannel.h>
#include <Swiften/Elements/Payload.h>

using namespace Swift;

class GetPrivateStorageRequestTest : public CppUnit::TestFixture {
		CPPUNIT_TEST_SUITE(GetPrivateStorageRequestTest);
		CPPUNIT_TEST(testSend);
		CPPUNIT_TEST(testHandleResponse);
		CPPUNIT_TEST(testHandleResponse_Error);
		CPPUNIT_TEST_SUITE_END();

	public:
		class MyPayload : public Payload {
			public:
				MyPayload(const std::string& text = "") : text(text) {}
				std::string text;
		};

	public:
		void setUp() {
			channel = new DummyIQChannel();
			router = new IQRouter(channel);
		}

		void tearDown() {
			delete router;
			delete channel;
		}

		void testSend() {
			GetPrivateStorageRequest<MyPayload>::ref request = GetPrivateStorageRequest<MyPayload>::create(router);
			request->send();

			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(channel->iqs_.size()));
			CPPUNIT_ASSERT_EQUAL(JID(), channel->iqs_[0]->getTo());
			CPPUNIT_ASSERT_EQUAL(IQ::Get, channel->iqs_[0]->getType());
			boost::shared_ptr<PrivateStorage> storage = channel->iqs_[0]->getPayload<PrivateStorage>();
			CPPUNIT_ASSERT(storage);
			boost::shared_ptr<MyPayload> payload = boost::dynamic_pointer_cast<MyPayload>(storage->getPayload());
			CPPUNIT_ASSERT(payload);
		}

		void testHandleResponse() {
			GetPrivateStorageRequest<MyPayload>::ref testling = GetPrivateStorageRequest<MyPayload>::create(router);
			testling->onResponse.connect(boost::bind(&GetPrivateStorageRequestTest::handleResponse, this, _1, _2));
			testling->send();
			channel->onIQReceived(createResponse("test-id", "foo"));

			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(responses.size()));
			CPPUNIT_ASSERT_EQUAL(std::string("foo"), boost::dynamic_pointer_cast<MyPayload>(responses[0])->text);
		}

		void testHandleResponse_Error() {
			GetPrivateStorageRequest<MyPayload>::ref testling = GetPrivateStorageRequest<MyPayload>::create(router);
			testling->onResponse.connect(boost::bind(&GetPrivateStorageRequestTest::handleResponse, this, _1, _2));
			testling->send();
			channel->onIQReceived(createError("test-id"));

			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(responses.size()));
			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(errors.size()));
		}

	private:
		void handleResponse(boost::shared_ptr<Payload> p, ErrorPayload::ref e) {
			if (e) {
				errors.push_back(*e);
			}
			else {
				responses.push_back(p);
			}
		}

		boost::shared_ptr<IQ> createResponse(const std::string& id, const std::string& text) {
			boost::shared_ptr<IQ> iq(new IQ(IQ::Result));
			boost::shared_ptr<PrivateStorage> storage(new PrivateStorage());
			storage->setPayload(boost::shared_ptr<Payload>(new MyPayload(text)));
			iq->addPayload(storage);
			iq->setID(id);
			return iq;
		}

		boost::shared_ptr<IQ> createError(const std::string& id) {
			boost::shared_ptr<IQ> iq(new IQ(IQ::Error));
			iq->setID(id);
			return iq;
		}
	
	private:
		IQRouter* router;
		DummyIQChannel* channel;
		std::vector< ErrorPayload > errors;
		std::vector< boost::shared_ptr<Payload> > responses;
};

CPPUNIT_TEST_SUITE_REGISTRATION(GetPrivateStorageRequestTest);