blob: 1ba70ad151810331857ac5f5f2d54605fcf969fa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
/*
* Copyright (c) 2010-2017 Isode Limited.
* All rights reserved.
* See the COPYING file for more information.
*/
#pragma once
#include <vector>
#include <Swiften/Client/StanzaChannel.h>
namespace Swift {
class DummyStanzaChannel : public StanzaChannel {
public:
DummyStanzaChannel() {}
virtual void sendStanza(std::shared_ptr<Stanza> stanza) {
sentStanzas.push_back(stanza);
}
void setAvailable(bool available) {
available_ = available;
onAvailableChanged(available);
}
virtual void sendIQ(std::shared_ptr<IQ> iq) {
sentStanzas.push_back(iq);
}
virtual void sendMessage(std::shared_ptr<Message> message) {
sentStanzas.push_back(message);
}
virtual void sendPresence(std::shared_ptr<Presence> presence) {
sentStanzas.push_back(presence);
}
virtual std::string getNewIQID() {
std::string id = "test-id";
if (uniqueIDs_) {
id += "-" + std::to_string(idCounter_++);
}
return id;
}
virtual bool isAvailable() const {
return available_;
}
virtual void setStreamManagementEnabled(bool enable) {
streamManagement_ = enable;
}
virtual bool getStreamManagementEnabled() const {
return streamManagement_;
}
template<typename T> bool isRequestAtIndex(size_t index, const JID& jid, IQ::Type type) {
if (index >= sentStanzas.size()) {
return false;
}
std::shared_ptr<IQ> iqStanza = std::dynamic_pointer_cast<IQ>(sentStanzas[index]);
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;
}
std::shared_ptr<IQ> iqStanza = std::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;
}
std::shared_ptr<IQ> iqStanza = std::dynamic_pointer_cast<IQ>(sentStanzas[index]);
return iqStanza && iqStanza->getType() == IQ::Error && iqStanza->getID() == id;
}
template<typename T> std::shared_ptr<T> getStanzaAtIndex(size_t index) {
if (sentStanzas.size() <= index) {
return std::shared_ptr<T>();
}
return std::dynamic_pointer_cast<T>(sentStanzas[index]);
}
template<typename T> size_t countSentStanzaOfType() {
size_t count = 0;
for (auto& stanza : sentStanzas) {
if (std::dynamic_pointer_cast<T>(stanza)) {
count++;
}
}
return count;
}
std::vector<Certificate::ref> getPeerCertificateChain() const {
return std::vector<Certificate::ref>();
}
std::vector<std::shared_ptr<Stanza> > sentStanzas;
bool available_ = true;
bool uniqueIDs_ = false;
unsigned int idCounter_ = 0;
bool streamManagement_ = false;
};
}
|
Swift