 28e2075280
			
		
	
	28e2075280
	
	
	
		
			
			trunk/talk git-svn-id: http://webrtc.googlecode.com/svn/trunk@4318 4adac7df-926f-26a2-2b94-8c16560cd09d
		
			
				
	
	
		
			850 lines
		
	
	
		
			30 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			850 lines
		
	
	
		
			30 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
|  * libjingle
 | |
|  * Copyright 2004, Google Inc.
 | |
|  *
 | |
|  * Redistribution and use in source and binary forms, with or without
 | |
|  * modification, are permitted provided that the following conditions are met:
 | |
|  *
 | |
|  *  1. Redistributions of source code must retain the above copyright notice,
 | |
|  *     this list of conditions and the following disclaimer.
 | |
|  *  2. Redistributions in binary form must reproduce the above copyright notice,
 | |
|  *     this list of conditions and the following disclaimer in the documentation
 | |
|  *     and/or other materials provided with the distribution.
 | |
|  *  3. The name of the author may not be used to endorse or promote products
 | |
|  *     derived from this software without specific prior written permission.
 | |
|  *
 | |
|  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 | |
|  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 | |
|  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
 | |
|  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 | |
|  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 | |
|  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 | |
|  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 | |
|  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 | |
|  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 | |
|  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | |
|  */
 | |
| 
 | |
| #include <string>
 | |
| #include <sstream>
 | |
| #include <iostream>
 | |
| 
 | |
| #include "talk/base/gunit.h"
 | |
| #include "talk/base/scoped_ptr.h"
 | |
| #include "talk/xmllite/xmlelement.h"
 | |
| #include "talk/xmpp/xmppengine.h"
 | |
| #include "talk/xmpp/rostermodule.h"
 | |
| #include "talk/xmpp/constants.h"
 | |
| #include "talk/xmpp/util_unittest.h"
 | |
| 
 | |
| #define TEST_OK(x) EXPECT_EQ((x),XMPP_RETURN_OK)
 | |
| #define TEST_BADARGUMENT(x) EXPECT_EQ((x),XMPP_RETURN_BADARGUMENT)
 | |
| 
 | |
| 
 | |
| namespace buzz {
 | |
| 
 | |
| class RosterModuleTest;
 | |
| 
 | |
| static void
 | |
| WriteString(std::ostream& os, const std::string& str) {
 | |
|   os<<str;
 | |
| }
 | |
| 
 | |
| static void
 | |
| WriteSubscriptionState(std::ostream& os, XmppSubscriptionState state)
 | |
| {
 | |
|   switch (state) {
 | |
|     case XMPP_SUBSCRIPTION_NONE:
 | |
|       os<<"none";
 | |
|       break;
 | |
|     case XMPP_SUBSCRIPTION_NONE_ASKED:
 | |
|       os<<"none_asked";
 | |
|       break;
 | |
|     case XMPP_SUBSCRIPTION_TO:
 | |
|       os<<"to";
 | |
|       break;
 | |
|     case XMPP_SUBSCRIPTION_FROM:
 | |
|       os<<"from";
 | |
|       break;
 | |
|     case XMPP_SUBSCRIPTION_FROM_ASKED:
 | |
|       os<<"from_asked";
 | |
|       break;
 | |
|     case XMPP_SUBSCRIPTION_BOTH:
 | |
|       os<<"both";
 | |
|       break;
 | |
|     default:
 | |
|       os<<"unknown";
 | |
|       break;
 | |
|   }
 | |
| }
 | |
| 
 | |
| static void
 | |
| WriteSubscriptionRequestType(std::ostream& os,
 | |
|                              XmppSubscriptionRequestType type) {
 | |
|   switch(type) {
 | |
|     case XMPP_REQUEST_SUBSCRIBE:
 | |
|       os<<"subscribe";
 | |
|       break;
 | |
|     case XMPP_REQUEST_UNSUBSCRIBE:
 | |
|       os<<"unsubscribe";
 | |
|       break;
 | |
|     case XMPP_REQUEST_SUBSCRIBED:
 | |
|       os<<"subscribed";
 | |
|       break;
 | |
|     case XMPP_REQUEST_UNSUBSCRIBED:
 | |
|       os<<"unsubscribe";
 | |
|       break;
 | |
|     default:
 | |
|       os<<"unknown";
 | |
|       break;
 | |
|   }
 | |
| }
 | |
| 
 | |
| static void
 | |
| WritePresenceShow(std::ostream& os, XmppPresenceShow show) {
 | |
|   switch(show) {
 | |
|     case XMPP_PRESENCE_AWAY:
 | |
|       os<<"away";
 | |
|       break;
 | |
|     case XMPP_PRESENCE_CHAT:
 | |
|       os<<"chat";
 | |
|       break;
 | |
|     case XMPP_PRESENCE_DND:
 | |
|       os<<"dnd";
 | |
|       break;
 | |
|     case XMPP_PRESENCE_XA:
 | |
|       os<<"xa";
 | |
|       break;
 | |
|     case XMPP_PRESENCE_DEFAULT:
 | |
|       os<<"[default]";
 | |
|       break;
 | |
|     default:
 | |
|       os<<"[unknown]";
 | |
|       break;
 | |
|   }
 | |
| }
 | |
| 
 | |
| static void
 | |
| WritePresence(std::ostream& os, const XmppPresence* presence) {
 | |
|   if (presence == NULL) {
 | |
|     os<<"NULL";
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   os<<"[Presence jid:";
 | |
|   WriteString(os, presence->jid().Str());
 | |
|   os<<" available:"<<presence->available();
 | |
|   os<<" presence_show:";
 | |
|   WritePresenceShow(os, presence->presence_show());
 | |
|   os<<" priority:"<<presence->priority();
 | |
|   os<<" status:";
 | |
|   WriteString(os, presence->status());
 | |
|   os<<"]"<<presence->raw_xml()->Str();
 | |
| }
 | |
| 
 | |
| static void
 | |
| WriteContact(std::ostream& os, const XmppRosterContact* contact) {
 | |
|   if (contact == NULL) {
 | |
|     os<<"NULL";
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   os<<"[Contact jid:";
 | |
|   WriteString(os, contact->jid().Str());
 | |
|   os<<" name:";
 | |
|   WriteString(os, contact->name());
 | |
|   os<<" subscription_state:";
 | |
|   WriteSubscriptionState(os, contact->subscription_state());
 | |
|   os<<" groups:[";
 | |
|   for(size_t i=0; i < contact->GetGroupCount(); ++i) {
 | |
|     os<<(i==0?"":", ");
 | |
|     WriteString(os, contact->GetGroup(i));
 | |
|   }
 | |
|   os<<"]]"<<contact->raw_xml()->Str();
 | |
| }
 | |
| 
 | |
| //! This session handler saves all calls to a string.  These are events and
 | |
| //! data delivered form the engine to application code.
 | |
| class XmppTestRosterHandler : public XmppRosterHandler {
 | |
| public:
 | |
|   XmppTestRosterHandler() {}
 | |
|   virtual ~XmppTestRosterHandler() {}
 | |
| 
 | |
|   virtual void SubscriptionRequest(XmppRosterModule*,
 | |
|                                    const Jid& requesting_jid,
 | |
|                                    XmppSubscriptionRequestType type,
 | |
|                                    const XmlElement* raw_xml) {
 | |
|     ss_<<"[SubscriptionRequest Jid:" << requesting_jid.Str()<<" type:";
 | |
|     WriteSubscriptionRequestType(ss_, type);
 | |
|     ss_<<"]"<<raw_xml->Str();
 | |
|   }
 | |
| 
 | |
|   //! Some type of presence error has occured
 | |
|   virtual void SubscriptionError(XmppRosterModule*,
 | |
|                                  const Jid& from,
 | |
|                                  const XmlElement* raw_xml) {
 | |
|     ss_<<"[SubscriptionError from:"<<from.Str()<<"]"<<raw_xml->Str();
 | |
|   }
 | |
| 
 | |
|   virtual void RosterError(XmppRosterModule*,
 | |
|                            const XmlElement* raw_xml) {
 | |
|     ss_<<"[RosterError]"<<raw_xml->Str();
 | |
|   }
 | |
| 
 | |
|   //! New presence information has come in
 | |
|   //! The user is notified with the presence object directly.  This info is also
 | |
|   //! added to the store accessable from the engine.
 | |
|   virtual void IncomingPresenceChanged(XmppRosterModule*,
 | |
|                                        const XmppPresence* presence) {
 | |
|     ss_<<"[IncomingPresenceChanged presence:";
 | |
|     WritePresence(ss_, presence);
 | |
|     ss_<<"]";
 | |
|   }
 | |
| 
 | |
|   //! A contact has changed
 | |
|   //! This indicates that the data for a contact may have changed.  No
 | |
|   //! contacts have been added or removed.
 | |
|   virtual void ContactChanged(XmppRosterModule* roster,
 | |
|                               const XmppRosterContact* old_contact,
 | |
|                               size_t index) {
 | |
|     ss_<<"[ContactChanged old_contact:";
 | |
|     WriteContact(ss_, old_contact);
 | |
|     ss_<<" index:"<<index<<" new_contact:";
 | |
|     WriteContact(ss_, roster->GetRosterContact(index));
 | |
|     ss_<<"]";
 | |
|   }
 | |
| 
 | |
|   //! A set of contacts have been added
 | |
|   //! These contacts may have been added in response to the original roster
 | |
|   //! request or due to a "roster push" from the server.
 | |
|   virtual void ContactsAdded(XmppRosterModule* roster,
 | |
|                              size_t index, size_t number) {
 | |
|     ss_<<"[ContactsAdded index:"<<index<<" number:"<<number;
 | |
|     for (size_t i = 0; i < number; ++i) {
 | |
|       ss_<<" "<<(index+i)<<":";
 | |
|       WriteContact(ss_, roster->GetRosterContact(index+i));
 | |
|     }
 | |
|     ss_<<"]";
 | |
|   }
 | |
| 
 | |
|   //! A contact has been removed
 | |
|   //! This contact has been removed form the list.
 | |
|   virtual void ContactRemoved(XmppRosterModule*,
 | |
|                               const XmppRosterContact* removed_contact,
 | |
|                               size_t index) {
 | |
|     ss_<<"[ContactRemoved old_contact:";
 | |
|     WriteContact(ss_, removed_contact);
 | |
|     ss_<<" index:"<<index<<"]";
 | |
|   }
 | |
| 
 | |
|   std::string Str() {
 | |
|     return ss_.str();
 | |
|   }
 | |
| 
 | |
|   std::string StrClear() {
 | |
|     std::string result = ss_.str();
 | |
|     ss_.str("");
 | |
|     return result;
 | |
|   }
 | |
| 
 | |
| private:
 | |
|   std::stringstream ss_;
 | |
| };
 | |
| 
 | |
| //! This is the class that holds all of the unit test code for the
 | |
| //! roster module
 | |
| class RosterModuleTest : public testing::Test {
 | |
| public:
 | |
|   RosterModuleTest() {}
 | |
|   static void RunLogin(RosterModuleTest* obj, XmppEngine* engine,
 | |
|                        XmppTestHandler* handler) {
 | |
|     // Needs to be similar to XmppEngineTest::RunLogin
 | |
|   }
 | |
| };
 | |
| 
 | |
| TEST_F(RosterModuleTest, TestPresence) {
 | |
|   XmlElement* status = new XmlElement(QN_GOOGLE_PSTN_CONFERENCE_STATUS);
 | |
|   status->AddAttr(QN_STATUS, STR_PSTN_CONFERENCE_STATUS_CONNECTING);
 | |
|   XmlElement presence_xml(QN_PRESENCE);
 | |
|   presence_xml.AddElement(status);
 | |
|   talk_base::scoped_ptr<XmppPresence> presence(XmppPresence::Create());
 | |
|   presence->set_raw_xml(&presence_xml);
 | |
|   EXPECT_EQ(presence->connection_status(), XMPP_CONNECTION_STATUS_CONNECTING);
 | |
| }
 | |
| 
 | |
| TEST_F(RosterModuleTest, TestOutgoingPresence) {
 | |
|   std::stringstream dump;
 | |
| 
 | |
|   talk_base::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | |
|   XmppTestHandler handler(engine.get());
 | |
|   XmppTestRosterHandler roster_handler;
 | |
| 
 | |
|   talk_base::scoped_ptr<XmppRosterModule> roster(XmppRosterModule::Create());
 | |
|   roster->set_roster_handler(&roster_handler);
 | |
| 
 | |
|   // Configure the roster module
 | |
|   roster->RegisterEngine(engine.get());
 | |
| 
 | |
|   // Set up callbacks
 | |
|   engine->SetOutputHandler(&handler);
 | |
|   engine->AddStanzaHandler(&handler);
 | |
|   engine->SetSessionHandler(&handler);
 | |
| 
 | |
|   // Set up minimal login info
 | |
|   engine->SetUser(Jid("david@my-server"));
 | |
|   // engine->SetPassword("david");
 | |
| 
 | |
|   // Do the whole login handshake
 | |
|   RunLogin(this, engine.get(), &handler);
 | |
|   EXPECT_EQ("", handler.OutputActivity());
 | |
| 
 | |
|   // Set some presence and broadcast it
 | |
|   TEST_OK(roster->outgoing_presence()->
 | |
|     set_available(XMPP_PRESENCE_AVAILABLE));
 | |
|   TEST_OK(roster->outgoing_presence()->set_priority(-37));
 | |
|   TEST_OK(roster->outgoing_presence()->set_presence_show(XMPP_PRESENCE_DND));
 | |
|   TEST_OK(roster->outgoing_presence()->
 | |
|     set_status("I'm off to the races!<>&"));
 | |
|   TEST_OK(roster->BroadcastPresence());
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence>"
 | |
|       "<priority>-37</priority>"
 | |
|       "<show>dnd</show>"
 | |
|       "<status>I'm off to the races!<>&</status>"
 | |
|     "</presence>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Try some more
 | |
|   TEST_OK(roster->outgoing_presence()->
 | |
|     set_available(XMPP_PRESENCE_UNAVAILABLE));
 | |
|   TEST_OK(roster->outgoing_presence()->set_priority(0));
 | |
|   TEST_OK(roster->outgoing_presence()->set_presence_show(XMPP_PRESENCE_XA));
 | |
|   TEST_OK(roster->outgoing_presence()->set_status("Gone fishin'"));
 | |
|   TEST_OK(roster->BroadcastPresence());
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence type=\"unavailable\">"
 | |
|       "<show>xa</show>"
 | |
|       "<status>Gone fishin'</status>"
 | |
|     "</presence>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Okay -- we are back on
 | |
|   TEST_OK(roster->outgoing_presence()->
 | |
|     set_available(XMPP_PRESENCE_AVAILABLE));
 | |
|   TEST_BADARGUMENT(roster->outgoing_presence()->set_priority(128));
 | |
|   TEST_OK(roster->outgoing_presence()->
 | |
|       set_presence_show(XMPP_PRESENCE_DEFAULT));
 | |
|   TEST_OK(roster->outgoing_presence()->set_status("Cookin' wit gas"));
 | |
|   TEST_OK(roster->BroadcastPresence());
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence>"
 | |
|       "<status>Cookin' wit gas</status>"
 | |
|     "</presence>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Set it via XML
 | |
|   XmlElement presence_input(QN_PRESENCE);
 | |
|   presence_input.AddAttr(QN_TYPE, "unavailable");
 | |
|   presence_input.AddElement(new XmlElement(QN_PRIORITY));
 | |
|   presence_input.AddText("42", 1);
 | |
|   presence_input.AddElement(new XmlElement(QN_STATUS));
 | |
|   presence_input.AddAttr(QN_XML_LANG, "es", 1);
 | |
|   presence_input.AddText("Hola Amigos!", 1);
 | |
|   presence_input.AddElement(new XmlElement(QN_STATUS));
 | |
|   presence_input.AddText("Hey there, friend!", 1);
 | |
|   TEST_OK(roster->outgoing_presence()->set_raw_xml(&presence_input));
 | |
|   TEST_OK(roster->BroadcastPresence());
 | |
| 
 | |
|   WritePresence(dump, roster->outgoing_presence());
 | |
|   EXPECT_EQ(dump.str(),
 | |
|     "[Presence jid: available:0 presence_show:[default] "
 | |
|               "priority:42 status:Hey there, friend!]"
 | |
|     "<cli:presence type=\"unavailable\" xmlns:cli=\"jabber:client\">"
 | |
|       "<cli:priority>42</cli:priority>"
 | |
|       "<cli:status xml:lang=\"es\">Hola Amigos!</cli:status>"
 | |
|       "<cli:status>Hey there, friend!</cli:status>"
 | |
|     "</cli:presence>");
 | |
|   dump.str("");
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence type=\"unavailable\">"
 | |
|       "<priority>42</priority>"
 | |
|       "<status xml:lang=\"es\">Hola Amigos!</status>"
 | |
|       "<status>Hey there, friend!</status>"
 | |
|     "</presence>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Construct a directed presence
 | |
|   talk_base::scoped_ptr<XmppPresence> directed_presence(XmppPresence::Create());
 | |
|   TEST_OK(directed_presence->set_available(XMPP_PRESENCE_AVAILABLE));
 | |
|   TEST_OK(directed_presence->set_priority(120));
 | |
|   TEST_OK(directed_presence->set_status("*very* available"));
 | |
|   TEST_OK(roster->SendDirectedPresence(directed_presence.get(),
 | |
|                                        Jid("myhoney@honey.net")));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence to=\"myhoney@honey.net\">"
 | |
|       "<priority>120</priority>"
 | |
|       "<status>*very* available</status>"
 | |
|     "</presence>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| }
 | |
| 
 | |
| TEST_F(RosterModuleTest, TestIncomingPresence) {
 | |
|   talk_base::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | |
|   XmppTestHandler handler(engine.get());
 | |
|   XmppTestRosterHandler roster_handler;
 | |
| 
 | |
|   talk_base::scoped_ptr<XmppRosterModule> roster(XmppRosterModule::Create());
 | |
|   roster->set_roster_handler(&roster_handler);
 | |
| 
 | |
|   // Configure the roster module
 | |
|   roster->RegisterEngine(engine.get());
 | |
| 
 | |
|   // Set up callbacks
 | |
|   engine->SetOutputHandler(&handler);
 | |
|   engine->AddStanzaHandler(&handler);
 | |
|   engine->SetSessionHandler(&handler);
 | |
| 
 | |
|   // Set up minimal login info
 | |
|   engine->SetUser(Jid("david@my-server"));
 | |
|   // engine->SetPassword("david");
 | |
| 
 | |
|   // Do the whole login handshake
 | |
|   RunLogin(this, engine.get(), &handler);
 | |
|   EXPECT_EQ("", handler.OutputActivity());
 | |
| 
 | |
|   // Load up with a bunch of data
 | |
|   std::string input;
 | |
|   input = "<presence from='maude@example.net/studio' "
 | |
|                     "to='david@my-server/test'/>"
 | |
|           "<presence from='walter@example.net/home' "
 | |
|                     "to='david@my-server/test'>"
 | |
|             "<priority>-10</priority>"
 | |
|             "<show>xa</show>"
 | |
|             "<status>Off bowling</status>"
 | |
|           "</presence>"
 | |
|           "<presence from='walter@example.net/alley' "
 | |
|                     "to='david@my-server/test'>"
 | |
|             "<priority>20</priority>"
 | |
|             "<status>Looking for toes...</status>"
 | |
|           "</presence>"
 | |
|           "<presence from='donny@example.net/alley' "
 | |
|                     "to='david@my-server/test'>"
 | |
|             "<priority>10</priority>"
 | |
|             "<status>Throwing rocks</status>"
 | |
|           "</presence>";
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[IncomingPresenceChanged "
 | |
|       "presence:[Presence jid:maude@example.net/studio available:1 "
 | |
|                  "presence_show:[default] priority:0 status:]"
 | |
|         "<cli:presence from=\"maude@example.net/studio\" "
 | |
|                       "to=\"david@my-server/test\" "
 | |
|                       "xmlns:cli=\"jabber:client\"/>]"
 | |
|     "[IncomingPresenceChanged "
 | |
|       "presence:[Presence jid:walter@example.net/home available:1 "
 | |
|                  "presence_show:xa priority:-10 status:Off bowling]"
 | |
|         "<cli:presence from=\"walter@example.net/home\" "
 | |
|                       "to=\"david@my-server/test\" "
 | |
|                       "xmlns:cli=\"jabber:client\">"
 | |
|           "<cli:priority>-10</cli:priority>"
 | |
|           "<cli:show>xa</cli:show>"
 | |
|           "<cli:status>Off bowling</cli:status>"
 | |
|         "</cli:presence>]"
 | |
|     "[IncomingPresenceChanged "
 | |
|       "presence:[Presence jid:walter@example.net/alley available:1 "
 | |
|                  "presence_show:[default] "
 | |
|                  "priority:20 status:Looking for toes...]"
 | |
|         "<cli:presence from=\"walter@example.net/alley\" "
 | |
|                        "to=\"david@my-server/test\" "
 | |
|                        "xmlns:cli=\"jabber:client\">"
 | |
|           "<cli:priority>20</cli:priority>"
 | |
|           "<cli:status>Looking for toes...</cli:status>"
 | |
|         "</cli:presence>]"
 | |
|     "[IncomingPresenceChanged "
 | |
|       "presence:[Presence jid:donny@example.net/alley available:1 "
 | |
|                  "presence_show:[default] priority:10 status:Throwing rocks]"
 | |
|         "<cli:presence from=\"donny@example.net/alley\" "
 | |
|                       "to=\"david@my-server/test\" "
 | |
|                       "xmlns:cli=\"jabber:client\">"
 | |
|           "<cli:priority>10</cli:priority>"
 | |
|           "<cli:status>Throwing rocks</cli:status>"
 | |
|         "</cli:presence>]");
 | |
|   EXPECT_EQ(handler.OutputActivity(), "");
 | |
|   handler.SessionActivity(); // Ignore the session output
 | |
| 
 | |
|   // Now look at the data structure we've built
 | |
|   EXPECT_EQ(roster->GetIncomingPresenceCount(), static_cast<size_t>(4));
 | |
|   EXPECT_EQ(roster->GetIncomingPresenceForJidCount(Jid("maude@example.net")),
 | |
|             static_cast<size_t>(1));
 | |
|   EXPECT_EQ(roster->GetIncomingPresenceForJidCount(Jid("walter@example.net")),
 | |
|             static_cast<size_t>(2));
 | |
| 
 | |
|   const XmppPresence * presence;
 | |
|   presence = roster->GetIncomingPresenceForJid(Jid("walter@example.net"), 1);
 | |
| 
 | |
|   std::stringstream dump;
 | |
|   WritePresence(dump, presence);
 | |
|   EXPECT_EQ(dump.str(),
 | |
|           "[Presence jid:walter@example.net/alley available:1 "
 | |
|             "presence_show:[default] priority:20 status:Looking for toes...]"
 | |
|               "<cli:presence from=\"walter@example.net/alley\" "
 | |
|                             "to=\"david@my-server/test\" "
 | |
|                             "xmlns:cli=\"jabber:client\">"
 | |
|                 "<cli:priority>20</cli:priority>"
 | |
|                 "<cli:status>Looking for toes...</cli:status>"
 | |
|               "</cli:presence>");
 | |
|   dump.str("");
 | |
| 
 | |
|   // Maude took off...
 | |
|   input = "<presence from='maude@example.net/studio' "
 | |
|                     "to='david@my-server/test' "
 | |
|                     "type='unavailable'>"
 | |
|             "<status>Stealing my rug back</status>"
 | |
|             "<priority>-10</priority>"
 | |
|           "</presence>";
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[IncomingPresenceChanged "
 | |
|       "presence:[Presence jid:maude@example.net/studio available:0 "
 | |
|                  "presence_show:[default] priority:-10 "
 | |
|                  "status:Stealing my rug back]"
 | |
|         "<cli:presence from=\"maude@example.net/studio\" "
 | |
|                       "to=\"david@my-server/test\" type=\"unavailable\" "
 | |
|                       "xmlns:cli=\"jabber:client\">"
 | |
|           "<cli:status>Stealing my rug back</cli:status>"
 | |
|           "<cli:priority>-10</cli:priority>"
 | |
|         "</cli:presence>]");
 | |
|   EXPECT_EQ(handler.OutputActivity(), "");
 | |
|   handler.SessionActivity(); // Ignore the session output
 | |
| }
 | |
| 
 | |
| TEST_F(RosterModuleTest, TestPresenceSubscription) {
 | |
|   talk_base::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | |
|   XmppTestHandler handler(engine.get());
 | |
|   XmppTestRosterHandler roster_handler;
 | |
| 
 | |
|   talk_base::scoped_ptr<XmppRosterModule> roster(XmppRosterModule::Create());
 | |
|   roster->set_roster_handler(&roster_handler);
 | |
| 
 | |
|   // Configure the roster module
 | |
|   roster->RegisterEngine(engine.get());
 | |
| 
 | |
|   // Set up callbacks
 | |
|   engine->SetOutputHandler(&handler);
 | |
|   engine->AddStanzaHandler(&handler);
 | |
|   engine->SetSessionHandler(&handler);
 | |
| 
 | |
|   // Set up minimal login info
 | |
|   engine->SetUser(Jid("david@my-server"));
 | |
|   // engine->SetPassword("david");
 | |
| 
 | |
|   // Do the whole login handshake
 | |
|   RunLogin(this, engine.get(), &handler);
 | |
|   EXPECT_EQ("", handler.OutputActivity());
 | |
| 
 | |
|   // Test incoming requests
 | |
|   std::string input;
 | |
|   input =
 | |
|     "<presence from='maude@example.net' type='subscribe'/>"
 | |
|     "<presence from='maude@example.net' type='unsubscribe'/>"
 | |
|     "<presence from='maude@example.net' type='subscribed'/>"
 | |
|     "<presence from='maude@example.net' type='unsubscribed'/>";
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[SubscriptionRequest Jid:maude@example.net type:subscribe]"
 | |
|       "<cli:presence from=\"maude@example.net\" type=\"subscribe\" "
 | |
|                     "xmlns:cli=\"jabber:client\"/>"
 | |
|     "[SubscriptionRequest Jid:maude@example.net type:unsubscribe]"
 | |
|       "<cli:presence from=\"maude@example.net\" type=\"unsubscribe\" "
 | |
|                     "xmlns:cli=\"jabber:client\"/>"
 | |
|     "[SubscriptionRequest Jid:maude@example.net type:subscribed]"
 | |
|       "<cli:presence from=\"maude@example.net\" type=\"subscribed\" "
 | |
|                     "xmlns:cli=\"jabber:client\"/>"
 | |
|     "[SubscriptionRequest Jid:maude@example.net type:unsubscribe]"
 | |
|       "<cli:presence from=\"maude@example.net\" type=\"unsubscribed\" "
 | |
|                     "xmlns:cli=\"jabber:client\"/>");
 | |
|   EXPECT_EQ(handler.OutputActivity(), "");
 | |
|   handler.SessionActivity(); // Ignore the session output
 | |
| 
 | |
|   TEST_OK(roster->RequestSubscription(Jid("maude@example.net")));
 | |
|   TEST_OK(roster->CancelSubscription(Jid("maude@example.net")));
 | |
|   TEST_OK(roster->ApproveSubscriber(Jid("maude@example.net")));
 | |
|   TEST_OK(roster->CancelSubscriber(Jid("maude@example.net")));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<presence to=\"maude@example.net\" type=\"subscribe\"/>"
 | |
|     "<presence to=\"maude@example.net\" type=\"unsubscribe\"/>"
 | |
|     "<presence to=\"maude@example.net\" type=\"subscribed\"/>"
 | |
|     "<presence to=\"maude@example.net\" type=\"unsubscribed\"/>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| }
 | |
| 
 | |
| TEST_F(RosterModuleTest, TestRosterReceive) {
 | |
|   talk_base::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | |
|   XmppTestHandler handler(engine.get());
 | |
|   XmppTestRosterHandler roster_handler;
 | |
| 
 | |
|   talk_base::scoped_ptr<XmppRosterModule> roster(XmppRosterModule::Create());
 | |
|   roster->set_roster_handler(&roster_handler);
 | |
| 
 | |
|   // Configure the roster module
 | |
|   roster->RegisterEngine(engine.get());
 | |
| 
 | |
|   // Set up callbacks
 | |
|   engine->SetOutputHandler(&handler);
 | |
|   engine->AddStanzaHandler(&handler);
 | |
|   engine->SetSessionHandler(&handler);
 | |
| 
 | |
|   // Set up minimal login info
 | |
|   engine->SetUser(Jid("david@my-server"));
 | |
|   // engine->SetPassword("david");
 | |
| 
 | |
|   // Do the whole login handshake
 | |
|   RunLogin(this, engine.get(), &handler);
 | |
|   EXPECT_EQ("", handler.OutputActivity());
 | |
| 
 | |
|   // Request a roster update
 | |
|   TEST_OK(roster->RequestRosterUpdate());
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),"");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"get\" id=\"2\">"
 | |
|       "<query xmlns=\"jabber:iq:roster\"/>"
 | |
|     "</iq>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Prime the roster with a starting set
 | |
|   std::string input =
 | |
|     "<iq to='david@myserver/test' type='result' id='2'>"
 | |
|       "<query xmlns='jabber:iq:roster'>"
 | |
|         "<item jid='maude@example.net' "
 | |
|               "name='Maude Lebowski' "
 | |
|               "subscription='none' "
 | |
|               "ask='subscribe'>"
 | |
|           "<group>Business Partners</group>"
 | |
|         "</item>"
 | |
|         "<item jid='walter@example.net' "
 | |
|               "name='Walter Sobchak' "
 | |
|               "subscription='both'>"
 | |
|           "<group>Friends</group>"
 | |
|           "<group>Bowling Team</group>"
 | |
|           "<group>Bowling League</group>"
 | |
|         "</item>"
 | |
|         "<item jid='donny@example.net' "
 | |
|               "name='Donny' "
 | |
|               "subscription='both'>"
 | |
|           "<group>Friends</group>"
 | |
|           "<group>Bowling Team</group>"
 | |
|           "<group>Bowling League</group>"
 | |
|         "</item>"
 | |
|         "<item jid='jeffrey@example.net' "
 | |
|               "name='The Big Lebowski' "
 | |
|               "subscription='to'>"
 | |
|           "<group>Business Partners</group>"
 | |
|         "</item>"
 | |
|         "<item jid='jesus@example.net' "
 | |
|               "name='Jesus Quintana' "
 | |
|               "subscription='from'>"
 | |
|           "<group>Bowling League</group>"
 | |
|         "</item>"
 | |
|       "</query>"
 | |
|     "</iq>";
 | |
| 
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[ContactsAdded index:0 number:5 "
 | |
|       "0:[Contact jid:maude@example.net name:Maude Lebowski "
 | |
|                  "subscription_state:none_asked "
 | |
|                  "groups:[Business Partners]]"
 | |
|         "<ros:item jid=\"maude@example.net\" name=\"Maude Lebowski\" "
 | |
|                   "subscription=\"none\" ask=\"subscribe\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Business Partners</ros:group>"
 | |
|         "</ros:item> "
 | |
|       "1:[Contact jid:walter@example.net name:Walter Sobchak "
 | |
|                  "subscription_state:both "
 | |
|                  "groups:[Friends, Bowling Team, Bowling League]]"
 | |
|         "<ros:item jid=\"walter@example.net\" name=\"Walter Sobchak\" "
 | |
|                   "subscription=\"both\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Friends</ros:group>"
 | |
|           "<ros:group>Bowling Team</ros:group>"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|         "</ros:item> "
 | |
|       "2:[Contact jid:donny@example.net name:Donny "
 | |
|                  "subscription_state:both "
 | |
|                  "groups:[Friends, Bowling Team, Bowling League]]"
 | |
|         "<ros:item jid=\"donny@example.net\" name=\"Donny\" "
 | |
|                   "subscription=\"both\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Friends</ros:group>"
 | |
|           "<ros:group>Bowling Team</ros:group>"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|         "</ros:item> "
 | |
|       "3:[Contact jid:jeffrey@example.net name:The Big Lebowski "
 | |
|                  "subscription_state:to "
 | |
|                  "groups:[Business Partners]]"
 | |
|         "<ros:item jid=\"jeffrey@example.net\" name=\"The Big Lebowski\" "
 | |
|                   "subscription=\"to\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Business Partners</ros:group>"
 | |
|         "</ros:item> "
 | |
|       "4:[Contact jid:jesus@example.net name:Jesus Quintana "
 | |
|                  "subscription_state:from groups:[Bowling League]]"
 | |
|         "<ros:item jid=\"jesus@example.net\" name=\"Jesus Quintana\" "
 | |
|                   "subscription=\"from\" xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|         "</ros:item>]");
 | |
|   EXPECT_EQ(handler.OutputActivity(), "");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Request that someone be added
 | |
|   talk_base::scoped_ptr<XmppRosterContact> contact(XmppRosterContact::Create());
 | |
|   TEST_OK(contact->set_jid(Jid("brandt@example.net")));
 | |
|   TEST_OK(contact->set_name("Brandt"));
 | |
|   TEST_OK(contact->AddGroup("Business Partners"));
 | |
|   TEST_OK(contact->AddGroup("Watchers"));
 | |
|   TEST_OK(contact->AddGroup("Friends"));
 | |
|   TEST_OK(contact->RemoveGroup("Friends")); // Maybe not...
 | |
|   TEST_OK(roster->RequestRosterChange(contact.get()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"set\" id=\"3\">"
 | |
|       "<query xmlns=\"jabber:iq:roster\">"
 | |
|         "<item jid=\"brandt@example.net\" "
 | |
|               "name=\"Brandt\">"
 | |
|           "<group>Business Partners</group>"
 | |
|           "<group>Watchers</group>"
 | |
|         "</item>"
 | |
|       "</query>"
 | |
|     "</iq>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Get the push from the server
 | |
|   input =
 | |
|     "<iq type='result' to='david@my-server/test' id='3'/>"
 | |
|     "<iq type='set' id='server_1'>"
 | |
|       "<query xmlns='jabber:iq:roster'>"
 | |
|         "<item jid='brandt@example.net' "
 | |
|               "name='Brandt' "
 | |
|               "subscription='none'>"
 | |
|           "<group>Business Partners</group>"
 | |
|           "<group>Watchers</group>"
 | |
|         "</item>"
 | |
|       "</query>"
 | |
|     "</iq>";
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[ContactsAdded index:5 number:1 "
 | |
|       "5:[Contact jid:brandt@example.net name:Brandt "
 | |
|                  "subscription_state:none "
 | |
|                  "groups:[Business Partners, Watchers]]"
 | |
|         "<ros:item jid=\"brandt@example.net\" name=\"Brandt\" "
 | |
|                   "subscription=\"none\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Business Partners</ros:group>"
 | |
|           "<ros:group>Watchers</ros:group>"
 | |
|         "</ros:item>]");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"result\" id=\"server_1\"/>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Get a contact update
 | |
|   input =
 | |
|     "<iq type='set' id='server_2'>"
 | |
|       "<query xmlns='jabber:iq:roster'>"
 | |
|         "<item jid='walter@example.net' "
 | |
|               "name='Walter Sobchak' "
 | |
|               "subscription='both'>"
 | |
|           "<group>Friends</group>"
 | |
|           "<group>Bowling Team</group>"
 | |
|           "<group>Bowling League</group>"
 | |
|           "<group>Not wrong, just an...</group>"
 | |
|         "</item>"
 | |
|       "</query>"
 | |
|     "</iq>";
 | |
| 
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[ContactChanged "
 | |
|       "old_contact:[Contact jid:walter@example.net name:Walter Sobchak "
 | |
|                            "subscription_state:both "
 | |
|                            "groups:[Friends, Bowling Team, Bowling League]]"
 | |
|         "<ros:item jid=\"walter@example.net\" name=\"Walter Sobchak\" "
 | |
|                   "subscription=\"both\" xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Friends</ros:group>"
 | |
|           "<ros:group>Bowling Team</ros:group>"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|           "</ros:item> "
 | |
|       "index:1 "
 | |
|       "new_contact:[Contact jid:walter@example.net name:Walter Sobchak "
 | |
|                            "subscription_state:both "
 | |
|                            "groups:[Friends, Bowling Team, Bowling League, "
 | |
|                                    "Not wrong, just an...]]"
 | |
|         "<ros:item jid=\"walter@example.net\" name=\"Walter Sobchak\" "
 | |
|                   "subscription=\"both\" xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Friends</ros:group>"
 | |
|           "<ros:group>Bowling Team</ros:group>"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|           "<ros:group>Not wrong, just an...</ros:group>"
 | |
|         "</ros:item>]");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"result\" id=\"server_2\"/>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Remove a contact
 | |
|   TEST_OK(roster->RequestRosterRemove(Jid("jesus@example.net")));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(), "");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"set\" id=\"4\">"
 | |
|       "<query xmlns=\"jabber:iq:roster\" jid=\"jesus@example.net\" "
 | |
|              "subscription=\"remove\"/>"
 | |
|     "</iq>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| 
 | |
|   // Response from the server
 | |
|   input =
 | |
|     "<iq type='result' to='david@my-server/test' id='4'/>"
 | |
|     "<iq type='set' id='server_3'>"
 | |
|       "<query xmlns='jabber:iq:roster'>"
 | |
|         "<item jid='jesus@example.net' "
 | |
|               "subscription='remove'>"
 | |
|         "</item>"
 | |
|       "</query>"
 | |
|     "</iq>";
 | |
|   TEST_OK(engine->HandleInput(input.c_str(), input.length()));
 | |
| 
 | |
|   EXPECT_EQ(roster_handler.StrClear(),
 | |
|     "[ContactRemoved "
 | |
|       "old_contact:[Contact jid:jesus@example.net name:Jesus Quintana "
 | |
|                            "subscription_state:from groups:[Bowling League]]"
 | |
|         "<ros:item jid=\"jesus@example.net\" name=\"Jesus Quintana\" "
 | |
|                   "subscription=\"from\" "
 | |
|                   "xmlns:ros=\"jabber:iq:roster\">"
 | |
|           "<ros:group>Bowling League</ros:group>"
 | |
|         "</ros:item> index:4]");
 | |
|   EXPECT_EQ(handler.OutputActivity(),
 | |
|     "<iq type=\"result\" id=\"server_3\"/>");
 | |
|   EXPECT_EQ(handler.SessionActivity(), "");
 | |
| }
 | |
| 
 | |
| }
 |