1 /* 2 * libjingle 3 * Copyright 2011, Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 // A fake XmppClient for use in unit tests. 29 30 #ifndef TALK_XMPP_FAKEXMPPCLIENT_H_ 31 #define TALK_XMPP_FAKEXMPPCLIENT_H_ 32 33 #include <string> 34 #include <vector> 35 36 #include "talk/xmpp/xmpptask.h" 37 38 namespace buzz { 39 40 class XmlElement; 41 42 class FakeXmppClient : public XmppTaskParentInterface, 43 public XmppClientInterface { 44 public: FakeXmppClient(rtc::TaskParent * parent)45 explicit FakeXmppClient(rtc::TaskParent* parent) 46 : XmppTaskParentInterface(parent) { 47 } 48 49 // As XmppTaskParentInterface GetClient()50 virtual XmppClientInterface* GetClient() { 51 return this; 52 } 53 ProcessStart()54 virtual int ProcessStart() { 55 return STATE_RESPONSE; 56 } 57 58 // As XmppClientInterface GetState()59 virtual XmppEngine::State GetState() const { 60 return XmppEngine::STATE_OPEN; 61 } 62 jid()63 virtual const Jid& jid() const { 64 return jid_; 65 } 66 NextId()67 virtual std::string NextId() { 68 // Implement if needed for tests. 69 return "0"; 70 } 71 SendStanza(const XmlElement * stanza)72 virtual XmppReturnStatus SendStanza(const XmlElement* stanza) { 73 sent_stanzas_.push_back(stanza); 74 return XMPP_RETURN_OK; 75 } 76 sent_stanzas()77 const std::vector<const XmlElement*>& sent_stanzas() { 78 return sent_stanzas_; 79 } 80 SendStanzaError(const XmlElement * pelOriginal,XmppStanzaError code,const std::string & text)81 virtual XmppReturnStatus SendStanzaError( 82 const XmlElement * pelOriginal, 83 XmppStanzaError code, 84 const std::string & text) { 85 // Implement if needed for tests. 86 return XMPP_RETURN_OK; 87 } 88 AddXmppTask(XmppTask * task,XmppEngine::HandlerLevel level)89 virtual void AddXmppTask(XmppTask* task, 90 XmppEngine::HandlerLevel level) { 91 tasks_.push_back(task); 92 } 93 RemoveXmppTask(XmppTask * task)94 virtual void RemoveXmppTask(XmppTask* task) { 95 std::remove(tasks_.begin(), tasks_.end(), task); 96 } 97 98 // As FakeXmppClient set_jid(const Jid & jid)99 void set_jid(const Jid& jid) { 100 jid_ = jid; 101 } 102 103 // Takes ownership of stanza. HandleStanza(XmlElement * stanza)104 void HandleStanza(XmlElement* stanza) { 105 for (std::vector<XmppTask*>::iterator task = tasks_.begin(); 106 task != tasks_.end(); ++task) { 107 if ((*task)->HandleStanza(stanza)) { 108 delete stanza; 109 return; 110 } 111 } 112 delete stanza; 113 } 114 115 private: 116 Jid jid_; 117 std::vector<XmppTask*> tasks_; 118 std::vector<const XmlElement*> sent_stanzas_; 119 }; 120 121 } // namespace buzz 122 123 #endif // TALK_XMPP_FAKEXMPPCLIENT_H_ 124