src/third_party/libjingle/source/talk/xmpp/fakexmppclient.h

   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 "webrtc/libjingle/xmpp/xmpptask.h"
  37
  38 namespace buzz {
  39
  40 class XmlElement;
  41
  42 class FakeXmppClient : public XmppTaskParentInterface,
  43                        public XmppClientInterface {
  44  public:
  45   explicit FakeXmppClient(rtc::TaskParent* parent)
  46       : XmppTaskParentInterface(parent) {
  47   }
  48
  49   // As XmppTaskParentInterface
  50   virtual XmppClientInterface* GetClient() {
  51     return this;
  52   }
  53
  54   virtual int ProcessStart() {
  55     return STATE_RESPONSE;
  56   }
  57
  58   // As XmppClientInterface
  59   virtual XmppEngine::State GetState() const {
  60     return XmppEngine::STATE_OPEN;
  61   }
  62
  63   virtual const Jid& jid() const {
  64     return jid_;
  65   }
  66
  67   virtual std::string NextId() {
  68     // Implement if needed for tests.
  69     return "0";
  70   }
  71
  72   virtual XmppReturnStatus SendStanza(const XmlElement* stanza) {
  73     sent_stanzas_.push_back(stanza);
  74     return XMPP_RETURN_OK;
  75   }
  76
  77   const std::vector<const XmlElement*>& sent_stanzas() {
  78     return sent_stanzas_;
  79   }
  80
  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
  89   virtual void AddXmppTask(XmppTask* task,
  90                            XmppEngine::HandlerLevel level) {
  91     tasks_.push_back(task);
  92   }
  93
  94   virtual void RemoveXmppTask(XmppTask* task) {
  95     std::remove(tasks_.begin(), tasks_.end(), task);
  96   }
  97
  98   // As FakeXmppClient
  99   void set_jid(const Jid& jid) {
 100     jid_ = jid;
 101   }
 102
 103   // Takes ownership of 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_