1 files changed, 8 insertions, 2 deletions
diff --git a/Swiften/Client/DummyStanzaChannel.h b/Swiften/Client/DummyStanzaChannel.h
index d43d68a..05066a8 100644
--- a/Swiften/Client/DummyStanzaChannel.h
+++ b/Swiften/Client/DummyStanzaChannel.h
@@ -11,16 +11,21 @@
 #include "Swiften/Client/StanzaChannel.h"
 namespace Swift {
        class DummyStanzaChannel : public StanzaChannel {
                public:
-                        DummyStanzaChannel() {}
+                        DummyStanzaChannel() : available_(true) {}
                        virtual void sendStanza(boost::shared_ptr<Stanza> stanza) {
                                sentStanzas.push_back(stanza);
                        }
+                        void setAvailable(bool available) {
+                                available_ = available;
+                                onAvailableChanged(available);
+                        }
                        virtual void sendIQ(boost::shared_ptr<IQ> iq) {
                                sentStanzas.push_back(iq);
                        }
                        virtual void sendMessage(boost::shared_ptr<Message> message) {
@@ -34,11 +39,11 @@ namespace Swift {
                        virtual String getNewIQID() {
                                return "test-id";
                        }
                        
                        virtual bool isAvailable() {
-                                return true;
+                                return available_;
                        }
                        virtual bool getStreamManagementEnabled() const {
                                return false;
                        }
@@ -47,7 +52,8 @@ namespace Swift {
                                boost::shared_ptr<IQ> iqStanza = boost::dynamic_pointer_cast<IQ>(sentStanzas[index]);
                                return iqStanza && iqStanza->getType() == type && iqStanza->getTo() == jid && iqStanza->getPayload<T>();
                        }
                        std::vector<boost::shared_ptr<Stanza> > sentStanzas;
+                        bool available_;
        };
 }