6 files changed, 132 insertions, 0 deletions
diff --git a/Swiften/Roster/UnitTest/Makefile.inc b/Swiften/Roster/UnitTest/Makefile.inc
new file mode 100644
index 0000000..5631641
--- /dev/null
+++ b/Swiften/Roster/UnitTest/Makefile.inc
@@ -0,0 +1,4 @@
+UNITTEST_SOURCES += \
+	Swiften/Roster/UnitTest/RosterTest.cpp \
+	Swiften/Roster/UnitTest/OfflineRosterFilterTest.cpp
+
diff --git a/Swiften/Roster/UnitTest/MockTreeWidget.h b/Swiften/Roster/UnitTest/MockTreeWidget.h
new file mode 100644
index 0000000..e6f6def
--- /dev/null
+++ b/Swiften/Roster/UnitTest/MockTreeWidget.h
@@ -0,0 +1,14 @@
+#ifndef SWIFTEN_MockTreeWidget_H
+#define SWIFTEN_MockTreeWidget_H
+
+#include "Swiften/Roster/TreeWidget.h"
+
+namespace Swift {
+
+class MockTreeWidget : public TreeWidget {
+	public:
+		virtual ~MockTreeWidget() {}
+};
+
+}
+#endif
diff --git a/Swiften/Roster/UnitTest/MockTreeWidgetFactory.h b/Swiften/Roster/UnitTest/MockTreeWidgetFactory.h
new file mode 100644
index 0000000..09e4742
--- /dev/null
+++ b/Swiften/Roster/UnitTest/MockTreeWidgetFactory.h
@@ -0,0 +1,31 @@
+#ifndef SWIFTEN_MockTreeWidgetFactory_H
+#define SWIFTEN_MockTreeWidgetFactory_H
+
+#include "Swiften/Roster/TreeWidgetFactory.h"
+#include "Swiften/Roster/UnitTest/MockTreeWidget.h"
+#include "Swiften/Roster/UnitTest/MockTreeWidgetItem.h"
+
+namespace Swift {
+
+class MockTreeWidgetItem;
+class MockTreeWidget;
+
+class MockTreeWidgetFactory : public TreeWidgetFactory {
+	public:
+		virtual ~MockTreeWidgetFactory() {}
+		virtual TreeWidget* createTreeWidget() {
+			return new MockTreeWidget();
+		};
+		virtual TreeWidgetItem* createTreeWidgetItem(TreeWidgetItem*) {
+			return new MockTreeWidgetItem();
+		};
+		virtual TreeWidgetItem* createTreeWidgetItem(TreeWidget*) {
+			return new MockTreeWidgetItem();
+		}
+};
+
+}
+
+#endif
+
+
diff --git a/Swiften/Roster/UnitTest/MockTreeWidgetItem.h b/Swiften/Roster/UnitTest/MockTreeWidgetItem.h
new file mode 100644
index 0000000..f352936
--- /dev/null
+++ b/Swiften/Roster/UnitTest/MockTreeWidgetItem.h
@@ -0,0 +1,26 @@
+#ifndef SWIFTEN_MockTreeWidgetItem_H
+#define SWIFTEN_MockTreeWidgetItem_H
+
+#include "Swiften/Base/String.h"
+#include "Swiften/Roster/TreeWidgetItem.h"
+
+#include <boost/signal.hpp>
+#include <boost/shared_ptr.hpp>
+
+namespace Swift {
+
+class MockTreeWidgetItem : public TreeWidgetItem {
+	public:
+		virtual ~MockTreeWidgetItem() {};
+		virtual void setText(const String&) {};
+		virtual void setExpanded(bool) {};
+		virtual void setTextColor(unsigned long) {};
+		virtual void setBackgroundColor(unsigned long) {};
+		virtual void show() {};
+		virtual void hide() {};
+};
+
+}
+#endif
+
+
diff --git a/Swiften/Roster/UnitTest/OfflineRosterFilterTest.cpp b/Swiften/Roster/UnitTest/OfflineRosterFilterTest.cpp
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/Swiften/Roster/UnitTest/OfflineRosterFilterTest.cpp
diff --git a/Swiften/Roster/UnitTest/RosterTest.cpp b/Swiften/Roster/UnitTest/RosterTest.cpp
new file mode 100644
index 0000000..b43a41c
--- /dev/null
+++ b/Swiften/Roster/UnitTest/RosterTest.cpp
@@ -0,0 +1,57 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <boost/shared_ptr.hpp>
+
+#include "Swiften/Roster/Roster.h"
+#include "Swiften/Roster/UnitTest/MockTreeWidget.h"
+#include "Swiften/Roster/UnitTest/MockTreeWidgetFactory.h"
+#include "Swiften/Roster/UnitTest/MockTreeWidgetItem.h"
+
+using namespace Swift;
+
+class RosterTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(RosterTest);
+		CPPUNIT_TEST(testGetGroup);
+		CPPUNIT_TEST_SUITE_END();
+
+	private:
+		Roster *roster_;
+		TreeWidget *widget_;
+		TreeWidgetFactory *factory_;
+		JID jid1_;
+		JID jid2_; 
+		JID jid3_;
+
+	public:
+
+		RosterTest() : jid1_(JID("a@b.c")), jid2_(JID("b@c.d")), jid3_(JID("c@d.e")) {}
+
+		void setUp() {
+			factory_ = new MockTreeWidgetFactory();
+			widget_ = factory_->createTreeWidget();
+			roster_ = new Roster(widget_, factory_);
+		}
+
+		void tearDown() {
+			delete roster_;
+			//delete widget_;
+			delete factory_;
+		}
+
+		void testGetGroup() {
+			roster_->addContact(jid1_, "Bert", "group1");
+			roster_->addContact(jid2_, "Ernie", "group2");
+			roster_->addContact(jid3_, "Cookie", "group1");
+
+			CPPUNIT_ASSERT_EQUAL(roster_->getGroup("group1"), roster_->getGroup("group1"));
+			CPPUNIT_ASSERT_EQUAL(roster_->getGroup("group2"), roster_->getGroup("group2"));
+			CPPUNIT_ASSERT_EQUAL(roster_->getGroup("group3"), roster_->getGroup("group3"));
+			CPPUNIT_ASSERT(roster_->getGroup("group1") != roster_->getGroup("group2"));
+			CPPUNIT_ASSERT(roster_->getGroup("group2") != roster_->getGroup("group3"));
+			CPPUNIT_ASSERT(roster_->getGroup("group3") != roster_->getGroup("group1"));
+		}
+
+};
+CPPUNIT_TEST_SUITE_REGISTRATION(RosterTest);
+