1 files changed, 152 insertions, 0 deletions
diff --git a/Swiften/Avatars/UnitTest/VCardAvatarManagerTest.cpp b/Swiften/Avatars/UnitTest/VCardAvatarManagerTest.cpp
new file mode 100644
index 0000000..8a0658c
--- /dev/null
+++ b/Swiften/Avatars/UnitTest/VCardAvatarManagerTest.cpp
@@ -0,0 +1,152 @@
+/*
+ * 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 <boost/bind.hpp>
+
+#include "Swiften/Elements/VCard.h"
+#include "Swiften/Avatars/VCardAvatarManager.h"
+#include "Swiften/Avatars/AvatarMemoryStorage.h"
+#include "Swiften/VCards/VCardMemoryStorage.h"
+#include "Swiften/VCards/VCardManager.h"
+#include "Swiften/MUC/MUCRegistry.h"
+#include "Swiften/Queries/IQRouter.h"
+#include "Swiften/Client/DummyStanzaChannel.h"
+#include "Swiften/StringCodecs/SHA1.h"
+#include "Swiften/StringCodecs/Hexify.h"
+
+using namespace Swift;
+
+class VCardAvatarManagerTest : public CppUnit::TestFixture {
+		CPPUNIT_TEST_SUITE(VCardAvatarManagerTest);
+		CPPUNIT_TEST(testGetAvatarHashKnownAvatar);
+		CPPUNIT_TEST(testGetAvatarHashEmptyAvatar);
+		CPPUNIT_TEST(testGetAvatarHashUnknownAvatarKnownVCardStoresAvatar);
+		CPPUNIT_TEST(testGetAvatarHashUnknownAvatarUnknownVCard);
+		CPPUNIT_TEST(testVCardUpdateTriggersUpdate);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		void setUp() {
+			ownJID = JID("foo@fum.com/bum");
+			stanzaChannel = new DummyStanzaChannel();
+			stanzaChannel->setAvailable(true);
+			iqRouter = new IQRouter(stanzaChannel);
+			mucRegistry = new DummyMUCRegistry();
+			avatarStorage = new AvatarMemoryStorage();
+			vcardStorage = new VCardMemoryStorage();
+			vcardManager = new VCardManager(ownJID, iqRouter, vcardStorage);
+			avatar1 = ByteArray("abcdefg");
+			avatar1Hash = Hexify::hexify(SHA1::getHash(avatar1));
+			user1 = JID("user1@bar.com/bla");
+			user2 = JID("user2@foo.com/baz");
+		}
+
+		void tearDown() {
+			delete vcardManager;
+			delete vcardStorage;
+			delete avatarStorage;
+			delete mucRegistry;
+			delete iqRouter;
+			delete stanzaChannel;
+		}
+
+		void testGetAvatarHashKnownAvatar() {
+			std::auto_ptr<VCardAvatarManager> testling = createManager();
+			storeVCardWithPhoto(user1.toBare(), avatar1);
+			avatarStorage->addAvatar(avatar1Hash, avatar1);
+
+			String result = testling->getAvatarHash(user1);
+
+			CPPUNIT_ASSERT_EQUAL(avatar1Hash, result);
+		}
+
+		void testGetAvatarHashEmptyAvatar() {
+			std::auto_ptr<VCardAvatarManager> testling = createManager();
+			storeEmptyVCard(user1.toBare());
+
+			String result = testling->getAvatarHash(user1);
+
+			CPPUNIT_ASSERT_EQUAL(String(), result);
+		}
+
+		void testGetAvatarHashUnknownAvatarKnownVCardStoresAvatar() {
+			std::auto_ptr<VCardAvatarManager> testling = createManager();
+			storeVCardWithPhoto(user1.toBare(), avatar1);
+
+			String result = testling->getAvatarHash(user1);
+
+			CPPUNIT_ASSERT_EQUAL(avatar1Hash, result);
+			CPPUNIT_ASSERT(avatarStorage->hasAvatar(avatar1Hash));
+			CPPUNIT_ASSERT_EQUAL(avatar1, avatarStorage->getAvatar(avatar1Hash));
+		}
+
+		void testGetAvatarHashUnknownAvatarUnknownVCard() {
+			std::auto_ptr<VCardAvatarManager> testling = createManager();
+
+			String result = testling->getAvatarHash(user1);
+
+			CPPUNIT_ASSERT_EQUAL(String(), result);
+		}
+
+		void testVCardUpdateTriggersUpdate() {
+			std::auto_ptr<VCardAvatarManager> testling = createManager();
+			vcardManager->requestVCard(user1);
+			sendVCardResult();
+
+			CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(changes.size()));
+		}
+
+	private:
+		std::auto_ptr<VCardAvatarManager> createManager() {
+			std::auto_ptr<VCardAvatarManager> result(new VCardAvatarManager(vcardManager, avatarStorage, mucRegistry));
+			result->onAvatarChanged.connect(boost::bind(&VCardAvatarManagerTest::handleAvatarChanged, this, _1));
+			return result;
+		}
+
+		void storeVCardWithPhoto(const JID& jid, const ByteArray& avatar) {
+			VCard::ref vcard(new VCard());
+			vcard->setPhoto(avatar);
+			vcardStorage->setVCard(jid, vcard);
+		}
+
+		void storeEmptyVCard(const JID& jid) {
+			VCard::ref vcard(new VCard());
+			vcardStorage->setVCard(jid, vcard);
+		}
+
+		void handleAvatarChanged(const JID& jid) {
+			changes.push_back(jid);
+		}
+
+		void sendVCardResult() {
+			VCard::ref vcard(new VCard());
+			vcard->setFullName("Foo Bar");
+			stanzaChannel->onIQReceived(IQ::createResult(JID("baz@fum.com/dum"), stanzaChannel->sentStanzas[0]->getID(), vcard));
+		}
+
+	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_;
+		};
+
+		JID ownJID;
+		DummyStanzaChannel* stanzaChannel;
+		IQRouter* iqRouter;
+		DummyMUCRegistry* mucRegistry;
+		AvatarMemoryStorage* avatarStorage;
+		VCardManager* vcardManager;
+		VCardMemoryStorage* vcardStorage;
+		ByteArray avatar1;
+		String avatar1Hash;
+		std::vector<JID> changes;
+		JID user1;
+		JID user2;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(VCardAvatarManagerTest);