rhubarb-lip-sync/rhubarb/lib/webrtc-8d2248ff/webrtc/libjingle/xmpp/pubsubclient_unittest.cc

280 lines
9.1 KiB
C++
Raw Normal View History

2016-06-21 20:13:05 +00:00
/*
* 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.
*/
#include <memory>
#include <string>
#include "webrtc/libjingle/xmllite/qname.h"
#include "webrtc/libjingle/xmllite/xmlelement.h"
#include "webrtc/libjingle/xmpp/constants.h"
#include "webrtc/libjingle/xmpp/fakexmppclient.h"
#include "webrtc/libjingle/xmpp/jid.h"
#include "webrtc/libjingle/xmpp/pubsubclient.h"
#include "webrtc/base/faketaskrunner.h"
#include "webrtc/base/gunit.h"
#include "webrtc/base/sigslot.h"
struct HandledPubSubItem {
std::string itemid;
std::string payload;
};
class TestPubSubItemsListener : public sigslot::has_slots<> {
public:
TestPubSubItemsListener() : error_count(0) {}
void OnItems(buzz::PubSubClient*,
const std::vector<buzz::PubSubItem>& items) {
for (std::vector<buzz::PubSubItem>::const_iterator item = items.begin();
item != items.end(); ++item) {
HandledPubSubItem handled_item;
handled_item.itemid = item->itemid;
if (item->elem->FirstElement() != NULL) {
handled_item.payload = item->elem->FirstElement()->Str();
}
this->items.push_back(handled_item);
}
}
void OnRequestError(buzz::PubSubClient* client,
const buzz::XmlElement* stanza) {
error_count++;
}
void OnPublishResult(buzz::PubSubClient* client,
const std::string& task_id,
const buzz::XmlElement* item) {
result_task_id = task_id;
}
void OnPublishError(buzz::PubSubClient* client,
const std::string& task_id,
const buzz::XmlElement* item,
const buzz::XmlElement* stanza) {
error_count++;
error_task_id = task_id;
}
void OnRetractResult(buzz::PubSubClient* client,
const std::string& task_id) {
result_task_id = task_id;
}
void OnRetractError(buzz::PubSubClient* client,
const std::string& task_id,
const buzz::XmlElement* stanza) {
error_count++;
error_task_id = task_id;
}
std::vector<HandledPubSubItem> items;
int error_count;
std::string error_task_id;
std::string result_task_id;
};
class PubSubClientTest : public testing::Test {
public:
PubSubClientTest() :
pubsubjid("room@domain.com"),
node("topic"),
itemid("key") {
runner.reset(new rtc::FakeTaskRunner());
xmpp_client = new buzz::FakeXmppClient(runner.get());
client.reset(new buzz::PubSubClient(xmpp_client, pubsubjid, node));
listener.reset(new TestPubSubItemsListener());
client->SignalItems.connect(
listener.get(), &TestPubSubItemsListener::OnItems);
client->SignalRequestError.connect(
listener.get(), &TestPubSubItemsListener::OnRequestError);
client->SignalPublishResult.connect(
listener.get(), &TestPubSubItemsListener::OnPublishResult);
client->SignalPublishError.connect(
listener.get(), &TestPubSubItemsListener::OnPublishError);
client->SignalRetractResult.connect(
listener.get(), &TestPubSubItemsListener::OnRetractResult);
client->SignalRetractError.connect(
listener.get(), &TestPubSubItemsListener::OnRetractError);
}
std::unique_ptr<rtc::FakeTaskRunner> runner;
// xmpp_client deleted by deleting runner.
buzz::FakeXmppClient* xmpp_client;
std::unique_ptr<buzz::PubSubClient> client;
std::unique_ptr<TestPubSubItemsListener> listener;
buzz::Jid pubsubjid;
std::string node;
std::string itemid;
};
TEST_F(PubSubClientTest, TestRequest) {
client->RequestItems();
std::string expected_iq =
"<cli:iq type=\"get\" to=\"room@domain.com\" id=\"0\" "
"xmlns:cli=\"jabber:client\">"
"<pub:pubsub xmlns:pub=\"http://jabber.org/protocol/pubsub\">"
"<pub:items node=\"topic\"/>"
"</pub:pubsub>"
"</cli:iq>";
ASSERT_EQ(1U, xmpp_client->sent_stanzas().size());
EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str());
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'>"
" <pubsub xmlns='http://jabber.org/protocol/pubsub'>"
" <items node='topic'>"
" <item id='key0'>"
" <value0a/>"
" </item>"
" <item id='key1'>"
" <value1a/>"
" </item>"
" </items>"
" </pubsub>"
"</iq>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
ASSERT_EQ(2U, listener->items.size());
EXPECT_EQ("key0", listener->items[0].itemid);
EXPECT_EQ("<pub:value0a xmlns:pub=\"http://jabber.org/protocol/pubsub\"/>",
listener->items[0].payload);
EXPECT_EQ("key1", listener->items[1].itemid);
EXPECT_EQ("<pub:value1a xmlns:pub=\"http://jabber.org/protocol/pubsub\"/>",
listener->items[1].payload);
std::string items_message =
"<message xmlns='jabber:client' from='room@domain.com'>"
" <event xmlns='http://jabber.org/protocol/pubsub#event'>"
" <items node='topic'>"
" <item id='key0'>"
" <value0b/>"
" </item>"
" <item id='key1'>"
" <value1b/>"
" </item>"
" </items>"
" </event>"
"</message>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(items_message));
ASSERT_EQ(4U, listener->items.size());
EXPECT_EQ("key0", listener->items[2].itemid);
EXPECT_EQ("<eve:value0b"
" xmlns:eve=\"http://jabber.org/protocol/pubsub#event\"/>",
listener->items[2].payload);
EXPECT_EQ("key1", listener->items[3].itemid);
EXPECT_EQ("<eve:value1b"
" xmlns:eve=\"http://jabber.org/protocol/pubsub#event\"/>",
listener->items[3].payload);
}
TEST_F(PubSubClientTest, TestRequestError) {
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>"
" <error type='auth'>"
" <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>"
" </error>"
"</iq>";
client->RequestItems();
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
EXPECT_EQ(1, listener->error_count);
}
TEST_F(PubSubClientTest, TestPublish) {
buzz::XmlElement* payload =
new buzz::XmlElement(buzz::QName(buzz::NS_PUBSUB, "value"));
std::string task_id;
client->PublishItem(itemid, payload, &task_id);
std::string expected_iq =
"<cli:iq type=\"set\" to=\"room@domain.com\" id=\"0\" "
"xmlns:cli=\"jabber:client\">"
"<pubsub xmlns=\"http://jabber.org/protocol/pubsub\">"
"<publish node=\"topic\">"
"<item id=\"key\">"
"<value/>"
"</item>"
"</publish>"
"</pubsub>"
"</cli:iq>";
ASSERT_EQ(1U, xmpp_client->sent_stanzas().size());
EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str());
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'/>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
EXPECT_EQ(task_id, listener->result_task_id);
}
TEST_F(PubSubClientTest, TestPublishError) {
buzz::XmlElement* payload =
new buzz::XmlElement(buzz::QName(buzz::NS_PUBSUB, "value"));
std::string task_id;
client->PublishItem(itemid, payload, &task_id);
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>"
" <error type='auth'>"
" <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>"
" </error>"
"</iq>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
EXPECT_EQ(1, listener->error_count);
EXPECT_EQ(task_id, listener->error_task_id);
}
TEST_F(PubSubClientTest, TestRetract) {
std::string task_id;
client->RetractItem(itemid, &task_id);
std::string expected_iq =
"<cli:iq type=\"set\" to=\"room@domain.com\" id=\"0\" "
"xmlns:cli=\"jabber:client\">"
"<pubsub xmlns=\"http://jabber.org/protocol/pubsub\">"
"<retract node=\"topic\" notify=\"true\">"
"<item id=\"key\"/>"
"</retract>"
"</pubsub>"
"</cli:iq>";
ASSERT_EQ(1U, xmpp_client->sent_stanzas().size());
EXPECT_EQ(expected_iq, xmpp_client->sent_stanzas()[0]->Str());
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='result' from='room@domain.com'/>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
EXPECT_EQ(task_id, listener->result_task_id);
}
TEST_F(PubSubClientTest, TestRetractError) {
std::string task_id;
client->RetractItem(itemid, &task_id);
std::string result_iq =
"<iq xmlns='jabber:client' id='0' type='error' from='room@domain.com'>"
" <error type='auth'>"
" <forbidden xmlns='urn:ietf:params:xml:ns:xmpp-stanzas'/>"
" </error>"
"</iq>";
xmpp_client->HandleStanza(buzz::XmlElement::ForStr(result_iq));
EXPECT_EQ(1, listener->error_count);
EXPECT_EQ(task_id, listener->error_task_id);
}