2 files changed, 129 insertions, 0 deletions
diff --git a/Swiften/EventLoop/UnitTest/EventLoopTest.cpp b/Swiften/EventLoop/UnitTest/EventLoopTest.cpp
new file mode 100644
index 0000000..9475ac9
--- /dev/null
+++ b/Swiften/EventLoop/UnitTest/EventLoopTest.cpp
@@ -0,0 +1,67 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <boost/thread.hpp>
+#include <boost/bind.hpp>
+
+#include "Swiften/EventLoop/EventOwner.h"
+#include "Swiften/EventLoop/SimpleEventLoop.h"
+#include "Swiften/Base/sleep.h"
+
+using namespace Swift;
+
+class EventLoopTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(EventLoopTest);
+		CPPUNIT_TEST(testPost);
+		CPPUNIT_TEST(testRemove);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		EventLoopTest() {}
+
+		void setUp() {
+			events_.clear();
+		}
+
+		void testPost() {
+			SimpleEventLoop testling;
+
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 1));
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 2));
+			testling.stop();
+			testling.run();
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(events_.size()));
+			CPPUNIT_ASSERT_EQUAL(1, events_[0]);
+			CPPUNIT_ASSERT_EQUAL(2, events_[1]);
+		}
+
+		void testRemove() {
+			SimpleEventLoop testling;
+			boost::shared_ptr<MyEventOwner> eventOwner1(new MyEventOwner());
+			boost::shared_ptr<MyEventOwner> eventOwner2(new MyEventOwner());
+
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 1), eventOwner1);
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 2), eventOwner2);
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 3), eventOwner1);
+			testling.postEvent(boost::bind(&EventLoopTest::logEvent, this, 4), eventOwner2);
+			testling.removeEventsFromOwner(eventOwner2);
+			testling.stop();
+			testling.run();
+
+			CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(events_.size()));
+			CPPUNIT_ASSERT_EQUAL(1, events_[0]);
+			CPPUNIT_ASSERT_EQUAL(3, events_[1]);
+		}
+	
+	private:
+		struct MyEventOwner : public EventOwner {};
+		void logEvent(int i) {
+			events_.push_back(i);
+		}
+
+	private:
+		std::vector<int> events_;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(EventLoopTest);
diff --git a/Swiften/EventLoop/UnitTest/SimpleEventLoopTest.cpp b/Swiften/EventLoop/UnitTest/SimpleEventLoopTest.cpp
new file mode 100644
index 0000000..14f24c7
--- /dev/null
+++ b/Swiften/EventLoop/UnitTest/SimpleEventLoopTest.cpp
@@ -0,0 +1,62 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <boost/thread.hpp>
+#include <boost/bind.hpp>
+
+#include "Swiften/EventLoop/SimpleEventLoop.h"
+#include "Swiften/Base/sleep.h"
+
+using namespace Swift;
+
+class SimpleEventLoopTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(SimpleEventLoopTest);
+		CPPUNIT_TEST(testRun);
+		CPPUNIT_TEST(testPostFromMainThread);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		SimpleEventLoopTest() {}
+
+		void setUp() {
+			counter_ = 0;
+		}
+
+		void testRun() {
+			SimpleEventLoop testling;
+			boost::thread thread(boost::bind(&SimpleEventLoopTest::runIncrementingThread, this, &testling));
+			testling.run();
+
+			CPPUNIT_ASSERT_EQUAL(10, counter_);
+		}
+
+		void testPostFromMainThread() {
+			SimpleEventLoop testling;
+			testling.postEvent(boost::bind(&SimpleEventLoopTest::incrementCounterAndStop, this, &testling));
+			testling.run();
+
+			CPPUNIT_ASSERT_EQUAL(1, counter_);
+		}
+
+	private:
+		void runIncrementingThread(SimpleEventLoop* loop) {
+			for (unsigned int i = 0; i < 10; ++i) {
+				Swift::sleep(1);
+				loop->postEvent(boost::bind(&SimpleEventLoopTest::incrementCounter, this));
+			}
+			loop->stop();
+		}
+
+		void incrementCounter() {
+			counter_++;
+		}
+
+		void incrementCounterAndStop(SimpleEventLoop* loop) {
+			counter_++;
+			loop->stop();
+		}
+
+		int counter_;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(SimpleEventLoopTest);