/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <Swift/Controllers/Chat/ChatControllerBase.h>

#include <sstream>

#include <map>

#include <boost/bind.hpp>

#include <boost/shared_ptr.hpp>

#include <boost/smart_ptr/make_shared.hpp>

#include <boost/date_time/posix_time/posix_time.hpp>

#include <boost/numeric/conversion/cast.hpp>

#include <boost/algorithm/string.hpp>

#include <Swiften/Base/format.h>

#include <Swiften/Base/Path.h>

#include <Swiften/Base/String.h>

#include <Swiften/Client/StanzaChannel.h>

#include <Swiften/Elements/Delay.h>

#include <Swiften/Elements/MUCInvitationPayload.h>

#include <Swiften/Elements/MUCUserPayload.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Disco/EntityCapsProvider.h>

#include <Swiften/Queries/Requests/GetSecurityLabelsCatalogRequest.h>

#include <Swiften/Avatars/AvatarManager.h>

#include <Swift/Controllers/Intl.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

#include <Swift/Controllers/UIEvents/JoinMUCUIEvent.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swift/Controllers/UIInterfaces/ChatWindowFactory.h>

#include <Swift/Controllers/XMPPEvents/MUCInviteEvent.h>

#include <Swift/Controllers/HighlightManager.h>

#include <Swift/Controllers/Highlighter.h>

#include <Swift/Controllers/Chat/AutoAcceptMUCInviteDecider.h>

#include <Swift/Controllers/Chat/ChatMessageParser.h>

namespace Swift {

ChatControllerBase::ChatControllerBase(const JID& self, StanzaChannel* stanzaChannel, IQRouter* iqRouter, ChatWindowFactory* chatWindowFactory, const JID &toJID, PresenceOracle* presenceOracle, AvatarManager* avatarManager, bool useDelayForLatency, UIEventStream* eventStream, EventController* eventController, TimerFactory* timerFactory, EntityCapsProvider* entityCapsProvider, HistoryController* historyController, MUCRegistry* mucRegistry, HighlightManager* highlightManager, boost::shared_ptr<ChatMessageParser> chatMessageParser, AutoAcceptMUCInviteDecider* autoAcceptMUCInviteDecider) : selfJID_(self), stanzaChannel_(stanzaChannel), iqRouter_(iqRouter), chatWindowFactory_(chatWindowFactory), toJID_(toJID), labelsEnabled_(false), presenceOracle_(presenceOracle), avatarManager_(avatarManager), useDelayForLatency_(useDelayForLatency), eventController_(eventController), timerFactory_(timerFactory), entityCapsProvider_(entityCapsProvider), historyController_(historyController), mucRegistry_(mucRegistry), chatMessageParser_(chatMessageParser), autoAcceptMUCInviteDecider_(autoAcceptMUCInviteDecider), eventStream_(eventStream) {

chatWindow_ = chatWindowFactory_->createChatWindow(toJID, eventStream);

chatWindow_->onAllMessagesRead.connect(boost::bind(&ChatControllerBase::handleAllMessagesRead, this));

chatWindow_->onSendMessageRequest.connect(boost::bind(&ChatControllerBase::handleSendMessageRequest, this, _1, _2));

chatWindow_->onLogCleared.connect(boost::bind(&ChatControllerBase::handleLogCleared, this));

entityCapsProvider_->onCapsChanged.connect(boost::bind(&ChatControllerBase::handleCapsChanged, this, _1));

highlighter_ = highlightManager->createHighlighter();

setOnline(stanzaChannel->isAvailable() && iqRouter->isAvailable());

createDayChangeTimer();

}

ChatControllerBase::~ChatControllerBase() {

delete chatWindow_;

}

void ChatControllerBase::handleLogCleared() {

cancelReplaces();

}

ChatWindow* ChatControllerBase::detachChatWindow() {

ChatWindow* chatWindow = chatWindow_;

chatWindow_ = NULL;

return chatWindow;

}

void ChatControllerBase::handleCapsChanged(const JID& jid) {

if (jid.compare(toJID_, JID::WithoutResource) == 0) {

handleBareJIDCapsChanged(jid);

}

}

void ChatControllerBase::setCanStartImpromptuChats(bool supportsImpromptu) {

if (chatWindow_) {

chatWindow_->setCanInitiateImpromptuChats(supportsImpromptu);

}

}

void ChatControllerBase::createDayChangeTimer() {

if (timerFactory_) {

boost::posix_time::ptime now = boost::posix_time::second_clock::local_time();

boost::posix_time::ptime midnight(now.date() + boost::gregorian::days(1));

int millisecondsUntilMidnight = boost::numeric_cast<int>((midnight - now).total_milliseconds());

dateChangeTimer_ = boost::shared_ptr<Timer>(timerFactory_->createTimer(millisecondsUntilMidnight));

dateChangeTimer_->onTick.connect(boost::bind(&ChatControllerBase::handleDayChangeTick, this));

dateChangeTimer_->start();

}

}

void ChatControllerBase::handleDayChangeTick() {

dateChangeTimer_->stop();

boost::posix_time::ptime now = boost::posix_time::second_clock::local_time();

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(str(format(QT_TRANSLATE_NOOP("", "The day is now %1%")) % std::string(boost::posix_time::to_iso_extended_string(now)).substr(0,10))), ChatWindow::DefaultDirection);

dayTicked();

createDayChangeTimer();

}

void ChatControllerBase::setEnabled(bool enabled) {

chatWindow_->setInputEnabled(enabled);

}

void ChatControllerBase::setOnline(bool online) {

setEnabled(online);

}

JID ChatControllerBase::getBaseJID() {

return JID(toJID_.toBare());

}

void ChatControllerBase::setAvailableServerFeatures(boost::shared_ptr<DiscoInfo> info) {

if (iqRouter_->isAvailable() && info->hasFeature(DiscoInfo::SecurityLabelsCatalogFeature)) {

GetSecurityLabelsCatalogRequest::ref request = GetSecurityLabelsCatalogRequest::create(getBaseJID(), iqRouter_);

request->onResponse.connect(boost::bind(&ChatControllerBase::handleSecurityLabelsCatalogResponse, this, _1, _2));

request->send();

} else {

chatWindow_->setSecurityLabelsEnabled(false);

labelsEnabled_ = false;

}

}

void ChatControllerBase::handleAllMessagesRead() {

if (!unreadMessages_.empty()) {

targetedUnreadMessages_.clear();

foreach (boost::shared_ptr<StanzaEvent> stanzaEvent, unreadMessages_) {

stanzaEvent->conclude();

}

unreadMessages_.clear();

chatWindow_->setUnreadMessageCount(0);

onUnreadCountChanged();

}

}

int ChatControllerBase::getUnreadCount() {

return boost::numeric_cast<int>(targetedUnreadMessages_.size());

}

void ChatControllerBase::handleSendMessageRequest(const std::string &body, bool isCorrectionMessage) {

if (!stanzaChannel_->isAvailable() || body.empty()) {

return;

}

boost::shared_ptr<Message> message(new Message());

message->setTo(toJID_);

message->setType(Swift::Message::Chat);

message->setBody(body);

if (labelsEnabled_) {

if (!isCorrectionMessage) {

lastLabel_ = chatWindow_->getSelectedSecurityLabel();

}

SecurityLabelsCatalog::Item labelItem = lastLabel_;

if (labelItem.getLabel()) {

message->addPayload(labelItem.getLabel());

}

}

preSendMessageRequest(message);

boost::posix_time::ptime now = boost::posix_time::microsec_clock::universal_time();

if (useDelayForLatency_) {

message->addPayload(boost::make_shared<Delay>(now, selfJID_));

}

if (isCorrectionMessage) {

message->addPayload(boost::shared_ptr<Replace> (new Replace(lastSentMessageStanzaID_)));

}

message->setID(lastSentMessageStanzaID_ = idGenerator_.generateID());

stanzaChannel_->sendMessage(message);

postSendMessage(message->getBody(), boost::dynamic_pointer_cast<Stanza>(message));

onActivity(message->getBody());

#ifdef SWIFT_EXPERIMENTAL_HISTORY

logMessage(body, selfJID_, toJID_, now, false);

#endif

}

void ChatControllerBase::handleSecurityLabelsCatalogResponse(boost::shared_ptr<SecurityLabelsCatalog> catalog, ErrorPayload::ref error) {

if (catalog && !error) {

if (catalog->getItems().size() == 0) {

chatWindow_->setSecurityLabelsEnabled(false);

labelsEnabled_ = false;

} else {

labelsEnabled_ = true;

chatWindow_->setAvailableSecurityLabels(catalog->getItems());

chatWindow_->setSecurityLabelsEnabled(true);

}

} else {

labelsEnabled_ = false;

chatWindow_->setSecurityLabelsError();

}

}

void ChatControllerBase::showChatWindow() {

chatWindow_->show();

}

void ChatControllerBase::activateChatWindow() {

chatWindow_->activate();

}

bool ChatControllerBase::hasOpenWindow() const {

return chatWindow_ && chatWindow_->isVisible();

}

std::string ChatControllerBase::addMessage(const std::string& message, const std::string& senderName, bool senderIsSelf, const boost::shared_ptr<SecurityLabel> label, const boost::filesystem::path& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

if (boost::starts_with(message, "/me ")) {

return chatWindow_->addAction(chatMessageParser_->parseMessageBody(String::getSplittedAtFirst(message, ' ').second), senderName, senderIsSelf, label, pathToString(avatarPath), time, highlight);

} else {

return chatWindow_->addMessage(chatMessageParser_->parseMessageBody(message,highlighter_->getNick(),senderIsSelf), senderName, senderIsSelf, label, pathToString(avatarPath), time, highlight);

}

}

void ChatControllerBase::replaceMessage(const std::string& message, const std::string& id, bool senderIsSelf, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

if (boost::starts_with(message, "/me ")) {

chatWindow_->replaceWithAction(chatMessageParser_->parseMessageBody(String::getSplittedAtFirst(message, ' ').second), id, time, highlight);

} else {

chatWindow_->replaceMessage(chatMessageParser_->parseMessageBody(message,highlighter_->getNick(),senderIsSelf), id, time, highlight);

}

}

bool ChatControllerBase::isFromContact(const JID& from) {

return from.toBare() == toJID_.toBare();

}

void ChatControllerBase::handleIncomingMessage(boost::shared_ptr<MessageEvent> messageEvent) {

preHandleIncomingMessage(messageEvent);

if (messageEvent->isReadable() && !messageEvent->getConcluded()) {

unreadMessages_.push_back(messageEvent);

if (messageEvent->targetsMe()) {

targetedUnreadMessages_.push_back(messageEvent);

}

}

boost::shared_ptr<Message> message = messageEvent->getStanza();

std::string body = message->getBody();

HighlightAction highlight;

if (message->isError()) {

if (!message->getTo().getResource().empty()) {

std::string errorMessage = str(format(QT_TRANSLATE_NOOP("", "Couldn't send message: %1%")) % getErrorMessage(message->getPayload<ErrorPayload>()));

chatWindow_->addErrorMessage(chatMessageParser_->parseMessageBody(errorMessage));

}

}

else if (messageEvent->getStanza()->getPayload<MUCInvitationPayload>()) {

handleMUCInvitation(messageEvent->getStanza());

return;

}

else if (messageEvent->getStanza()->getPayload<MUCUserPayload>() && messageEvent->getStanza()->getPayload<MUCUserPayload>()->getInvite()) {

handleMediatedMUCInvitation(messageEvent->getStanza());

return;

}

else {

if (!messageEvent->isReadable()) {

return;

}

showChatWindow();

JID from = message->getFrom();

std::vector<boost::shared_ptr<Delay> > delayPayloads = message->getPayloads<Delay>();

for (size_t i = 0; useDelayForLatency_ && i < delayPayloads.size(); i++) {

if (!delayPayloads[i]->getFrom()) {

continue;

}

boost::posix_time::ptime now = boost::posix_time::microsec_clock::universal_time();

std::ostringstream s;

s << "The following message took " << (now - delayPayloads[i]->getStamp()).total_milliseconds() / 1000.0 << " seconds to be delivered from " << delayPayloads[i]->getFrom()->toString() << ".";

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(std::string(s.str())), ChatWindow::DefaultDirection);

}

boost::shared_ptr<SecurityLabel> label = message->getPayload<SecurityLabel>();

// Determine the timestamp

boost::posix_time::ptime timeStamp = boost::posix_time::microsec_clock::universal_time();

boost::optional<boost::posix_time::ptime> messageTimeStamp = getMessageTimestamp(message);

if (messageTimeStamp) {

timeStamp = *messageTimeStamp;

}

onActivity(body);

// Highlight

if (!isIncomingMessageFromMe(message)) {

highlight = highlighter_->findAction(body, senderDisplayNameFromMessage(from));

}

boost::shared_ptr<Replace> replace = message->getPayload<Replace>();

if (replace) {

std::string body = message->getBody();

// Should check if the user has a previous message

std::map<JID, std::string>::iterator lastMessage;

lastMessage = lastMessagesUIID_.find(from);

if (lastMessage != lastMessagesUIID_.end()) {

replaceMessage(body, lastMessagesUIID_[from], isIncomingMessageFromMe(message), timeStamp, highlight);

}

}

else {

lastMessagesUIID_[from] = addMessage(body, senderDisplayNameFromMessage(from), isIncomingMessageFromMe(message), label, avatarManager_->getAvatarPath(from), timeStamp, highlight);

}

logMessage(body, from, selfJID_, timeStamp, true);

}

chatWindow_->show();

chatWindow_->setUnreadMessageCount(boost::numeric_cast<int>(unreadMessages_.size()));

onUnreadCountChanged();

postHandleIncomingMessage(messageEvent, highlight);

}

std::string ChatControllerBase::getErrorMessage(boost::shared_ptr<ErrorPayload> error) {

std::string defaultMessage = QT_TRANSLATE_NOOP("", "Error sending message");

if (!error->getText().empty()) {

return error->getText();

}

else {

switch (error->getCondition()) {

case ErrorPayload::BadRequest: return QT_TRANSLATE_NOOP("", "Bad request");

case ErrorPayload::Conflict: return QT_TRANSLATE_NOOP("", "Conflict");

case ErrorPayload::FeatureNotImplemented: return QT_TRANSLATE_NOOP("", "This feature is not implemented");

case ErrorPayload::Forbidden: return QT_TRANSLATE_NOOP("", "Forbidden");

case ErrorPayload::Gone: return QT_TRANSLATE_NOOP("", "Recipient can no longer be contacted");

case ErrorPayload::InternalServerError: return QT_TRANSLATE_NOOP("", "Internal server error");

case ErrorPayload::ItemNotFound: return QT_TRANSLATE_NOOP("", "Item not found");

case ErrorPayload::JIDMalformed: return QT_TRANSLATE_NOOP("", "JID Malformed");

case ErrorPayload::NotAcceptable: return QT_TRANSLATE_NOOP("", "Message was rejected");

case ErrorPayload::NotAllowed: return QT_TRANSLATE_NOOP("", "Not allowed");

case ErrorPayload::NotAuthorized: return QT_TRANSLATE_NOOP("", "Not authorized");

case ErrorPayload::PaymentRequired: return QT_TRANSLATE_NOOP("", "Payment is required");

case ErrorPayload::RecipientUnavailable: return QT_TRANSLATE_NOOP("", "Recipient is unavailable");

case ErrorPayload::Redirect: return QT_TRANSLATE_NOOP("", "Redirect");

case ErrorPayload::RegistrationRequired: return QT_TRANSLATE_NOOP("", "Registration required");

case ErrorPayload::RemoteServerNotFound: return QT_TRANSLATE_NOOP("", "Recipient's server not found");

case ErrorPayload::RemoteServerTimeout: return QT_TRANSLATE_NOOP("", "Remote server timeout");

case ErrorPayload::ResourceConstraint: return QT_TRANSLATE_NOOP("", "The server is low on resources");

case ErrorPayload::ServiceUnavailable: return QT_TRANSLATE_NOOP("", "The service is unavailable");

case ErrorPayload::SubscriptionRequired: return QT_TRANSLATE_NOOP("", "A subscription is required");

case ErrorPayload::UndefinedCondition: return QT_TRANSLATE_NOOP("", "Undefined condition");

case ErrorPayload::UnexpectedRequest: return QT_TRANSLATE_NOOP("", "Unexpected request");

}

}

assert(false);

return defaultMessage;

}

void ChatControllerBase::handleGeneralMUCInvitation(MUCInviteEvent::ref event) {

unreadMessages_.push_back(event);

chatWindow_->show();

chatWindow_->setUnreadMessageCount(boost::numeric_cast<int>(unreadMessages_.size()));

onUnreadCountChanged();

chatWindow_->addMUCInvitation(senderDisplayNameFromMessage(event->getInviter()), event->getRoomJID(), event->getReason(), event->getPassword(), event->getDirect(), event->getImpromptu());

eventController_->handleIncomingEvent(event);

}

void ChatControllerBase::handleMUCInvitation(Message::ref message) {

MUCInvitationPayload::ref invite = message->getPayload<MUCInvitationPayload>();

if (autoAcceptMUCInviteDecider_->isAutoAcceptedInvite(message->getFrom(), invite)) {

eventStream_->send(boost::make_shared<JoinMUCUIEvent>(invite->getJID(), boost::optional<std::string>(), boost::optional<std::string>(), false, false, true));

} else {

MUCInviteEvent::ref inviteEvent = boost::make_shared<MUCInviteEvent>(toJID_, invite->getJID(), invite->getReason(), invite->getPassword(), true, invite->getIsImpromptu());

handleGeneralMUCInvitation(inviteEvent);

}

}

void ChatControllerBase::handleMediatedMUCInvitation(Message::ref message) {

MUCUserPayload::Invite invite = *message->getPayload<MUCUserPayload>()->getInvite();

JID from = message->getFrom();

std::string reason;

if (!invite.reason.empty()) {

reason = invite.reason;

}

std::string password;

if (message->getPayload<MUCUserPayload>()->getPassword()) {

password = *message->getPayload<MUCUserPayload>()->getPassword();

}

MUCInviteEvent::ref inviteEvent = boost::make_shared<MUCInviteEvent>(invite.from, from, reason, password, false, false);

handleGeneralMUCInvitation(inviteEvent);

}

}

/*

* Copyright (c) 2010 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <cppunit/extensions/HelperMacros.h>

#include <cppunit/extensions/TestFactoryRegistry.h>

#include <Swiften/Avatars/NullAvatarManager.h>

#include <Swiften/Base/Algorithm.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Client/ClientBlockListManager.h>

#include <Swiften/Client/DummyNickManager.h>

#include <Swiften/Client/DummyStanzaChannel.h>

#include <Swiften/Client/NickResolver.h>

#include <Swiften/Crypto/CryptoProvider.h>

#include <Swiften/Crypto/PlatformCryptoProvider.h>

#include <Swiften/Disco/CapsProvider.h>

#include <Swiften/Disco/EntityCapsManager.h>

#include <Swiften/EventLoop/DummyEventLoop.h>

#include <Swiften/FileTransfer/UnitTest/DummyFileTransferManager.h>

#include <Swiften/Jingle/JingleSessionManager.h>

#include <Swiften/MUC/MUCRegistry.h>

#include <Swiften/Presence/PresenceOracle.h>

#include <Swiften/Presence/SubscriptionManager.h>

#include <Swiften/Queries/DummyIQChannel.h>

#include <Swiften/Queries/IQRouter.h>

#include <Swiften/Roster/XMPPRosterImpl.h>

#include <Swiften/VCards/VCardMemoryStorage.h>

// #include <Swiften/Elements/Payload.h>

// #include <Swiften/Elements/RosterItemPayload.h>

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

#include <Swift/Controllers/Roster/ContactRosterItem.h>

#include <Swift/Controllers/Roster/GroupRosterItem.h>

#include <Swift/Controllers/Roster/Roster.h>

#include <Swift/Controllers/Roster/RosterController.h>

#include <Swift/Controllers/Settings/DummySettingsProvider.h>

#include <Swift/Controllers/UIEvents/RenameRosterItemUIEvent.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UnitTest/MockMainWindowFactory.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

using namespace Swift;

#define CHILDREN mainWindow_->roster->getRoot()->getChildren()

class DummyCapsProvider : public CapsProvider {

DiscoInfo::ref getCaps(const std::string&) const {return DiscoInfo::ref(new DiscoInfo());}

};

class RosterControllerTest : public CppUnit::TestFixture {

CPPUNIT_TEST_SUITE(RosterControllerTest);

CPPUNIT_TEST(testAdd);

CPPUNIT_TEST(testAddSubscription);

CPPUNIT_TEST(testReceiveRename);

CPPUNIT_TEST(testReceiveRegroup);

CPPUNIT_TEST(testSendRename);

CPPUNIT_TEST(testPresence);

CPPUNIT_TEST(testHighestPresence);

CPPUNIT_TEST(testNotHighestPresence);

CPPUNIT_TEST(testUnavailablePresence);

CPPUNIT_TEST_SUITE_END();

public:

void setUp() {

jid_ = JID("testjid@swift.im/swift");

xmppRoster_ = new XMPPRosterImpl();

avatarManager_ = new NullAvatarManager();

mainWindowFactory_ = new MockMainWindowFactory();

mucRegistry_ = new MUCRegistry();

nickResolver_ = new NickResolver(jid_.toBare(), xmppRoster_, NULL, mucRegistry_);

channel_ = new DummyIQChannel();

router_ = new IQRouter(channel_);

stanzaChannel_ = new DummyStanzaChannel();

presenceOracle_ = new PresenceOracle(stanzaChannel_);

subscriptionManager_ = new SubscriptionManager(stanzaChannel_);

eventController_ = new EventController();

uiEventStream_ = new UIEventStream();

settings_ = new DummySettingsProvider();

nickManager_ = new DummyNickManager();

capsProvider_ = new DummyCapsProvider();

entityCapsManager_ = new EntityCapsManager(capsProvider_, stanzaChannel_);

jingleSessionManager_ = new JingleSessionManager(router_);

ftManager_ = new DummyFileTransferManager();

ftOverview_ = new FileTransferOverview(ftManager_);

clientBlockListManager_ = new ClientBlockListManager(router_);

crypto_ = PlatformCryptoProvider::create();

vcardStorage_ = new VCardMemoryStorage(crypto_);

vcardManager_ = new VCardManager(jid_, router_, vcardStorage_);

rosterController_ = new RosterController(jid_, xmppRoster_, avatarManager_, mainWindowFactory_, nickManager_, nickResolver_, presenceOracle_, subscriptionManager_, eventController_, uiEventStream_, router_, settings_, entityCapsManager_, ftOverview_, clientBlockListManager_, vcardManager_);

mainWindow_ = mainWindowFactory_->last;

}

void tearDown() {

delete rosterController_;

delete vcardManager_;

delete vcardStorage_;

delete crypto_;

delete clientBlockListManager_;

delete ftManager_;

delete jingleSessionManager_;

delete nickManager_;

delete nickResolver_;

delete mucRegistry_;

delete mainWindowFactory_;

delete avatarManager_;

delete router_;

delete channel_;

delete eventController_;

delete subscriptionManager_;

delete presenceOracle_;

delete stanzaChannel_;

delete uiEventStream_;

delete settings_;

delete xmppRoster_;

}

GroupRosterItem* groupChild(size_t i) {

return dynamic_cast<GroupRosterItem*>(CHILDREN[i]);

}

JID withResource(const JID& jid, const std::string& resource) {

return JID(jid.toBare().toString() + "/" + resource);

}

void testPresence() {

std::vector<std::string> groups;

groups.push_back("testGroup1");

groups.push_back("testGroup2");

JID from("test@testdomain.com");

xmppRoster_->addContact(from, "name", groups, RosterItemPayload::Both);

Presence::ref presence(new Presence());

presence->setFrom(withResource(from, "bob"));

presence->setPriority(2);

presence->setStatus("So totally here");

stanzaChannel_->onPresenceReceived(presence);

ContactRosterItem* item = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[0])->getChildren()[0]);

CPPUNIT_ASSERT(item);

CPPUNIT_ASSERT_EQUAL(presence->getStatus(), item->getStatusText());

ContactRosterItem* item2 = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[1])->getChildren()[0]);

CPPUNIT_ASSERT(item2);

CPPUNIT_ASSERT_EQUAL(presence->getStatus(), item2->getStatusText());

}

void testHighestPresence() {

std::vector<std::string> groups;

groups.push_back("testGroup1");

JID from("test@testdomain.com");

xmppRoster_->addContact(from, "name", groups, RosterItemPayload::Both);

Presence::ref lowPresence(new Presence());

lowPresence->setFrom(withResource(from, "bob"));

lowPresence->setPriority(2);

lowPresence->setStatus("Not here");

Presence::ref highPresence(new Presence());

highPresence->setFrom(withResource(from, "bert"));

highPresence->setPriority(10);

highPresence->setStatus("So totally here");

stanzaChannel_->onPresenceReceived(lowPresence);

stanzaChannel_->onPresenceReceived(highPresence);

ContactRosterItem* item = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[0])->getChildren()[0]);

CPPUNIT_ASSERT(item);

CPPUNIT_ASSERT_EQUAL(highPresence->getStatus(), item->getStatusText());

}

void testNotHighestPresence() {

std::vector<std::string> groups;

groups.push_back("testGroup1");

JID from("test@testdomain.com");

xmppRoster_->addContact(from, "name", groups, RosterItemPayload::Both);

Presence::ref lowPresence(new Presence());

lowPresence->setFrom(withResource(from, "bob"));

lowPresence->setPriority(2);

lowPresence->setStatus("Not here");

Presence::ref highPresence(new Presence());

highPresence->setFrom(withResource(from, "bert"));

highPresence->setPriority(10);

highPresence->setStatus("So totally here");

stanzaChannel_->onPresenceReceived(highPresence);

stanzaChannel_->onPresenceReceived(lowPresence);

ContactRosterItem* item = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[0])->getChildren()[0]);

CPPUNIT_ASSERT(item);

CPPUNIT_ASSERT_EQUAL(highPresence->getStatus(), item->getStatusText());

}

void testUnavailablePresence() {

std::vector<std::string> groups;

groups.push_back("testGroup1");

JID from("test@testdomain.com");

xmppRoster_->addContact(from, "name", groups, RosterItemPayload::Both);

Presence::ref lowPresence(new Presence());

lowPresence->setFrom(withResource(from, "bob"));

lowPresence->setPriority(2);

lowPresence->setStatus("Not here");

Presence::ref highPresence(new Presence());

highPresence->setFrom(withResource(from, "bert"));

highPresence->setPriority(10);

highPresence->setStatus("So totally here");

Presence::ref highPresenceOffline(new Presence());

highPresenceOffline->setFrom(withResource(from, "bert"));

highPresenceOffline->setType(Presence::Unavailable);

Presence::ref lowPresenceOffline(new Presence());

lowPresenceOffline->setFrom(withResource(from, "bob"));

lowPresenceOffline->setStatus("Signing out");

lowPresenceOffline->setType(Presence::Unavailable);

stanzaChannel_->onPresenceReceived(lowPresence);

stanzaChannel_->onPresenceReceived(highPresence);

stanzaChannel_->onPresenceReceived(highPresenceOffline);

ContactRosterItem* item = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[0])->getChildren()[0]);

CPPUNIT_ASSERT(item);

/* A verification that if the test fails, it's the RosterController, not the PresenceOracle. */

Presence::ref high = presenceOracle_->getHighestPriorityPresence(from);

CPPUNIT_ASSERT_EQUAL(Presence::Available, high->getType());

CPPUNIT_ASSERT_EQUAL(lowPresence->getStatus(), high->getStatus());

CPPUNIT_ASSERT_EQUAL(StatusShow::Online, item->getStatusShow());

CPPUNIT_ASSERT_EQUAL(lowPresence->getStatus(), item->getStatusText());

stanzaChannel_->onPresenceReceived(lowPresenceOffline);

item = dynamic_cast<ContactRosterItem*>(dynamic_cast<GroupRosterItem*>(CHILDREN[0])->getChildren()[0]);

CPPUNIT_ASSERT(item);

/* A verification that if the test fails, it's the RosterController, not the PresenceOracle. */

high = presenceOracle_->getHighestPriorityPresence(from);

CPPUNIT_ASSERT_EQUAL(Presence::Unavailable, high->getType());

CPPUNIT_ASSERT_EQUAL(lowPresenceOffline->getStatus(), high->getStatus());

CPPUNIT_ASSERT_EQUAL(StatusShow::None, item->getStatusShow());

CPPUNIT_ASSERT_EQUAL(lowPresenceOffline->getStatus(), item->getStatusText());

}

void testAdd() {

std::vector<std::string> groups;

groups.push_back("testGroup1");

groups.push_back("testGroup2");

xmppRoster_->addContact(JID("test@testdomain.com/bob"), "name", groups, RosterItemPayload::Both);

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

//CPPUNIT_ASSERT_EQUAL(std::string("Bob"), xmppRoster_->getNameForJID(JID("foo@bar.com")));

}

void testAddSubscription() {

std::vector<std::string> groups;

JID jid("test@testdomain.com");

xmppRoster_->addContact(jid, "name", groups, RosterItemPayload::None);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

xmppRoster_->addContact(jid, "name", groups, RosterItemPayload::To);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

xmppRoster_->addContact(jid, "name", groups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

}

void testReceiveRename() {

std::vector<std::string> groups;

JID jid("test@testdomain.com");

xmppRoster_->addContact(jid, "name", groups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

CPPUNIT_ASSERT_EQUAL(std::string("name"), groupChild(0)->getChildren()[0]->getDisplayName());

xmppRoster_->addContact(jid, "NewName", groups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

CPPUNIT_ASSERT_EQUAL(std::string("NewName"), groupChild(0)->getChildren()[0]->getDisplayName());

}

void testReceiveRegroup() {

std::vector<std::string> oldGroups;

std::vector<std::string> newGroups;

newGroups.push_back("A Group");

std::vector<std::string> newestGroups;

newestGroups.push_back("Best Group");

JID jid("test@testdomain.com");

xmppRoster_->addContact(jid, "", oldGroups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

CPPUNIT_ASSERT_EQUAL(jid.toString(), groupChild(0)->getChildren()[0]->getDisplayName());

xmppRoster_->addContact(jid, "new name", newGroups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

CPPUNIT_ASSERT_EQUAL(std::string("new name"), groupChild(0)->getChildren()[0]->getDisplayName());

CPPUNIT_ASSERT_EQUAL(std::string("A Group"), groupChild(0)->getDisplayName());

xmppRoster_->addContact(jid, "new name", newestGroups, RosterItemPayload::Both);

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

CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(groupChild(0)->getChildren().size()));

CPPUNIT_ASSERT_EQUAL(std::string("new name"), groupChild(0)->getChildren()[0]->getDisplayName());

CPPUNIT_ASSERT_EQUAL(std::string("Best Group"), groupChild(0)->getDisplayName());

}

void testSendRename() {

JID jid("testling@wonderland.lit");

std::vector<std::string> groups;

groups.push_back("Friends");

groups.push_back("Enemies");

xmppRoster_->addContact(jid, "Bob", groups, RosterItemPayload::From);

CPPUNIT_ASSERT_EQUAL(groups.size(), xmppRoster_->getGroupsForJID(jid).size());

uiEventStream_->send(boost::shared_ptr<UIEvent>(new RenameRosterItemUIEvent(jid, "Robert")));

CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), channel_->iqs_.size());

CPPUNIT_ASSERT_EQUAL(IQ::Set, channel_->iqs_[0]->getType());

boost::shared_ptr<RosterPayload> payload = channel_->iqs_[0]->getPayload<RosterPayload>();

CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), payload->getItems().size());

RosterItemPayload item = payload->getItems()[0];

CPPUNIT_ASSERT_EQUAL(jid, item.getJID());

CPPUNIT_ASSERT_EQUAL(std::string("Robert"), item.getName());

CPPUNIT_ASSERT_EQUAL(groups.size(), item.getGroups().size());

assertVectorsEqual(groups, item.getGroups(), __LINE__);

}

void assertVectorsEqual(const std::vector<std::string>& v1, const std::vector<std::string>& v2, int line) {

foreach (const std::string& entry, v1) {

if (std::find(v2.begin(), v2.end(), entry) == v2.end()) {

std::stringstream stream;

stream << "Couldn't find " << entry << " in v2 (line " << line << ")";

CPPUNIT_FAIL(stream.str());

}

}

}

private:

JID jid_;

XMPPRosterImpl* xmppRoster_;

MUCRegistry* mucRegistry_;

AvatarManager* avatarManager_;

MockMainWindowFactory* mainWindowFactory_;

NickManager* nickManager_;

NickResolver* nickResolver_;

RosterController* rosterController_;

DummyIQChannel* channel_;

DummyStanzaChannel* stanzaChannel_;

IQRouter* router_;

PresenceOracle* presenceOracle_;

SubscriptionManager* subscriptionManager_;

EventController* eventController_;

UIEventStream* uiEventStream_;

MockMainWindow* mainWindow_;

DummySettingsProvider* settings_;

DummyCapsProvider* capsProvider_;

EntityCapsManager* entityCapsManager_;

JingleSessionManager* jingleSessionManager_;

FileTransferManager* ftManager_;

FileTransferOverview* ftOverview_;

ClientBlockListManager* clientBlockListManager_;

CryptoProvider* crypto_;

VCardStorage* vcardStorage_;

VCardManager* vcardManager_;

};

CPPUNIT_TEST_SUITE_REGISTRATION(RosterControllerTest);

/*

* Copyright (c) 2010 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <cppunit/extensions/HelperMacros.h>

#include <cppunit/extensions/TestFactoryRegistry.h>

#include "Swift/Controllers/PreviousStatusStore.h"

using namespace Swift;

class PreviousStatusStoreTest : public CppUnit::TestFixture {

CPPUNIT_TEST_SUITE(PreviousStatusStoreTest);

CPPUNIT_TEST(testGetAll);

//CPPUNIT_TEST(testGetAllLimited);

//CPPUNIT_TEST(testGetSuggestionsInexact);

//CPPUNIT_TEST(testGetSuggestionsExact);

CPPUNIT_TEST_SUITE_END();

public:

void setUp() {

store_ = new PreviousStatusStore();

store_->addStatus(StatusShow::Online, "At home in the study");

store_->addStatus(StatusShow::DND, "In a meeting");

store_->addStatus(StatusShow::DND, "With a client");

store_->addStatus(StatusShow::Away, "Walking the elephant");

store_->addStatus(StatusShow::Online, "In the office, at my desk");

}

void tearDown() {

}

void testGetAll() {

}

private:

PreviousStatusStore* store_;

};

CPPUNIT_TEST_SUITE_REGISTRATION(PreviousStatusStoreTest);