git-svn-id: http://webrtc.googlecode.com/svn/trunk@6799 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 "webrtc/base/gunit.h"
 | 
						|
#include "webrtc/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);
 | 
						|
  rtc::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;
 | 
						|
 | 
						|
  rtc::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
						|
  XmppTestHandler handler(engine.get());
 | 
						|
  XmppTestRosterHandler roster_handler;
 | 
						|
 | 
						|
  rtc::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
 | 
						|
  rtc::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) {
 | 
						|
  rtc::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
						|
  XmppTestHandler handler(engine.get());
 | 
						|
  XmppTestRosterHandler roster_handler;
 | 
						|
 | 
						|
  rtc::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) {
 | 
						|
  rtc::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
						|
  XmppTestHandler handler(engine.get());
 | 
						|
  XmppTestRosterHandler roster_handler;
 | 
						|
 | 
						|
  rtc::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) {
 | 
						|
  rtc::scoped_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
						|
  XmppTestHandler handler(engine.get());
 | 
						|
  XmppTestRosterHandler roster_handler;
 | 
						|
 | 
						|
  rtc::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
 | 
						|
  rtc::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(), "");
 | 
						|
}
 | 
						|
 | 
						|
}
 |