1 files changed, 74 insertions, 0 deletions
diff --git a/Swiften/LinkLocal/UnitTest/MockDNSSDService.h b/Swiften/LinkLocal/UnitTest/MockDNSSDService.h
new file mode 100644
index 0000000..5d857fb
--- /dev/null
+++ b/Swiften/LinkLocal/UnitTest/MockDNSSDService.h
@@ -0,0 +1,74 @@
+#pragma once
+
+#include <vector>
+#include <map>
+#include <boost/bind.hpp>
+
+#include "Swiften/EventLoop/MainEventLoop.h"
+#include "Swiften/LinkLocal/DNSSDService.h"
+
+namespace Swift {
+	class MockDNSSDService : public DNSSDService {
+		public:
+			MockDNSSDService() {
+			}
+			
+			void start() {
+			}
+
+			void stop() {
+			}
+
+			virtual void registerService(const String&, int, const LinkLocalServiceInfo&) {
+				assert(false);
+			}
+
+			virtual void updateService(const LinkLocalServiceInfo&) {
+				assert(false);
+			}
+
+			virtual void unregisterService() {
+				assert(false);
+			}
+
+			virtual void startResolvingService(const LinkLocalServiceID& id) {
+				resolvingServices.push_back(id);
+				broadcastServiceInfo(id);
+			}
+
+			virtual void stopResolvingService(const LinkLocalServiceID& id) {
+				resolvingServices.erase(std::remove(resolvingServices.begin(), resolvingServices.end(), id), resolvingServices.end());
+			}
+			
+			virtual void resolveHostname(const String&, int) {
+				assert(false);
+			}
+
+			void addService(const LinkLocalServiceID& id) {
+				MainEventLoop::postEvent(boost::bind(boost::ref(onServiceAdded), id));
+			}
+
+			void setServiceInfo(const LinkLocalServiceID& id, const DNSSDService::ResolveResult& info) {
+				serviceInfo.insert(std::make_pair(id, info));
+				broadcastServiceInfo(id);
+			}
+
+		private:
+			void broadcastServiceInfo(const LinkLocalServiceID& id) {
+				if (std::find(resolvingServices.begin(), resolvingServices.end(), id) != resolvingServices.end()) {
+					ServiceInfoMap::const_iterator i = serviceInfo.find(id);
+					if (i != serviceInfo.end()) {
+						MainEventLoop::postEvent(
+								boost::bind(boost::ref(onServiceResolved), id, i->second));
+					}
+				}
+			}
+
+		private:
+			typedef std::map<LinkLocalServiceID,DNSSDService::ResolveResult> ServiceInfoMap;
+			ServiceInfoMap serviceInfo;
+			std::vector<LinkLocalServiceID> resolvingServices;
+	};
+
+
+}