summaryrefslogtreecommitdiffstats
blob: 2c9c52684495cfdce545151c8e8663a8d2c03dbc (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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/optional.hpp>

#include "Swiften/Presence/PresenceOracle.h"
#include "Swiften/Client/DummyStanzaChannel.h"

using namespace Swift;

class PresencePointerPair {
	public:
		boost::shared_ptr<Presence> one;
		boost::shared_ptr<Presence> two;
};

class SubscriptionRequestInfo {
	public:
		boost::optional<JID> jid;
		boost::optional<String> reason;
};

class PresenceOracleTest : public CppUnit::TestFixture
{
		CPPUNIT_TEST_SUITE(PresenceOracleTest);
		CPPUNIT_TEST(testFirstPresence);
		CPPUNIT_TEST(testSecondPresence);
		CPPUNIT_TEST(testSubscriptionRequest);
		CPPUNIT_TEST_SUITE_END();

	private:
		PresenceOracle* oracle_;
		DummyStanzaChannel* stanzaChannel_;

	public:

		void handlePresenceChange(boost::shared_ptr<Presence> newPresence, boost::shared_ptr<Presence> lastPresence, PresencePointerPair* out) {
			CPPUNIT_ASSERT(out->one.get() == NULL);
			CPPUNIT_ASSERT(out->two.get() == NULL);
			out->one = newPresence;
			out->two = lastPresence;
			CPPUNIT_ASSERT(newPresence.get());
			CPPUNIT_ASSERT_EQUAL(newPresence, out->one);
		}
		
		void handlePresenceSubscriptionRequest(const JID& jid, const String& reason, SubscriptionRequestInfo* info) {
			CPPUNIT_ASSERT(!info->jid);
			CPPUNIT_ASSERT(!info->reason);
			info->jid = jid;
			info->reason = reason;
		}

		void setUp() {
			stanzaChannel_ = new DummyStanzaChannel();
			oracle_ = new PresenceOracle(stanzaChannel_);
		}

		void tearDown() {
			delete oracle_;
			delete stanzaChannel_;
		}

		void testFirstPresence() {
			PresencePointerPair out;
			oracle_->onPresenceChange.connect(boost::bind(&PresenceOracleTest::handlePresenceChange, this, _1, _2, &out));

			SubscriptionRequestInfo info;
			oracle_->onPresenceSubscriptionRequest.connect(boost::bind(&PresenceOracleTest::handlePresenceSubscriptionRequest, this, _1, _2, &info));

			boost::shared_ptr<Presence> sentPresence(new Presence("blarb"));
			stanzaChannel_->onPresenceReceived(sentPresence);

			CPPUNIT_ASSERT(!info.jid);
			CPPUNIT_ASSERT(!info.reason);
			CPPUNIT_ASSERT(out.two.get() == NULL);
			CPPUNIT_ASSERT_EQUAL(sentPresence, out.one);
		}
		
		void testSecondPresence() {
			PresencePointerPair out;
			oracle_->onPresenceChange.connect(boost::bind(&PresenceOracleTest::handlePresenceChange, this, _1, _2, &out));

			boost::shared_ptr<Presence> sentPresence1(new Presence("blarb"));
			stanzaChannel_->onPresenceReceived(sentPresence1);
			CPPUNIT_ASSERT_EQUAL(sentPresence1, out.one);
			out.one = boost::shared_ptr<Presence>();
			
			SubscriptionRequestInfo info;
			oracle_->onPresenceSubscriptionRequest.connect(boost::bind(&PresenceOracleTest::handlePresenceSubscriptionRequest, this, _1, _2, &info));

			boost::shared_ptr<Presence> sentPresence2(new Presence("test2"));
			stanzaChannel_->onPresenceReceived(sentPresence2);

			CPPUNIT_ASSERT(!info.jid);
			CPPUNIT_ASSERT(!info.reason);
			CPPUNIT_ASSERT_EQUAL(sentPresence1, out.two);
			CPPUNIT_ASSERT_EQUAL(sentPresence2, out.one);
		}

		void testSubscriptionRequest() {
			PresencePointerPair out;
			oracle_->onPresenceChange.connect(boost::bind(&PresenceOracleTest::handlePresenceChange, this, _1, _2, &out));
			SubscriptionRequestInfo info;
			oracle_->onPresenceSubscriptionRequest.connect(boost::bind(&PresenceOracleTest::handlePresenceSubscriptionRequest, this, _1, _2, &info));

			String reasonText = "Because I want to";
			JID sentJID = JID("me@example.com");

			boost::shared_ptr<Presence> sentPresence(new Presence());
			sentPresence->setType(Presence::Subscribe);
			sentPresence->setFrom(sentJID);
			sentPresence->setStatus(reasonText);
			stanzaChannel_->onPresenceReceived(sentPresence);

			CPPUNIT_ASSERT(info.jid);
			CPPUNIT_ASSERT(info.reason);
			CPPUNIT_ASSERT_EQUAL(sentJID, info.jid.get());
			CPPUNIT_ASSERT_EQUAL(reasonText, info.reason.get());

			CPPUNIT_ASSERT(!out.two);
			CPPUNIT_ASSERT(!out.one);
		}
};
CPPUNIT_TEST_SUITE_REGISTRATION(PresenceOracleTest);