1 files changed, 120 insertions, 0 deletions
diff --git a/Swift/Controllers/Chat/UnitTest/MUCControllerTest.cpp b/Swift/Controllers/Chat/UnitTest/MUCControllerTest.cpp
index fbc6901..aeae562 100644
--- a/Swift/Controllers/Chat/UnitTest/MUCControllerTest.cpp
+++ b/Swift/Controllers/Chat/UnitTest/MUCControllerTest.cpp
@@ -6,8 +6,23 @@
 
 #include <cppunit/extensions/HelperMacros.h>
 #include <cppunit/extensions/TestFactoryRegistry.h>
+#include "3rdParty/hippomocks.h"
 
+#include "Swift/Controllers/XMPPEvents/EventController.h"
+#include "Swiften/Presence/PresenceSender.h"
+#include "Swiften/Avatars/NullAvatarManager.h"
 #include "Swift/Controllers/Chat/MUCController.h"
+#include "Swift/Controllers/UIInterfaces/ChatWindow.h"
+#include "Swift/Controllers/UIInterfaces/ChatWindowFactory.h"
+#include "Swift/Controllers/NickResolver.h"
+#include "Swiften/Roster/XMPPRoster.h"
+#include "Swift/Controllers/UIEvents/UIEventStream.h"
+#include "Swift/Controllers/UnitTest/MockChatWindow.h"
+#include "Swiften/Client/DummyStanzaChannel.h"
+#include "Swiften/Queries/DummyIQChannel.h"
+#include "Swiften/Presence/PresenceOracle.h"
+#include "Swiften/Network/TimerFactory.h"
+#include "Swiften/Elements/MUCUserPayload.h"
 
 using namespace Swift;
 
@@ -17,15 +32,102 @@ class MUCControllerTest : public CppUnit::TestFixture
 	CPPUNIT_TEST(testJoinPartStringContructionSimple);
 	CPPUNIT_TEST(testJoinPartStringContructionMixed);
 	CPPUNIT_TEST(testAppendToJoinParts);
+	CPPUNIT_TEST(testAddressedToSelf);
+	CPPUNIT_TEST(testNotAddressedToSelf);
+	CPPUNIT_TEST(testAddressedToSelfBySelf);
 	CPPUNIT_TEST_SUITE_END();
 
 public:
 	MUCControllerTest() {};
 
 	void setUp() {
+		self_ = JID("girl@wonderland.lit/rabbithole");
+		muc_ = JID("teaparty@rooms.wonderland.lit");
+		nick_ = "aLiCe";
+		mocks_ = new MockRepository();
+		stanzaChannel_ = new DummyStanzaChannel();
+		iqChannel_ = new DummyIQChannel();
+		iqRouter_ = new IQRouter(iqChannel_);
+		eventController_ = new EventController();
+		chatWindowFactory_ = mocks_->InterfaceMock<ChatWindowFactory>();
+		presenceOracle_ = new PresenceOracle(stanzaChannel_);
+		presenceSender_ = new PresenceSender(stanzaChannel_);
+		uiEventStream_ = new UIEventStream();
+		avatarManager_ = new NullAvatarManager();
+		TimerFactory* timerFactory = NULL;
+		window_ = new MockChatWindow();//mocks_->InterfaceMock<ChatWindow>();
+		mocks_->ExpectCall(chatWindowFactory_, ChatWindowFactory::createChatWindow).With(muc_, uiEventStream_).Return(window_);
+		controller_ = new MUCController (self_, muc_, nick_, stanzaChannel_, presenceSender_,
+				iqRouter_, chatWindowFactory_, presenceOracle_, avatarManager_, uiEventStream_, false, timerFactory, eventController_);
 	};
 
 	void tearDown() {
+		delete controller_;
+		delete eventController_;
+		delete presenceOracle_;
+		delete mocks_;
+		delete uiEventStream_;
+		delete stanzaChannel_;
+		delete presenceSender_;
+		delete iqRouter_;
+		delete iqChannel_;
+		delete avatarManager_;
+	}
+
+	void finishJoin() {
+		Presence::ref presence(new Presence());
+		presence->setFrom(JID(muc_.toString() + "/" + nick_));
+		MUCUserPayload::ref status(new MUCUserPayload());
+		MUCUserPayload::StatusCode code;
+		code.code = 110;
+		status->addStatusCode(code);
+		presence->addPayload(status);
+		stanzaChannel_->onPresenceReceived(presence);
+	}
+
+	void testAddressedToSelf() {
+		finishJoin();
+		Message::ref message(new Message());
+
+		message = Message::ref(new Message());
+		message->setFrom(JID(muc_.toString() + "/otherperson"));
+		message->setBody(nick_ + ": hi there");
+		message->setType(Message::Groupchat);
+		controller_->handleIncomingMessage(MessageEvent::ref(new MessageEvent(message)));
+		CPPUNIT_ASSERT_EQUAL((size_t)1, eventController_->getEvents().size());
+
+		message->setFrom(JID(muc_.toString() + "/other"));
+		message->setBody("Hi there " + nick_);
+		message->setType(Message::Groupchat);
+		controller_->handleIncomingMessage(MessageEvent::ref(new MessageEvent(message)));
+		CPPUNIT_ASSERT_EQUAL((size_t)2, eventController_->getEvents().size());
+
+		message = Message::ref(new Message());
+		message->setFrom(JID(muc_.toString() + "/other2"));
+		message->setBody("Hi " + nick_.getLowerCase());
+		message->setType(Message::Groupchat);
+		controller_->handleIncomingMessage(MessageEvent::ref(new MessageEvent(message)));
+		CPPUNIT_ASSERT_EQUAL((size_t)3, eventController_->getEvents().size());
+	}
+
+	void testNotAddressedToSelf() {
+		finishJoin();
+		Message::ref message(new Message());
+		message->setFrom(JID(muc_.toString() + "/other3"));
+		message->setBody("Hi there Hatter");
+		message->setType(Message::Groupchat);
+		controller_->handleIncomingMessage(MessageEvent::ref(new MessageEvent(message)));
+		CPPUNIT_ASSERT_EQUAL((size_t)0, eventController_->getEvents().size());
+	}
+
+	void testAddressedToSelfBySelf() {
+		finishJoin();
+		Message::ref message(new Message());
+		message->setFrom(JID(muc_.toString() + "/" + nick_));
+		message->setBody("Hi there " + nick_);
+		message->setType(Message::Groupchat);
+		controller_->handleIncomingMessage(MessageEvent::ref(new MessageEvent(message)));
+		CPPUNIT_ASSERT_EQUAL((size_t)0, eventController_->getEvents().size());
 	}
 
 	void checkEqual(const std::vector<NickJoinPart>& expected, const std::vector<NickJoinPart>& actual) {
@@ -95,6 +197,24 @@ public:
 		list.push_back(NickJoinPart("Ernie", JoinThenPart));
 		CPPUNIT_ASSERT_EQUAL(String("Kev joined then left the room, Remko has left the room, Bert left then rejoined the room and Ernie joined then left the room."), MUCController::generateJoinPartString(list));
 	}
+
+private:
+	JID self_;
+	JID muc_;
+	String nick_;
+	StanzaChannel* stanzaChannel_;
+	IQChannel* iqChannel_;
+	IQRouter* iqRouter_;
+	EventController* eventController_;
+	ChatWindowFactory* chatWindowFactory_;
+	MUCController* controller_;
+//	NickResolver* nickResolver_;
+	PresenceOracle* presenceOracle_;
+	AvatarManager* avatarManager_;
+	PresenceSender* presenceSender_;
+	MockRepository* mocks_;
+	UIEventStream* uiEventStream_;
+	MockChatWindow* window_;
 };
 
 CPPUNIT_TEST_SUITE_REGISTRATION(MUCControllerTest);