OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright 2011 The WebRTC Project Authors. All rights reserved. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 |
| 11 #include <memory> |
| 12 #include <string> |
| 13 |
| 14 #include "webrtc/libjingle/xmllite/qname.h" |
| 15 #include "webrtc/libjingle/xmllite/xmlelement.h" |
| 16 #include "webrtc/libjingle/xmpp/constants.h" |
| 17 #include "webrtc/libjingle/xmpp/fakexmppclient.h" |
| 18 #include "webrtc/libjingle/xmpp/jid.h" |
| 19 #include "webrtc/libjingle/xmpp/pubsubclient.h" |
| 20 #include "webrtc/base/faketaskrunner.h" |
| 21 #include "webrtc/base/gunit.h" |
| 22 #include "webrtc/base/sigslot.h" |
| 23 |
| 24 struct HandledPubSubItem { |
| 25 std::string itemid; |
| 26 std::string payload; |
| 27 }; |
| 28 |
| 29 class TestPubSubItemsListener : public sigslot::has_slots<> { |
| 30 public: |
| 31 TestPubSubItemsListener() : error_count(0) {} |
| 32 |
| 33 void OnItems(buzz::PubSubClient*, |
| 34 const std::vector<buzz::PubSubItem>& items) { |
| 35 for (std::vector<buzz::PubSubItem>::const_iterator item = items.begin(); |
| 36 item != items.end(); ++item) { |
| 37 HandledPubSubItem handled_item; |
| 38 handled_item.itemid = item->itemid; |
| 39 if (item->elem->FirstElement() != NULL) { |
| 40 handled_item.payload = item->elem->FirstElement()->Str(); |
| 41 } |
| 42 this->items.push_back(handled_item); |
| 43 } |
| 44 } |
| 45 |
| 46 void OnRequestError(buzz::PubSubClient* client, |
| 47 const buzz::XmlElement* stanza) { |
| 48 error_count++; |
| 49 } |
| 50 |
| 51 void OnPublishResult(buzz::PubSubClient* client, |
| 52 const std::string& task_id, |
| 53 const buzz::XmlElement* item) { |
| 54 result_task_id = task_id; |
| 55 } |
| 56 |
| 57 void OnPublishError(buzz::PubSubClient* client, |
| 58 const std::string& task_id, |
| 59 const buzz::XmlElement* item, |
| 60 const buzz::XmlElement* stanza) { |
| 61 error_count++; |
| 62 error_task_id = task_id; |
| 63 } |
| 64 |
| 65 void OnRetractResult(buzz::PubSubClient* client, |
| 66 const std::string& task_id) { |
| 67 result_task_id = task_id; |
| 68 } |
| 69 |
| 70 void OnRetractError(buzz::PubSubClient* client, |
| 71 const std::string& task_id, |
| 72 const buzz::XmlElement* stanza) { |
| 73 error_count++; |
| 74 error_task_id = task_id; |
| 75 } |
| 76 |
| 77 std::vector<HandledPubSubItem> items; |
| 78 int error_count; |
| 79 std::string error_task_id; |
| 80 std::string result_task_id; |
| 81 }; |
| 82 |
| 83 class PubSubClientTest : public testing::Test { |
| 84 public: |
| 85 PubSubClientTest() : |
| 86 pubsubjid("room@domain.com"), |
| 87 node("topic"), |
| 88 itemid("key") { |
| 89 runner.reset(new rtc::FakeTaskRunner()); |
| 90 xmpp_client = new buzz::FakeXmppClient(runner.get()); |
| 91 client.reset(new buzz::PubSubClient(xmpp_client, pubsubjid, node)); |
| 92 listener.reset(new TestPubSubItemsListener()); |
| 93 client->SignalItems.connect( |
| 94 listener.get(), &TestPubSubItemsListener::OnItems); |
| 95 client->SignalRequestError.connect( |
| 96 listener.get(), &TestPubSubItemsListener::OnRequestError); |
| 97 client->SignalPublishResult.connect( |
| 98 listener.get(), &TestPubSubItemsListener::OnPublishResult); |
| 99 client->SignalPublishError.connect( |
| 100 listener.get(), &TestPubSubItemsListener::OnPublishError); |
| 101 client->SignalRetractResult.connect( |
| 102 listener.get(), &TestPubSubItemsListener::OnRetractResult); |
| 103 client->SignalRetractError.connect( |
| 104 listener.get(), &TestPubSubItemsListener::OnRetractError); |
| 105 } |
| 106 |
| 107 std::unique_ptr<rtc::FakeTaskRunner> runner; |
| 108 // xmpp_client deleted by deleting runner. |
| 109 buzz::FakeXmppClient* xmpp_client; |
| 110 std::unique_ptr<buzz::PubSubClient> client; |
| 111 std::unique_ptr<TestPubSubItemsListener> listener; |
| 112 buzz::Jid pubsubjid; |
| 113 std::string node; |
| 114 std::string itemid; |
| 115 }; |
| 116 |
| 117 TEST_F(PubSubClientTest, TestRequest) { |
| 118 client->RequestItems(); |
| 119 |
| 120 std::string expected_iq = |
| 121 "<cli:iq type=\"get\" to=\"room@domain.com\" id=\"0\" " |
| 122 "xmlns:cli=\"jabber:client\">" |
| 123 "<pub:pubsub xmlns:pub=\"http://jabber.org/protocol/pubsub\">" |
| 124 "<pub:items node=\"topic\"/>" |
| 125 "</pub:pubsub>" |
| 126 "</cli:iq>"; |
| 127 |
| 128 ASSERT_EQ(1U, xmpp_client->sent_stanzas().size()); |
| 129 EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str()); |
| 130 |
| 131 std::string result_iq = |
| 132 "<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'>" |
| 133 " <pubsub xmlns='http://jabber.org/protocol/pubsub'>" |
| 134 " <items node='topic'>" |
| 135 " <item id='key0'>" |
| 136 " <value0a/>" |
| 137 " </item>" |
| 138 " <item id='key1'>" |
| 139 " <value1a/>" |
| 140 " </item>" |
| 141 " </items>" |
| 142 " </pubsub>" |
| 143 "</iq>"; |
| 144 |
| 145 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 146 ASSERT_EQ(2U, listener->items.size()); |
| 147 EXPECT_EQ("key0", listener->items[0].itemid); |
| 148 EXPECT_EQ("<pub:value0a xmlns:pub=\"http://jabber.org/protocol/pubsub\"/>", |
| 149 listener->items[0].payload); |
| 150 EXPECT_EQ("key1", listener->items[1].itemid); |
| 151 EXPECT_EQ("<pub:value1a xmlns:pub=\"http://jabber.org/protocol/pubsub\"/>", |
| 152 listener->items[1].payload); |
| 153 |
| 154 std::string items_message = |
| 155 "<message xmlns='jabber:client' from='room@domain.com'>" |
| 156 " <event xmlns='http://jabber.org/protocol/pubsub#event'>" |
| 157 " <items node='topic'>" |
| 158 " <item id='key0'>" |
| 159 " <value0b/>" |
| 160 " </item>" |
| 161 " <item id='key1'>" |
| 162 " <value1b/>" |
| 163 " </item>" |
| 164 " </items>" |
| 165 " </event>" |
| 166 "</message>"; |
| 167 |
| 168 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(items_message)); |
| 169 ASSERT_EQ(4U, listener->items.size()); |
| 170 EXPECT_EQ("key0", listener->items[2].itemid); |
| 171 EXPECT_EQ("<eve:value0b" |
| 172 " xmlns:eve=\"http://jabber.org/protocol/pubsub#event\"/>", |
| 173 listener->items[2].payload); |
| 174 EXPECT_EQ("key1", listener->items[3].itemid); |
| 175 EXPECT_EQ("<eve:value1b" |
| 176 " xmlns:eve=\"http://jabber.org/protocol/pubsub#event\"/>", |
| 177 listener->items[3].payload); |
| 178 } |
| 179 |
| 180 TEST_F(PubSubClientTest, TestRequestError) { |
| 181 std::string result_iq = |
| 182 "<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>" |
| 183 " <error type='auth'>" |
| 184 " <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>" |
| 185 " </error>" |
| 186 "</iq>"; |
| 187 |
| 188 client->RequestItems(); |
| 189 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 190 EXPECT_EQ(1, listener->error_count); |
| 191 } |
| 192 |
| 193 TEST_F(PubSubClientTest, TestPublish) { |
| 194 buzz::XmlElement* payload = |
| 195 new buzz::XmlElement(buzz::QName(buzz::NS_PUBSUB, "value")); |
| 196 |
| 197 std::string task_id; |
| 198 client->PublishItem(itemid, payload, &task_id); |
| 199 |
| 200 std::string expected_iq = |
| 201 "<cli:iq type=\"set\" to=\"room@domain.com\" id=\"0\" " |
| 202 "xmlns:cli=\"jabber:client\">" |
| 203 "<pubsub xmlns=\"http://jabber.org/protocol/pubsub\">" |
| 204 "<publish node=\"topic\">" |
| 205 "<item id=\"key\">" |
| 206 "<value/>" |
| 207 "</item>" |
| 208 "</publish>" |
| 209 "</pubsub>" |
| 210 "</cli:iq>"; |
| 211 |
| 212 ASSERT_EQ(1U, xmpp_client->sent_stanzas().size()); |
| 213 EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str()); |
| 214 |
| 215 std::string result_iq = |
| 216 "<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'/>"; |
| 217 |
| 218 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 219 EXPECT_EQ(task_id, listener->result_task_id); |
| 220 } |
| 221 |
| 222 TEST_F(PubSubClientTest, TestPublishError) { |
| 223 buzz::XmlElement* payload = |
| 224 new buzz::XmlElement(buzz::QName(buzz::NS_PUBSUB, "value")); |
| 225 |
| 226 std::string task_id; |
| 227 client->PublishItem(itemid, payload, &task_id); |
| 228 |
| 229 std::string result_iq = |
| 230 "<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>" |
| 231 " <error type='auth'>" |
| 232 " <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>" |
| 233 " </error>" |
| 234 "</iq>"; |
| 235 |
| 236 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 237 EXPECT_EQ(1, listener->error_count); |
| 238 EXPECT_EQ(task_id, listener->error_task_id); |
| 239 } |
| 240 |
| 241 TEST_F(PubSubClientTest, TestRetract) { |
| 242 std::string task_id; |
| 243 client->RetractItem(itemid, &task_id); |
| 244 |
| 245 std::string expected_iq = |
| 246 "<cli:iq type=\"set\" to=\"room@domain.com\" id=\"0\" " |
| 247 "xmlns:cli=\"jabber:client\">" |
| 248 "<pubsub xmlns=\"http://jabber.org/protocol/pubsub\">" |
| 249 "<retract node=\"topic\" notify=\"true\">" |
| 250 "<item id=\"key\"/>" |
| 251 "</retract>" |
| 252 "</pubsub>" |
| 253 "</cli:iq>"; |
| 254 |
| 255 ASSERT_EQ(1U, xmpp_client->sent_stanzas().size()); |
| 256 EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str()); |
| 257 |
| 258 std::string result_iq = |
| 259 "<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'/>"; |
| 260 |
| 261 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 262 EXPECT_EQ(task_id, listener->result_task_id); |
| 263 } |
| 264 |
| 265 TEST_F(PubSubClientTest, TestRetractError) { |
| 266 std::string task_id; |
| 267 client->RetractItem(itemid, &task_id); |
| 268 |
| 269 std::string result_iq = |
| 270 "<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>" |
| 271 " <error type='auth'>" |
| 272 " <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>" |
| 273 " </error>" |
| 274 "</iq>"; |
| 275 |
| 276 xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq)); |
| 277 EXPECT_EQ(1, listener->error_count); |
| 278 EXPECT_EQ(task_id, listener->error_task_id); |
| 279 } |
OLD | NEW |