+ * Copyright (c) 2010-2016 Isode Limited.

+#include <map>

+

+#include <boost/bind.hpp>

+

+#include <Swiften/Elements/RosterPayload.h>

+#include <Swiften/EventLoop/DummyEventLoop.h>

+#include <Swiften/LinkLocal/DNSSD/DNSSDServiceID.h>

+#include <Swiften/LinkLocal/LinkLocalServiceBrowser.h>

+#include <Swiften/LinkLocal/LinkLocalServiceInfo.h>

+

+#include <Slimber/LinkLocalPresenceManager.h>

+ CPPUNIT_TEST_SUITE(LinkLocalPresenceManagerTest);

+ CPPUNIT_TEST(testConstructor);

+ CPPUNIT_TEST(testServiceAdded);

+ CPPUNIT_TEST(testServiceRemoved);

+ CPPUNIT_TEST(testServiceChanged);

+ CPPUNIT_TEST(testGetRoster);

+ CPPUNIT_TEST(testGetAllPresence);

+ CPPUNIT_TEST(testGetRoster_InfoWithNick);

+ CPPUNIT_TEST(testGetRoster_InfoWithFirstName);

+ CPPUNIT_TEST(testGetRoster_InfoWithLastName);

+ CPPUNIT_TEST(testGetRoster_InfoWithFirstAndLastName);

+ CPPUNIT_TEST(testGetRoster_NoInfo);

+ CPPUNIT_TEST(testGetServiceForJID);

+ CPPUNIT_TEST(testGetServiceForJID_NoMatch);

+ CPPUNIT_TEST_SUITE_END();

+

+ public:

+ void setUp() {

+ eventLoop = new DummyEventLoop();

+ querier = std::make_shared<FakeDNSSDQuerier>("wonderland.lit", eventLoop);

+ browser = new LinkLocalServiceBrowser(querier);

+ browser->start();

+ }

+

+ void tearDown() {

+ browser->stop();

+ delete browser;

+ delete eventLoop;

+ }

+

+ void testConstructor() {

+ addService("alice@wonderland");

+ addService("rabbit@teaparty");

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(testling->getRoster()->getItems().size()));

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(testling->getAllPresence().size()));

+ }

+

+ void testServiceAdded() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "Alice");

+

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(rosterChanges.size()));

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(rosterChanges[0]->getItems().size()));

+ boost::optional<RosterItemPayload> item = rosterChanges[0]->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT(item);

+ CPPUNIT_ASSERT_EQUAL(std::string("Alice"), item->getName());

+ CPPUNIT_ASSERT_EQUAL(RosterItemPayload::Both, item->getSubscription());

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(presenceChanges.size()));

+ CPPUNIT_ASSERT(StatusShow::Online == presenceChanges[0]->getShow());

+ CPPUNIT_ASSERT(JID("alice@wonderland") == presenceChanges[0]->getFrom());

+ }

+

+ void testServiceRemoved() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+ removeService("alice@wonderland");

+

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(rosterChanges.size()));

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(rosterChanges[1]->getItems().size()));

+ boost::optional<RosterItemPayload> item = rosterChanges[1]->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT(item);

+ CPPUNIT_ASSERT_EQUAL(RosterItemPayload::Remove, item->getSubscription());

+ }

+

+ void testServiceChanged() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+ updateServicePresence("alice@wonderland", LinkLocalServiceInfo::Away, "I'm Away");

+

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(rosterChanges.size()));

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(presenceChanges.size()));

+ CPPUNIT_ASSERT(StatusShow::Away == presenceChanges[1]->getShow());

+ CPPUNIT_ASSERT(JID("alice@wonderland") == presenceChanges[1]->getFrom());

+ CPPUNIT_ASSERT_EQUAL(std::string("I'm Away"), presenceChanges[1]->getStatus());

+ }

+

+ void testGetAllPresence() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+ addService("rabbit@teaparty");

+ updateServicePresence("rabbit@teaparty", LinkLocalServiceInfo::Away, "Partying");

+

+ std::vector<std::shared_ptr<Presence> > presences = testling->getAllPresence();

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(presences.size()));

+ // The order doesn't matter

+ CPPUNIT_ASSERT(JID("rabbit@teaparty") == presences[0]->getFrom());

+ CPPUNIT_ASSERT(StatusShow::Away == presences[0]->getShow());

+ CPPUNIT_ASSERT(JID("alice@wonderland") == presences[1]->getFrom());

+ CPPUNIT_ASSERT(StatusShow::Online == presences[1]->getShow());

+ }

+

+ void testGetRoster() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "Alice");

+ addService("rabbit@teaparty", "Rabbit");

+

+ std::shared_ptr<RosterPayload> roster = testling->getRoster();

+ CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(roster->getItems().size()));

+ boost::optional<RosterItemPayload> item;

+ item = roster->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT(item);

+ CPPUNIT_ASSERT_EQUAL(std::string("Alice"), item->getName());

+ CPPUNIT_ASSERT_EQUAL(RosterItemPayload::Both, item->getSubscription());

+ item = roster->getItem(JID("rabbit@teaparty"));

+ CPPUNIT_ASSERT(item);

+ CPPUNIT_ASSERT_EQUAL(std::string("Rabbit"), item->getName());

+ CPPUNIT_ASSERT_EQUAL(RosterItemPayload::Both, item->getSubscription());

+ }

+

+ void testGetRoster_InfoWithNick() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "Alice", "Alice In", "Wonderland");

+

+ boost::optional<RosterItemPayload> item = testling->getRoster()->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT_EQUAL(std::string("Alice"), item->getName());

+ }

+

+ void testGetRoster_InfoWithFirstName() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "", "Alice In", "");

+

+ boost::optional<RosterItemPayload> item = testling->getRoster()->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT_EQUAL(std::string("Alice In"), item->getName());

+ }

+

+ void testGetRoster_InfoWithLastName() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "", "", "Wonderland");

+

+ boost::optional<RosterItemPayload> item = testling->getRoster()->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT_EQUAL(std::string("Wonderland"), item->getName());

+ }

+

+ void testGetRoster_InfoWithFirstAndLastName() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland", "", "Alice In", "Wonderland");

+

+ boost::optional<RosterItemPayload> item = testling->getRoster()->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT_EQUAL(std::string("Alice In Wonderland"), item->getName());

+ }

+

+ void testGetRoster_NoInfo() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+

+ boost::optional<RosterItemPayload> item = testling->getRoster()->getItem(JID("alice@wonderland"));

+ CPPUNIT_ASSERT_EQUAL(std::string(""), item->getName());

+ }

+

+ void testGetServiceForJID() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+ addService("rabbit@teaparty");

+ addService("queen@garden");

+

+ boost::optional<LinkLocalService> service = testling->getServiceForJID(JID("rabbit@teaparty"));

+ CPPUNIT_ASSERT(service);

+ CPPUNIT_ASSERT_EQUAL(std::string("rabbit@teaparty"), service->getID().getName());

+ }

+

+ void testGetServiceForJID_NoMatch() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(createTestling());

+

+ addService("alice@wonderland");

+ addService("queen@garden");

+

+ CPPUNIT_ASSERT(!testling->getServiceForJID(JID("rabbit@teaparty")));

+ }

+

+ private:

+ std::shared_ptr<LinkLocalPresenceManager> createTestling() {

+ std::shared_ptr<LinkLocalPresenceManager> testling(

+ new LinkLocalPresenceManager(browser));

+ testling->onRosterChanged.connect(boost::bind(

+ &LinkLocalPresenceManagerTest::handleRosterChanged, this, _1));

+ testling->onPresenceChanged.connect(boost::bind(

+ &LinkLocalPresenceManagerTest::handlePresenceChanged, this, _1));

+ return testling;

+ }

+

+ void addService(const std::string& name, const std::string& nickName = std::string(), const std::string& firstName = std::string(), const std::string& lastName = std::string()) {

+ DNSSDServiceID service(name, "local.");

+ LinkLocalServiceInfo info;

+ info.setFirstName(firstName);

+ info.setLastName(lastName);

+ info.setNick(nickName);

+ querier->setServiceInfo(service, DNSSDResolveServiceQuery::Result(name + "._presence._tcp.local", "rabbithole.local", 1234, info.toTXTRecord()));

+ querier->addService(service);

+ eventLoop->processEvents();

+ }

+

+ void removeService(const std::string& name) {

+ DNSSDServiceID service(name, "local.");

+ querier->removeService(DNSSDServiceID(name, "local."));

+ eventLoop->processEvents();

+ }

+

+ void updateServicePresence(const std::string& name, LinkLocalServiceInfo::Status status, const std::string& message) {

+ DNSSDServiceID service(name, "local.");

+ LinkLocalServiceInfo info;

+ info.setStatus(status);

+ info.setMessage(message);

+ querier->setServiceInfo(service, DNSSDResolveServiceQuery::Result(name + "._presence._tcp.local", "rabbithole.local", 1234, info.toTXTRecord()));

+ eventLoop->processEvents();

+ }

+

+ void handleRosterChanged(std::shared_ptr<RosterPayload> payload) {

+ rosterChanges.push_back(payload);

+ }

+

+ void handlePresenceChanged(std::shared_ptr<Presence> presence) {

+ presenceChanges.push_back(presence);

+ }

+

+ private:

+ DummyEventLoop* eventLoop;

+ std::shared_ptr<FakeDNSSDQuerier> querier;

+ LinkLocalServiceBrowser* browser;

+ std::vector< std::shared_ptr<RosterPayload> > rosterChanges;

+ std::vector< std::shared_ptr<Presence> > presenceChanges;

+ * Copyright (c) 2010-2016 Isode Limited.

+#include <Slimber/Menulet.h>

+#include <Slimber/MenuletController.h>

+ CPPUNIT_TEST_SUITE(MenuletControllerTest);

+ CPPUNIT_TEST(testConstructor);

+ CPPUNIT_TEST(testUpdate);

+ CPPUNIT_TEST(testSetXMPPStatus_Online);

+ CPPUNIT_TEST(testSetXMPPStatus_Offline);

+ CPPUNIT_TEST(testSetUserNames);

+ CPPUNIT_TEST(testSetUserNames_NoUsers);

+ CPPUNIT_TEST_SUITE_END();

+

+ public:

+ void setUp() {

+ menulet = new FakeMenulet();

+ }

+

+ void tearDown() {

+ delete menulet;

+ }

+

+ void testConstructor() {

+ MenuletController testling(menulet);

+

+ CPPUNIT_ASSERT_EQUAL(8, static_cast<int>(menulet->items.size()));

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("No online users"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("[Offline] "), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*About*"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*Restart*"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*Exit*"), menulet->items[i++]);

+ }

+

+ void testUpdate() {

+ MenuletController testling(menulet);

+

+ testling.setXMPPStatus("You are connected", MenuletController::Online);

+

+ CPPUNIT_ASSERT_EQUAL(8, static_cast<int>(menulet->items.size()));

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("No online users"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("[Online] You are connected"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*About*"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*Restart*"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("*Exit*"), menulet->items[i++]);

+ }

+

+ void testSetXMPPStatus_Online() {

+ MenuletController testling(menulet);

+

+ testling.setXMPPStatus("You are connected", MenuletController::Online);

+

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("No online users"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("[Online] You are connected"), menulet->items[i++]);

+ }

+

+

+ void testSetXMPPStatus_Offline() {

+ MenuletController testling(menulet);

+

+ testling.setXMPPStatus("You are not connected", MenuletController::Offline);

+

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("No online users"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("[Offline] You are not connected"), menulet->items[i++]);

+ }

+

+ void testSetUserNames() {

+ MenuletController testling(menulet);

+

+ std::vector<std::string> users;

+ users.push_back("Alice In Wonderland");

+ users.push_back("The Mad Hatter");

+ testling.setUserNames(users);

+

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("Online users:"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string(" Alice In Wonderland"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string(" The Mad Hatter"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ }

+

+ void testSetUserNames_NoUsers() {

+ MenuletController testling(menulet);

+

+ std::vector<std::string> users;

+ testling.setUserNames(users);

+

+ size_t i = 0;

+ CPPUNIT_ASSERT_EQUAL(std::string("No online users"), menulet->items[i++]);

+ CPPUNIT_ASSERT_EQUAL(std::string("-"), menulet->items[i++]);

+ }

+

+ private:

+ struct FakeMenulet : public Menulet {

+ virtual void clear() {

+ items.clear();

+ }

+

+ virtual void addItem(const std::string& name, const std::string& icon = std::string()) {

+ std::string result;

+ if (!icon.empty()) {

+ result += "[" + icon + "] ";

+ }

+ result += name;

+ items.push_back(result);

+ }

+

+ virtual void addAboutItem() {

+ items.push_back("*About*");

+ }

+

+ virtual void addRestartItem() {

+ items.push_back("*Restart*");

+ }

+

+ virtual void addExitItem() {

+ items.push_back("*Exit*");

+ }

+

+ virtual void addSeparator() {

+ items.push_back("-");

+ }

+

+ virtual void setIcon(const std::string& i) {

+ icon = i;

+ }

+

+ std::vector<std::string> items;

+ std::string icon;

+ };

+

+ FakeMenulet* menulet;