| Index: third_party/xmpp/rostermodule_unittest.cc
 | 
| diff --git a/third_party/xmpp/rostermodule_unittest.cc b/third_party/xmpp/rostermodule_unittest.cc
 | 
| new file mode 100644
 | 
| index 0000000000000000000000000000000000000000..59d3e31f562e4037f9b81cc6ddbfc119f041a9ca
 | 
| --- /dev/null
 | 
| +++ b/third_party/xmpp/rostermodule_unittest.cc
 | 
| @@ -0,0 +1,832 @@
 | 
| +/*
 | 
| + *  Copyright 2004 The WebRTC Project Authors. All rights reserved.
 | 
| + *
 | 
| + *  Use of this source code is governed by a BSD-style license
 | 
| + *  that can be found in the LICENSE file in the root of the source
 | 
| + *  tree. An additional intellectual property rights grant can be found
 | 
| + *  in the file PATENTS.  All contributing project authors may
 | 
| + *  be found in the AUTHORS file in the root of the source tree.
 | 
| + */
 | 
| +
 | 
| +#include <iostream>
 | 
| +#include <memory>
 | 
| +#include <sstream>
 | 
| +#include <string>
 | 
| +
 | 
| +#include "third_party/xmllite/xmlelement.h"
 | 
| +#include "third_party/xmpp/constants.h"
 | 
| +#include "third_party/xmpp/rostermodule.h"
 | 
| +#include "third_party/xmpp/util_unittest.h"
 | 
| +#include "third_party/xmpp/xmppengine.h"
 | 
| +#include "webrtc/base/gunit.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);
 | 
| +  std::unique_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;
 | 
| +
 | 
| +  std::unique_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
| +  XmppTestHandler handler(engine.get());
 | 
| +  XmppTestRosterHandler roster_handler;
 | 
| +
 | 
| +  std::unique_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
 | 
| +  std::unique_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) {
 | 
| +  std::unique_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
| +  XmppTestHandler handler(engine.get());
 | 
| +  XmppTestRosterHandler roster_handler;
 | 
| +
 | 
| +  std::unique_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) {
 | 
| +  std::unique_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
| +  XmppTestHandler handler(engine.get());
 | 
| +  XmppTestRosterHandler roster_handler;
 | 
| +
 | 
| +  std::unique_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) {
 | 
| +  std::unique_ptr<XmppEngine> engine(XmppEngine::Create());
 | 
| +  XmppTestHandler handler(engine.get());
 | 
| +  XmppTestRosterHandler roster_handler;
 | 
| +
 | 
| +  std::unique_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
 | 
| +  std::unique_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(), "");
 | 
| +}
 | 
| +
 | 
| +}
 | 
| 
 |