+ 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 = boost::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");

+ boost::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() {

+ boost::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() {

+ boost::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() {

+ boost::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() {

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

+

+ addService("alice@wonderland");

+ addService("rabbit@teaparty");

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

+

+ std::vector<boost::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() {

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

+

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

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

+

+ boost::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() {

+ boost::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() {

+ boost::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() {

+ boost::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() {

+ boost::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() {

+ boost::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() {

+ boost::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() {

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

+

+ addService("alice@wonderland");

+ addService("queen@garden");

+

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

+ }

+

+ private:

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

+ boost::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(boost::shared_ptr<RosterPayload> payload) {

+ rosterChanges.push_back(payload);

+ }

+

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

+ presenceChanges.push_back(presence);

+ }

+

+ private:

+ DummyEventLoop* eventLoop;

+ boost::shared_ptr<FakeDNSSDQuerier> querier;

+ LinkLocalServiceBrowser* browser;

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

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