You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
5.2 KiB
74 lines
5.2 KiB
#include "../src/avatars/groupavatarmanager.h" |
|
#include "../src/channelsettingsdialog.h" |
|
#include "../db/db_manager.h" |
|
#include "../transport/gui_bridge.h" |
|
#include <QtTest> |
|
#include <QTemporaryDir> |
|
#include <QPushButton> |
|
extern "C" { |
|
#include "utun_instance.h" |
|
#include "chat/chat_core.h" |
|
#include "chat/chat_event.h" |
|
#include "chat/group_avatar.h" |
|
#include "routing_layer/topo_node_sqlite.h" |
|
#include "mem.h" |
|
#include "debug_config.h" |
|
} |
|
static UTUN_INSTANCE* instance; |
|
extern "C" int gui_bridge_is_uasync_ready(void) { return instance != nullptr; } |
|
extern "C" UTUN_INSTANCE* gui_bridge_get_inst(void) { return instance; } |
|
extern "C" void gui_bridge_post_uasync_fn(void (*fn)(void*), void* arg) { uasync_call_soon(instance->ua, arg, fn); } |
|
static void coreEvent(UTUN_INSTANCE*, int type, const uint8_t* data, int length) { |
|
if (type == CHAT_EVT_GROUP_AVATAR_RESULT && GroupAvatarManager::instance()) GroupAvatarManager::instance()->onResult(data, length); |
|
} |
|
class GroupAvatarTest : public QObject { |
|
Q_OBJECT |
|
private slots: |
|
void preparePublishDisplayPermissionAndClear() { |
|
debug_config_init(); debug_set_level(DEBUG_LEVEL_WARN); debug_set_category_level(DEBUG_CATEGORY_SYS, DEBUG_LEVEL_ERROR); |
|
UASYNC* ua = uasync_create(); QVERIFY(ua); QTemporaryDir directory; QVERIFY(directory.isValid()); |
|
const QByteArray config = "[global]\nmy_private_key=704f2e012c8fa8768130cb0f988a997dccb628372bc5ceccacc78dcbfec5916f\n" |
|
"my_public_key=b3193173def895bd0fcea6f86af077c7d77216f10395275f627ac18242ec0f01\n" |
|
"[chatserver]\ngroup_autoconnect=0\nmember_files_enabled=0\nstorage_autoload=0\nmember_files_path=" + directory.path().toUtf8() + "/files\n"; |
|
instance = utun_instance_create_from_str(ua, config.constData()); QVERIFY(instance); |
|
QCOMPARE(utun_core_start(instance), 0); QCOMPARE(chat_service_start(instance), 0); chat_core_create_channel_auto(instance, "avatars"); |
|
DbManager db; db.setDb(instance->topo_sqlite_db); db.setMyNodeId(instance->node_id); |
|
QCOMPARE(db.getChannels().size(), 1); const QString channel = db.getChannels()[0].channelId; |
|
QTimer poll; connect(&poll, &QTimer::timeout, this, [ua] { uasync_poll(ua, 0); }); poll.start(1); |
|
chat_event_set_handler(instance, coreEvent); |
|
{ |
|
GroupAvatarManager manager(&db); |
|
ChannelSettingsDialog owner(channel, &db); QVERIFY(owner.findChild<QPushButton*>("uploadGroupAvatar")->isEnabled()); |
|
QImage image(512, 400, QImage::Format_RGB32); |
|
for (int y = 0; y < image.height(); y++) for (int x = 0; x < image.width(); x++) |
|
image.setPixel(x, y, qRgb((x * 73 + y * 39) & 255, (x * 19 + y * 81) & 255, (x * 17 + y * 53) & 255)); |
|
const QString path = directory.path() + "/photo.png"; QVERIFY(image.save(path)); |
|
QSignalSpy result(&manager, &GroupAvatarManager::status); manager.upload(channel, path); |
|
QTRY_VERIFY_WITH_TIMEOUT(db.getChannels()[0].avatarVersion > 0, 5000); |
|
QVERIFY(db.getChannelAvatar(channel).size() <= GROUP_AVATAR_MAX); |
|
QTRY_VERIFY_WITH_TIMEOUT(!manager.image(channel).isNull(), 5000); |
|
QVERIFY(manager.image(channel).width() <= 128); QCOMPARE(manager.image(channel).width(), manager.image(channel).height()); |
|
const QByteArray saved = db.getChannelAvatar(channel); |
|
const ChannelRow original = db.getChannels()[0]; |
|
quint64 version = db.getChannels()[0].avatarVersion; QVERIFY(result.size() >= 2); |
|
QSignalSpy changed(&manager, &GroupAvatarManager::changed); |
|
for (int i = 0; i < 10; i++) QVERIFY(!manager.image(channel).isNull()); QCOMPARE(changed.count(), 0); |
|
sqlite3_stmt* st; QCOMPARE(sqlite3_prepare_v2(instance->topo_sqlite_db, |
|
"UPDATE channels SET ed25519_privkey=NULL WHERE channel_id=?", -1, &st, nullptr), SQLITE_OK); |
|
const QByteArray ch = channel.toLatin1(); sqlite3_bind_text(st, 1, ch.constData(), ch.size(), SQLITE_TRANSIENT); |
|
QCOMPARE(sqlite3_step(st), SQLITE_DONE); sqlite3_finalize(st); |
|
ChannelSettingsDialog reader(channel, &db); |
|
QVERIFY(!reader.findChild<QPushButton*>("uploadGroupAvatar")->isEnabled()); QVERIFY(!reader.findChild<QPushButton*>("removeGroupAvatar")->isEnabled()); |
|
manager.remove(channel); QTest::qWait(50); QCOMPARE(db.getChannels()[0].avatarVersion, version); QCOMPARE(db.getChannelAvatar(channel), saved); |
|
QCOMPARE(topo_node_sqlite_channel_put(instance->topo_sqlite_db, ch.constData(), original.name.toUtf8().constData(), original.ownerNodeId, |
|
(const uint8_t*)original.x25519_pubkey.constData(), nullptr, (const uint8_t*)original.ed25519_pubkey.constData(), |
|
(const uint8_t*)original.ed25519_privkey.constData(), (const uint8_t*)original.signature.constData()), 0); |
|
manager.remove(channel); QTRY_VERIFY_WITH_TIMEOUT(db.getChannels()[0].avatarVersion > version, 5000); |
|
QVERIFY(db.getChannelAvatar(channel).isEmpty()); QVERIFY(manager.image(channel).isNull()); |
|
} |
|
poll.stop(); chat_event_set_handler(instance, nullptr); db.setDb(nullptr); |
|
utun_instance_destroy(instance); instance = nullptr; uasync_destroy(ua, 0); |
|
} |
|
}; |
|
QTEST_MAIN(GroupAvatarTest) |
|
#include "test_group_avatars.moc"
|
|
|