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
111
112
113
114
115
116
117
118
119
120
|
/*
* Copyright (c) 2010 Remko Tronçon
* Licensed under the GNU General Public License v3.
* See Documentation/Licenses/GPLv3.txt for more information.
*/
#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <boost/bind.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include "Swiften/StreamManagement/StanzaAckRequester.h"
#include "Swiften/Elements/Message.h"
using namespace Swift;
namespace Swift {
class StanzaAckRequesterTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(StanzaAckRequesterTest);
CPPUNIT_TEST(testHandleStanzaSent_RequestsAck);
CPPUNIT_TEST(testHandleAckReceived_AcksStanza);
CPPUNIT_TEST(testHandleAckReceived_AcksMultipleStanzas);
CPPUNIT_TEST(testHandleAckReceived_MultipleAcks);
CPPUNIT_TEST(testHandleAckReceived_WrapAround);
CPPUNIT_TEST_SUITE_END();
public:
void setUp() {
acksRequested = 0;
}
void testHandleStanzaSent_RequestsAck() {
std::auto_ptr<StanzaAckRequester> testling(createRequester());
testling->handleStanzaSent(createMessage("m1"));
CPPUNIT_ASSERT_EQUAL(1, acksRequested);
}
void testHandleAckReceived_AcksStanza() {
std::auto_ptr<StanzaAckRequester> testling(createRequester());
testling->handleStanzaSent(createMessage("m1"));
testling->handleAckReceived(1);
CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(ackedStanzas.size()));
CPPUNIT_ASSERT_EQUAL(String("m1"), ackedStanzas[0]->getID());
}
void testHandleAckReceived_AcksMultipleStanzas() {
std::auto_ptr<StanzaAckRequester> testling(createRequester());
testling->handleStanzaSent(createMessage("m1"));
testling->handleStanzaSent(createMessage("m2"));
testling->handleAckReceived(2);
CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(ackedStanzas.size()));
CPPUNIT_ASSERT_EQUAL(String("m1"), ackedStanzas[0]->getID());
CPPUNIT_ASSERT_EQUAL(String("m2"), ackedStanzas[1]->getID());
}
void testHandleAckReceived_MultipleAcks() {
std::auto_ptr<StanzaAckRequester> testling(createRequester());
testling->handleStanzaSent(createMessage("m1"));
testling->handleAckReceived(1);
testling->handleStanzaSent(createMessage("m2"));
testling->handleStanzaSent(createMessage("m3"));
testling->handleAckReceived(3);
CPPUNIT_ASSERT_EQUAL(3, static_cast<int>(ackedStanzas.size()));
CPPUNIT_ASSERT_EQUAL(String("m1"), ackedStanzas[0]->getID());
CPPUNIT_ASSERT_EQUAL(String("m2"), ackedStanzas[1]->getID());
CPPUNIT_ASSERT_EQUAL(String("m3"), ackedStanzas[2]->getID());
}
// Handle stanza ack count wrapping, as per the XEP
void testHandleAckReceived_WrapAround() {
std::auto_ptr<StanzaAckRequester> testling(createRequester());
testling->lastHandledStanzasCount = boost::numeric_cast<unsigned int>((1ULL<<32) - 1);
testling->handleStanzaSent(createMessage("m1"));
testling->handleStanzaSent(createMessage("m2"));
testling->handleAckReceived(1);
CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(ackedStanzas.size()));
CPPUNIT_ASSERT_EQUAL(String("m1"), ackedStanzas[0]->getID());
CPPUNIT_ASSERT_EQUAL(String("m2"), ackedStanzas[1]->getID());
}
private:
Message::ref createMessage(const String& id) {
Message::ref result(new Message());
result->setID(id);
return result;
}
StanzaAckRequester* createRequester() {
StanzaAckRequester* requester = new StanzaAckRequester();
requester->onRequestAck.connect(boost::bind(&StanzaAckRequesterTest::handleRequestAck, this));
requester->onStanzaAcked.connect(boost::bind(&StanzaAckRequesterTest::handleStanzaAcked, this, _1));
return requester;
}
void handleRequestAck() {
acksRequested++;
}
void handleStanzaAcked(boost::shared_ptr<Stanza> stanza) {
ackedStanzas.push_back(stanza);
}
private:
int acksRequested;
std::vector< boost::shared_ptr<Stanza> > ackedStanzas;
};
}
CPPUNIT_TEST_SUITE_REGISTRATION(Swift::StanzaAckRequesterTest);
|
Swift