1 files changed, 63 insertions, 0 deletions
diff --git a/Swiften/LinkLocal/DNSSD/Fake/FakeDNSSDQuerier.h b/Swiften/LinkLocal/DNSSD/Fake/FakeDNSSDQuerier.h
new file mode 100644
index 0000000..f2ec17b
--- /dev/null
+++ b/Swiften/LinkLocal/DNSSD/Fake/FakeDNSSDQuerier.h
@@ -0,0 +1,63 @@
+#pragma once
+
+#include <boost/shared_ptr.hpp>
+#include <boost/enable_shared_from_this.hpp>
+#include <list>
+#include <set>
+
+#include "Swiften/Base/foreach.h"
+#include "Swiften/EventLoop/EventOwner.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDQuerier.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDResolveServiceQuery.h"
+
+namespace Swift {
+	class ByteArray;
+	class FakeDNSSDQuery;
+	class FakeDNSSDBrowseQuery;
+
+	class FakeDNSSDQuerier : 
+			public DNSSDQuerier, 
+			public EventOwner,
+			public boost::enable_shared_from_this<FakeDNSSDQuerier> {
+		public:
+			FakeDNSSDQuerier(const String& domain);
+
+			boost::shared_ptr<DNSSDBrowseQuery> createBrowseQuery();
+			boost::shared_ptr<DNSSDRegisterQuery> createRegisterQuery(
+					const String& name, int port, const ByteArray& info);
+			boost::shared_ptr<DNSSDResolveServiceQuery> createResolveServiceQuery(
+					const DNSSDServiceID&);
+			boost::shared_ptr<DNSSDResolveHostnameQuery> createResolveHostnameQuery(
+					const String& hostname, int interfaceIndex);
+
+			void addRunningQuery(boost::shared_ptr<FakeDNSSDQuery>);
+			void removeRunningQuery(boost::shared_ptr<FakeDNSSDQuery>);
+
+			void addService(const DNSSDServiceID& id);
+			void removeService(const DNSSDServiceID& id);
+			void setServiceInfo(const DNSSDServiceID& id, const DNSSDResolveServiceQuery::Result& info);
+			bool isServiceRegistered(const String& name, int port, const ByteArray& info);
+
+			void setBrowseError();
+			void setRegisterError();
+
+		private:
+			template<typename T>
+			std::vector< boost::shared_ptr<T> > getQueries() const {
+				std::vector< boost::shared_ptr<T> > result;
+				foreach(const boost::shared_ptr<FakeDNSSDQuery>& query, runningQueries) {
+					if (boost::shared_ptr<T> resultQuery = boost::dynamic_pointer_cast<T>(query)) {
+						result.push_back(resultQuery);
+					}
+				}
+				return result;
+			}
+
+		private:
+			String domain;
+			std::list< boost::shared_ptr<FakeDNSSDQuery> > runningQueries;
+			std::set<DNSSDServiceID> services;
+			typedef std::map<DNSSDServiceID,DNSSDResolveServiceQuery::Result> ServiceInfoMap;
+			ServiceInfoMap serviceInfo;
+	};
+}