mirror of
https://github.com/qTox/qTox.git
synced 2024-03-22 14:00:36 +08:00
372 lines
12 KiB
C++
372 lines
12 KiB
C++
|
|
/*
|
|
Copyright © 2019 by The qTox Project Contributors
|
|
|
|
This file is part of qTox, a Qt-based graphical interface for Tox.
|
|
|
|
qTox is libre software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
qTox is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with qTox. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "src/model/notificationgenerator.h"
|
|
|
|
#include "mock/mockcoreidhandler.h"
|
|
#include "mock/mockgroupquery.h"
|
|
|
|
#include <QObject>
|
|
#include <QtTest/QtTest>
|
|
|
|
namespace
|
|
{
|
|
class MockNotificationSettings : public INotificationSettings
|
|
{
|
|
virtual bool getNotify() const override { return true; }
|
|
|
|
virtual void setNotify(bool newValue) override {}
|
|
|
|
virtual bool getShowWindow() const override { return true; }
|
|
virtual void setShowWindow(bool newValue) override {}
|
|
|
|
virtual bool getDesktopNotify() const override { return true; }
|
|
virtual void setDesktopNotify(bool enabled) override {}
|
|
|
|
virtual bool getNotifySound() const override { return true; }
|
|
virtual void setNotifySound(bool newValue) override {}
|
|
|
|
virtual bool getNotifyHide() const override { return notifyHide; }
|
|
virtual void setNotifyHide(bool newValue) override { notifyHide = newValue; };
|
|
|
|
virtual bool getBusySound() const override { return true; }
|
|
virtual void setBusySound(bool newValue) override {}
|
|
|
|
virtual bool getGroupAlwaysNotify() const override { return true; }
|
|
virtual void setGroupAlwaysNotify(bool newValue) override {}
|
|
private:
|
|
bool notifyHide = false;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
class TestNotificationGenerator : public QObject
|
|
{
|
|
Q_OBJECT
|
|
|
|
private slots:
|
|
void init();
|
|
void testSingleFriendMessage();
|
|
void testMultipleFriendMessages();
|
|
void testNotificationClear();
|
|
void testGroupMessage();
|
|
void testMultipleGroupMessages();
|
|
void testMultipleFriendSourceMessages();
|
|
void testMultipleGroupSourceMessages();
|
|
void testMixedSourceMessages();
|
|
void testFileTransfer();
|
|
void testFileTransferAfterMessage();
|
|
void testGroupInvitation();
|
|
void testGroupInviteUncounted();
|
|
void testFriendRequest();
|
|
void testFriendRequestUncounted();
|
|
void testSimpleFriendMessage();
|
|
void testSimpleFileTransfer();
|
|
void testSimpleGroupMessage();
|
|
void testSimpleFriendRequest();
|
|
void testSimpleGroupInvite();
|
|
void testSimpleMessageToggle();
|
|
|
|
private:
|
|
std::unique_ptr<INotificationSettings> notificationSettings;
|
|
std::unique_ptr<NotificationGenerator> notificationGenerator;
|
|
std::unique_ptr<MockGroupQuery> groupQuery;
|
|
std::unique_ptr<MockCoreIdHandler> coreIdHandler;
|
|
};
|
|
|
|
void TestNotificationGenerator::init()
|
|
{
|
|
notificationSettings.reset(new MockNotificationSettings());
|
|
notificationGenerator.reset(new NotificationGenerator(*notificationSettings, nullptr));
|
|
groupQuery.reset(new MockGroupQuery());
|
|
coreIdHandler.reset(new MockCoreIdHandler());
|
|
}
|
|
|
|
void TestNotificationGenerator::testSingleFriendMessage()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friendName");
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test");
|
|
QVERIFY(notificationData.title == "friendName");
|
|
QVERIFY(notificationData.message == "test");
|
|
}
|
|
|
|
void TestNotificationGenerator::testMultipleFriendMessages()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friendName");
|
|
notificationGenerator->friendMessageNotification(&f, "test");
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test2");
|
|
QVERIFY(notificationData.title == "2 message(s) from friendName");
|
|
QVERIFY(notificationData.message == "test2");
|
|
|
|
notificationData = notificationGenerator->friendMessageNotification(&f, "test3");
|
|
QVERIFY(notificationData.title == "3 message(s) from friendName");
|
|
QVERIFY(notificationData.message == "test3");
|
|
}
|
|
|
|
void TestNotificationGenerator::testNotificationClear()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friendName");
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test");
|
|
|
|
// On notification clear we shouldn't see a notification count from the friend
|
|
notificationGenerator->onNotificationActivated();
|
|
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test2");
|
|
QVERIFY(notificationData.title == "friendName");
|
|
QVERIFY(notificationData.message == "test2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testGroupMessage()
|
|
{
|
|
Group g(0, GroupId(0), "groupName", false, "selfName", *groupQuery, *coreIdHandler);
|
|
auto sender = groupQuery->getGroupPeerPk(0, 0);
|
|
g.updateUsername(sender, "sender1");
|
|
|
|
auto notificationData = notificationGenerator->groupMessageNotification(&g, sender, "test");
|
|
QVERIFY(notificationData.title == "groupName");
|
|
QVERIFY(notificationData.message == "sender1: test");
|
|
}
|
|
|
|
void TestNotificationGenerator::testMultipleGroupMessages()
|
|
{
|
|
Group g(0, GroupId(0), "groupName", false, "selfName", *groupQuery, *coreIdHandler);
|
|
|
|
auto sender = groupQuery->getGroupPeerPk(0, 0);
|
|
g.updateUsername(sender, "sender1");
|
|
|
|
auto sender2 = groupQuery->getGroupPeerPk(0, 1);
|
|
g.updateUsername(sender2, "sender2");
|
|
|
|
notificationGenerator->groupMessageNotification(&g, sender, "test1");
|
|
|
|
auto notificationData = notificationGenerator->groupMessageNotification(&g, sender2, "test2");
|
|
QVERIFY(notificationData.title == "2 message(s) from groupName");
|
|
QVERIFY(notificationData.message == "sender2: test2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testMultipleFriendSourceMessages()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend1");
|
|
|
|
Friend f2(1, ToxPk());
|
|
f2.setName("friend2");
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test1");
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f2, "test2");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from 2 chats");
|
|
QVERIFY(notificationData.message == "friend1, friend2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testMultipleGroupSourceMessages()
|
|
{
|
|
Group g(0, GroupId(QByteArray(32, 0)), "groupName", false, "selfName", *groupQuery, *coreIdHandler);
|
|
Group g2(1, GroupId(QByteArray(32, 1)), "groupName2", false, "selfName", *groupQuery, *coreIdHandler);
|
|
|
|
auto sender = groupQuery->getGroupPeerPk(0, 0);
|
|
g.updateUsername(sender, "sender1");
|
|
|
|
notificationGenerator->groupMessageNotification(&g, sender, "test1");
|
|
auto notificationData = notificationGenerator->groupMessageNotification(&g2, sender, "test1");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from 2 chats");
|
|
QVERIFY(notificationData.message == "groupName, groupName2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testMixedSourceMessages()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
Group g(0, GroupId(QByteArray(32, 0)), "group", false, "selfName", *groupQuery, *coreIdHandler);
|
|
|
|
auto sender = groupQuery->getGroupPeerPk(0, 0);
|
|
g.updateUsername(sender, "sender1");
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test1");
|
|
auto notificationData = notificationGenerator->groupMessageNotification(&g, sender, "test2");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from 2 chats");
|
|
QVERIFY(notificationData.message == "friend, group");
|
|
|
|
notificationData = notificationGenerator->fileTransferNotification(&f, "file", 0);
|
|
QVERIFY(notificationData.title == "3 message(s) from 2 chats");
|
|
QVERIFY(notificationData.message == "friend, group");
|
|
}
|
|
|
|
void TestNotificationGenerator::testFileTransfer()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
auto notificationData = notificationGenerator->fileTransferNotification(&f, "file", 5 * 1024 * 1024 /* 5MB */);
|
|
|
|
QVERIFY(notificationData.title == "friend - file transfer");
|
|
QVERIFY(notificationData.message == "file (5.00MiB)");
|
|
}
|
|
|
|
void TestNotificationGenerator::testFileTransferAfterMessage()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test1");
|
|
auto notificationData = notificationGenerator->fileTransferNotification(&f, "file", 5 * 1024 * 1024 /* 5MB */);
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from friend");
|
|
QVERIFY(notificationData.message == "Incoming file transfer");
|
|
}
|
|
|
|
void TestNotificationGenerator::testGroupInvitation()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
auto notificationData = notificationGenerator->groupInvitationNotification(&f);
|
|
|
|
QVERIFY(notificationData.title == "friend invites you to join a group.");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testGroupInviteUncounted()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test");
|
|
notificationGenerator->groupInvitationNotification(&f);
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test2");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from friend");
|
|
QVERIFY(notificationData.message == "test2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testFriendRequest()
|
|
{
|
|
ToxPk sender(QByteArray(32, 0));
|
|
|
|
auto notificationData = notificationGenerator->friendRequestNotification(sender, "request");
|
|
|
|
QVERIFY(notificationData.title == "Friend request received from 0000000000000000000000000000000000000000000000000000000000000000");
|
|
QVERIFY(notificationData.message == "request");
|
|
}
|
|
|
|
void TestNotificationGenerator::testFriendRequestUncounted()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
ToxPk sender(QByteArray(32, 0));
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test");
|
|
notificationGenerator->friendRequestNotification(sender, "request");
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test2");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from friend");
|
|
QVERIFY(notificationData.message == "test2");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleFriendMessage()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test");
|
|
|
|
QVERIFY(notificationData.title == "New message");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleFileTransfer()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
|
|
auto notificationData = notificationGenerator->fileTransferNotification(&f, "file", 0);
|
|
|
|
QVERIFY(notificationData.title == "Incoming file transfer");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleGroupMessage()
|
|
{
|
|
Group g(0, GroupId(0), "groupName", false, "selfName", *groupQuery, *coreIdHandler);
|
|
auto sender = groupQuery->getGroupPeerPk(0, 0);
|
|
g.updateUsername(sender, "sender1");
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
|
|
auto notificationData = notificationGenerator->groupMessageNotification(&g, sender, "test");
|
|
QVERIFY(notificationData.title == "New group message");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleFriendRequest()
|
|
{
|
|
ToxPk sender(QByteArray(32, 0));
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
|
|
auto notificationData = notificationGenerator->friendRequestNotification(sender, "request");
|
|
|
|
QVERIFY(notificationData.title == "Friend request received");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleGroupInvite()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
auto notificationData = notificationGenerator->groupInvitationNotification(&f);
|
|
|
|
QVERIFY(notificationData.title == "Group invite received");
|
|
QVERIFY(notificationData.message == "");
|
|
}
|
|
|
|
void TestNotificationGenerator::testSimpleMessageToggle()
|
|
{
|
|
Friend f(0, ToxPk());
|
|
f.setName("friend");
|
|
|
|
notificationSettings->setNotifyHide(true);
|
|
|
|
notificationGenerator->friendMessageNotification(&f, "test");
|
|
|
|
notificationSettings->setNotifyHide(false);
|
|
|
|
auto notificationData = notificationGenerator->friendMessageNotification(&f, "test2");
|
|
|
|
QVERIFY(notificationData.title == "2 message(s) from friend");
|
|
QVERIFY(notificationData.message == "test2");
|
|
}
|
|
|
|
QTEST_GUILESS_MAIN(TestNotificationGenerator)
|
|
#include "notificationgenerator_test.moc"
|