1 files changed, 17 insertions, 1 deletions
diff --git a/Swiften/Client/DummyStanzaChannel.h b/Swiften/Client/DummyStanzaChannel.h
index b9f05c3..c2f3919 100644
--- a/Swiften/Client/DummyStanzaChannel.h
+++ b/Swiften/Client/DummyStanzaChannel.h
@@ -8,7 +8,7 @@
 
 #include <vector>
 
-#include "Swiften/Client/StanzaChannel.h"
+#include <Swiften/Client/StanzaChannel.h>
 
 namespace Swift {
 	class DummyStanzaChannel : public StanzaChannel {
@@ -56,6 +56,22 @@ namespace Swift {
 				return iqStanza && iqStanza->getType() == type && iqStanza->getTo() == jid && iqStanza->getPayload<T>();
 			}
 
+			bool isResultAtIndex(size_t index, const std::string& id) {
+				if (index >= sentStanzas.size()) {
+					return false;
+				}
+				boost::shared_ptr<IQ> iqStanza = boost::dynamic_pointer_cast<IQ>(sentStanzas[index]);
+				return iqStanza && iqStanza->getType() == IQ::Result && iqStanza->getID() == id;
+			}
+
+			bool isErrorAtIndex(size_t index, const std::string& id) {
+				if (index >= sentStanzas.size()) {
+					return false;
+				}
+				boost::shared_ptr<IQ> iqStanza = boost::dynamic_pointer_cast<IQ>(sentStanzas[index]);
+				return iqStanza && iqStanza->getType() == IQ::Error && iqStanza->getID() == id;
+			}
+
 			template<typename T> boost::shared_ptr<T> getStanzaAtIndex(size_t index) {
 				if (sentStanzas.size() <= index) {
 					return boost::shared_ptr<T>();