3 files changed, 176 insertions, 1 deletions
diff --git a/Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp b/Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp
index f15daf4..f77a8cb 100644
--- a/Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp
+++ b/Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp
@@ -29,6 +29,8 @@ class LinkLocalPresenceManagerTest : public CppUnit::TestFixture {
 		CPPUNIT_TEST(testGetRoster_InfoWithLastName);
 		CPPUNIT_TEST(testGetRoster_InfoWithFirstAndLastName);
 		CPPUNIT_TEST(testGetRoster_NoInfo);
+		CPPUNIT_TEST(testGetServiceForJID);
+		CPPUNIT_TEST(testGetServiceForJID_NoMatch);
 		CPPUNIT_TEST_SUITE_END();
 
 	public:
@@ -176,6 +178,27 @@ class LinkLocalPresenceManagerTest : public CppUnit::TestFixture {
 			CPPUNIT_ASSERT_EQUAL(String(""), item->getName());
 		}
 
+		void testGetServiceForJID() {
+			std::auto_ptr<LinkLocalPresenceManager> testling(createTestling());
+
+			addService("alice@wonderland");
+			addService("rabbit@teaparty");
+			addService("queen@garden");
+
+			boost::optional<LinkLocalService> service = testling->getServiceForJID(JID("rabbit@teaparty"));
+			CPPUNIT_ASSERT(service);
+			CPPUNIT_ASSERT_EQUAL(String("rabbit@teaparty"), service->getID().getName());
+		}
+
+		void testGetServiceForJID_NoMatch() {
+			std::auto_ptr<LinkLocalPresenceManager> testling(createTestling());
+
+			addService("alice@wonderland");
+			addService("queen@garden");
+
+			CPPUNIT_ASSERT(!testling->getServiceForJID(JID("rabbit@teaparty")));
+		}
+
 	private:
 		std::auto_ptr<LinkLocalPresenceManager> createTestling() {
 			std::auto_ptr<LinkLocalPresenceManager> testling(
diff --git a/Slimber/UnitTest/Makefile.inc b/Slimber/UnitTest/Makefile.inc
index f2c4db8..5bc0c8a 100644
--- a/Slimber/UnitTest/Makefile.inc
+++ b/Slimber/UnitTest/Makefile.inc
@@ -1,2 +1,3 @@
 UNITTEST_SOURCES += \
-	Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp
+	Slimber/UnitTest/LinkLocalPresenceManagerTest.cpp \
+	Slimber/UnitTest/MenuletControllerTest.cpp
diff --git a/Slimber/UnitTest/MenuletControllerTest.cpp b/Slimber/UnitTest/MenuletControllerTest.cpp
new file mode 100644
index 0000000..c666679
--- /dev/null
+++ b/Slimber/UnitTest/MenuletControllerTest.cpp
@@ -0,0 +1,151 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+
+#include "Slimber/Menulet.h"
+#include "Slimber/MenuletController.h"
+
+using namespace Swift;
+
+class MenuletControllerTest : public CppUnit::TestFixture {
+		CPPUNIT_TEST_SUITE(MenuletControllerTest);
+		CPPUNIT_TEST(testConstructor);
+		CPPUNIT_TEST(testUpdate);
+		CPPUNIT_TEST(testSetXMPPStatus_Online);
+		CPPUNIT_TEST(testSetXMPPStatus_Offline);
+		CPPUNIT_TEST(testSetUserNames);
+		CPPUNIT_TEST(testSetUserNames_NoUsers);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		void setUp() {
+			menulet = new FakeMenulet();
+		}
+
+		void tearDown() {
+			delete menulet;
+		}
+
+		void testConstructor() {
+			MenuletController testling(menulet);
+
+			CPPUNIT_ASSERT_EQUAL(8, static_cast<int>(menulet->items.size()));
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("No online users"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("[Offline] "), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*About*"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*Restart*"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*Exit*"), menulet->items[i++]);
+		}
+
+		void testUpdate() {
+			MenuletController testling(menulet);
+
+			testling.setXMPPStatus("You are connected", MenuletController::Online);
+
+			CPPUNIT_ASSERT_EQUAL(8, static_cast<int>(menulet->items.size()));
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("No online users"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("[Online] You are connected"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*About*"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*Restart*"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("*Exit*"), menulet->items[i++]);
+		}
+
+		void testSetXMPPStatus_Online() {
+			MenuletController testling(menulet);
+
+			testling.setXMPPStatus("You are connected", MenuletController::Online);
+
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("No online users"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("[Online] You are connected"), menulet->items[i++]);
+		}
+
+
+		void testSetXMPPStatus_Offline() {
+			MenuletController testling(menulet);
+
+			testling.setXMPPStatus("You are not connected", MenuletController::Offline);
+
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("No online users"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("[Offline] You are not connected"), menulet->items[i++]);
+		}
+
+		void testSetUserNames() {
+			MenuletController testling(menulet);
+
+			std::vector<String> users;
+			users.push_back("Alice In Wonderland");
+			users.push_back("The Mad Hatter");
+			testling.setUserNames(users);
+
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("Online users:"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("  Alice In Wonderland"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("  The Mad Hatter"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+		}
+
+		void testSetUserNames_NoUsers() {
+			MenuletController testling(menulet);
+
+			std::vector<String> users;
+			testling.setUserNames(users);
+
+			int i = 0;
+			CPPUNIT_ASSERT_EQUAL(String("No online users"), menulet->items[i++]);
+			CPPUNIT_ASSERT_EQUAL(String("-"), menulet->items[i++]);
+		}
+	
+	private:
+		struct FakeMenulet : public Menulet {
+			virtual void clear() {
+				items.clear();
+			}
+
+			virtual void addItem(const String& name, const String& icon = String()) {
+				String result;
+				if (!icon.isEmpty()) {
+					result += "[" + icon + "] ";
+				}
+				result += name;
+				items.push_back(result);
+			}
+
+			virtual void addAboutItem() {
+				items.push_back("*About*");
+			}
+
+			virtual void addRestartItem() {
+				items.push_back("*Restart*");
+			}
+
+			virtual void addExitItem() {
+				items.push_back("*Exit*");
+			}
+
+			virtual void addSeparator() {
+				items.push_back("-");
+			}
+
+			virtual void setIcon(const String& i) {
+				icon = i;
+			}
+
+			std::vector<String> items;
+			String icon;
+		};
+
+		FakeMenulet* menulet;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(MenuletControllerTest);