2 files changed, 147 insertions, 1 deletions
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..d50e2c0
--- /dev/null
+++ b/Slimber/UnitTest/MenuletControllerTest.cpp
@@ -0,0 +1,145 @@
+#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(7, 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("*Exit*"), menulet->items[i++]);
+		}
+
+		void testUpdate() {
+			MenuletController testling(menulet);
+
+			testling.setXMPPStatus("You are connected", MenuletController::Online);
+
+			CPPUNIT_ASSERT_EQUAL(7, 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("*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 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);