/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

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

#include "Swiften/Elements/VCardUpdate.h"
#include "Swiften/Avatars/AvatarManager.h"
#include "Swiften/Avatars/AvatarStorage.h"
#include "Swiften/MUC/MUCRegistry.h"
#include "Swiften/Queries/IQRouter.h"
#include "Swiften/Client/DummyStanzaChannel.h"

using namespace Swift;

class AvatarManagerTest : public CppUnit::TestFixture {
		CPPUNIT_TEST_SUITE(AvatarManagerTest);
		CPPUNIT_TEST(testUpdate_UpdateNewHash);
		/*&
		CPPUNIT_TEST(testUpdate_UpdateNewHashAlreadyHaveAvatar);
		CPPUNIT_TEST(testUpdate_UpdateNewHashFromMUC);
		CPPUNIT_TEST(testUpdate_UpdateSameHash);*/
		CPPUNIT_TEST(testUpdate_UpdateWithError);
		/*
		CPPUNIT_TEST(testUpdate_UpdateNewHashSameThanOtherUser);
		CPPUNIT_TEST(testReceiveVCard);
		CPPUNIT_TEST(testGetAvatarPath);
		CPPUNIT_TEST(testGetAvatarPathFromMUC);*/
		CPPUNIT_TEST_SUITE_END();

	public:
		void setUp() {
			stanzaChannel_ = new DummyStanzaChannel();
			iqRouter_ = new IQRouter(stanzaChannel_);
			mucRegistry_ = new DummyMUCRegistry();
			avatarStorage_ = new DummyAvatarStorage();
		}

		void tearDown() {
			delete avatarStorage_;
			delete mucRegistry_;
			delete iqRouter_;
			delete stanzaChannel_;
		}

		void testUpdate_UpdateNewHash() {
			std::auto_ptr<AvatarManager> testling = createManager();
			stanzaChannel_->onPresenceReceived(createPresenceWithPhotoHash());

			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(stanzaChannel_->sentStanzas.size()));
			// TODO
			//CPPUNIT_ASSERT(stanzaChannel_->isRequestAtIndex<VCardUpdate>(0, JID("foo@bar.com"), IQ::Get));
		}

/*
		void testUpdate_UpdateNewHashAlreadyHaveAvatar() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testUpdate_UpdateNewHashFromMUC() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testUpdate_UpdateSameHash() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testUpdate_UpdateNewHashSameThanOtherUser() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testReceiveVCard() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testGetAvatarPath() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}

		void testGetAvatarPathFromMUC() {
			std::auto_ptr<AvatarManager> testling = createManager();
		}
		*/

		void testUpdate_UpdateWithError() {
			std::auto_ptr<AvatarManager> testling = createManager();
			boost::shared_ptr<Presence> update = createPresenceWithPhotoHash();
			update->addPayload(boost::shared_ptr<ErrorPayload>(new ErrorPayload()));
			stanzaChannel_->onPresenceReceived(update);

			CPPUNIT_ASSERT_EQUAL(0, static_cast<int>(stanzaChannel_->sentStanzas.size()));
		}


	private:
		std::auto_ptr<AvatarManager> createManager() {
			return std::auto_ptr<AvatarManager>(new AvatarManager(stanzaChannel_, iqRouter_, avatarStorage_, mucRegistry_));
		}

		boost::shared_ptr<Presence> createPresenceWithPhotoHash() {
			boost::shared_ptr<Presence> presence(new Presence());
			presence->setFrom(JID("foo@bar.com/baz"));
			presence->addPayload(boost::shared_ptr<VCardUpdate>(new VCardUpdate("aef56135bcce35eb24a43fcd684005b4ca286497")));
			return presence;
		}

	private:
		struct DummyMUCRegistry : public MUCRegistry {
			bool isMUC(const JID& jid) const { return std::find(mucs_.begin(), mucs_.end(), jid) != mucs_.end(); }
			std::vector<JID> mucs_;
		};
		struct DummyAvatarStorage : public AvatarStorage {
			virtual bool hasAvatar(const String& hash) const { return avatars.find(hash) != avatars.end(); }
			virtual void addAvatar(const String& hash, const ByteArray& avatar) { avatars[hash] = avatar; }
			virtual boost::filesystem::path getAvatarPath(const String& hash) const {
				return boost::filesystem::path("/avatars") / hash.getUTF8String();
			}
			std::map<String, ByteArray> avatars;
		};
		DummyStanzaChannel* stanzaChannel_;
		IQRouter* iqRouter_;
		DummyMUCRegistry* mucRegistry_;
		DummyAvatarStorage* avatarStorage_;
};

CPPUNIT_TEST_SUITE_REGISTRATION(AvatarManagerTest);