108 lines
2.5 KiB
C++
108 lines
2.5 KiB
C++
/*
|
|
* Copyright 2011 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
// A fake XmppClient for use in unit tests.
|
|
|
|
#ifndef WEBRTC_LIBJINGLE_XMPP_FAKEXMPPCLIENT_H_
|
|
#define WEBRTC_LIBJINGLE_XMPP_FAKEXMPPCLIENT_H_
|
|
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "webrtc/libjingle/xmpp/xmpptask.h"
|
|
|
|
namespace buzz {
|
|
|
|
class XmlElement;
|
|
|
|
class FakeXmppClient : public XmppTaskParentInterface,
|
|
public XmppClientInterface {
|
|
public:
|
|
explicit FakeXmppClient(rtc::TaskParent* parent)
|
|
: XmppTaskParentInterface(parent) {
|
|
}
|
|
|
|
// As XmppTaskParentInterface
|
|
virtual XmppClientInterface* GetClient() {
|
|
return this;
|
|
}
|
|
|
|
virtual int ProcessStart() {
|
|
return STATE_RESPONSE;
|
|
}
|
|
|
|
// As XmppClientInterface
|
|
virtual XmppEngine::State GetState() const {
|
|
return XmppEngine::STATE_OPEN;
|
|
}
|
|
|
|
virtual const Jid& jid() const {
|
|
return jid_;
|
|
}
|
|
|
|
virtual std::string NextId() {
|
|
// Implement if needed for tests.
|
|
return "0";
|
|
}
|
|
|
|
virtual XmppReturnStatus SendStanza(const XmlElement* stanza) {
|
|
sent_stanzas_.push_back(stanza);
|
|
return XMPP_RETURN_OK;
|
|
}
|
|
|
|
const std::vector<const XmlElement*>& sent_stanzas() {
|
|
return sent_stanzas_;
|
|
}
|
|
|
|
virtual XmppReturnStatus SendStanzaError(
|
|
const XmlElement * pelOriginal,
|
|
XmppStanzaError code,
|
|
const std::string & text) {
|
|
// Implement if needed for tests.
|
|
return XMPP_RETURN_OK;
|
|
}
|
|
|
|
virtual void AddXmppTask(XmppTask* task,
|
|
XmppEngine::HandlerLevel level) {
|
|
tasks_.push_back(task);
|
|
}
|
|
|
|
virtual void RemoveXmppTask(XmppTask* task) {
|
|
std::remove(tasks_.begin(), tasks_.end(), task);
|
|
}
|
|
|
|
// As FakeXmppClient
|
|
void set_jid(const Jid& jid) {
|
|
jid_ = jid;
|
|
}
|
|
|
|
// Takes ownership of stanza.
|
|
void HandleStanza(XmlElement* stanza) {
|
|
for (std::vector<XmppTask*>::iterator task = tasks_.begin();
|
|
task != tasks_.end(); ++task) {
|
|
if ((*task)->HandleStanza(stanza)) {
|
|
delete stanza;
|
|
return;
|
|
}
|
|
}
|
|
delete stanza;
|
|
}
|
|
|
|
private:
|
|
Jid jid_;
|
|
std::vector<XmppTask*> tasks_;
|
|
std::vector<const XmlElement*> sent_stanzas_;
|
|
};
|
|
|
|
} // namespace buzz
|
|
|
|
#endif // WEBRTC_LIBJINGLE_XMPP_FAKEXMPPCLIENT_H_
|