amnezia-client/client/tests/testComplexOperations.cpp

#include <QJsonDocument>
#include <QJsonObject>
#include <QUuid>
#include <QSignalSpy>
#include <QTest>

#include "core/controllers/coreController.h"
#include "core/models/serverConfig.h"
#include "vpnConnection.h"
#include "secureQSettings.h"

using namespace amnezia;

class TestComplexOperations : public QObject
{
    Q_OBJECT

private:
    CoreController* m_coreController;
    SecureQSettings* m_settings;

    QString getValueFromIni(const QString &key)
    {
        QSettings settings("test_vars.ini", QSettings::IniFormat);
        return settings.value(key).toString();
    }

private slots:
    void initTestCase() {
        QString testOrg = "AmneziaVPN-Test-" + QUuid::createUuid().toString();
        m_settings = new SecureQSettings(testOrg, "amnezia-client", nullptr, false);
        
        auto vpnConnection = QSharedPointer<VpnConnection>::create(nullptr, nullptr);
        m_coreController = new CoreController(vpnConnection, m_settings, nullptr, this);
    }

    void cleanupTestCase() {
        m_settings->clearSettings();
        delete m_coreController;
        delete m_settings;
    }

    void init() {
        m_settings->clearSettings();
        if (m_coreController->m_serversModel) {
            m_coreController->m_serversModel->updateModel(QVector<ServerConfig>(), -1, false);
        }
    }

    void testComplexOperationSequence() {
        QString awgKey = getValueFromIni("configs/TEST_CONFIG_AWG");
        QString xrayKey = getValueFromIni("configs/TEST_CONFIG_WG");
        QString wgKey = getValueFromIni("configs/TEST_CONFIG_XRAY");

        QSignalSpy importFinishedSpy(m_coreController->m_importCoreController, &ImportController::importFinished);
        QSignalSpy serverAddedSpy(m_coreController->m_serversRepository, &SecureServersRepository::serverAdded);
        QSignalSpy serverEditedSpy(m_coreController->m_serversRepository, &SecureServersRepository::serverEdited);
        QSignalSpy serverRemovedSpy(m_coreController->m_serversRepository, &SecureServersRepository::serverRemoved);
        QSignalSpy defaultServerChangedSpy(m_coreController->m_serversRepository, &SecureServersRepository::defaultServerChanged);

        auto importResult1 = m_coreController->m_importCoreController->extractConfigFromData(awgKey);
        m_coreController->m_importCoreController->importConfig(importResult1.config);
        auto importResult2 = m_coreController->m_importCoreController->extractConfigFromData(xrayKey);
        m_coreController->m_importCoreController->importConfig(importResult2.config);
        auto importResult3 = m_coreController->m_importCoreController->extractConfigFromData(wgKey);
        m_coreController->m_importCoreController->importConfig(importResult3.config);

        QVERIFY2(importFinishedSpy.count() == 3, "importFinished should be emitted 3 times");
        QVERIFY2(serverAddedSpy.count() == 3, "serverAdded should be emitted 3 times");
        QVERIFY2(defaultServerChangedSpy.count() == 2, "defaultServerChanged should be emitted 2 times (0->1, 1->2, first import doesn't emit as default is already 0)");
        QVERIFY2(m_coreController->m_serversRepository->serversCount() == 3, "Should have 3 servers");
        QVERIFY2(m_coreController->m_serversRepository->defaultServerIndex() == 2, "Default should be index 2");

        ServerConfig server0 = m_coreController->m_serversController->getServerConfig(0);
        server0.visit([](auto& arg) {
            arg.description = "Edited First Server";
        });
        m_coreController->m_serversController->editServer(0, server0);

        QVERIFY2(serverEditedSpy.count() == 1, "serverEdited should be emitted");
        QString editedDesc0 = m_coreController->m_serversRepository->server(0).description();
        QVERIFY2(editedDesc0 == "Edited First Server", "First server should be edited");

        m_coreController->m_serversController->removeServer(1);

        QVERIFY2(serverRemovedSpy.count() == 1, "serverRemoved should be emitted");
        QVERIFY2(m_coreController->m_serversRepository->serversCount() == 2, "Should have 2 servers");
        QVERIFY2(m_coreController->m_serversRepository->defaultServerIndex() == 1, "Default should be index 1 (was 2, removed 1)");

        m_coreController->m_serversController->setDefaultServerIndex(0);

        QVERIFY2(defaultServerChangedSpy.count() == 4, "defaultServerChanged should be emitted again");
        QVERIFY2(m_coreController->m_serversRepository->defaultServerIndex() == 0, "Default should be index 0");

        ServerConfig server0After = m_coreController->m_serversController->getServerConfig(0);
        server0After.visit([](auto& arg) {
            arg.description = "Final Edited Server";
        });
        m_coreController->m_serversController->editServer(0, server0After);

        QVERIFY2(serverEditedSpy.count() == 2, "serverEdited should be emitted again");
        QString finalDesc0 = m_coreController->m_serversRepository->server(0).description();
        QVERIFY2(finalDesc0 == "Final Edited Server", "First server should be edited again");

        QVERIFY2(m_coreController->m_serversRepository->serversCount() == 2, "Final servers count should be 2");
        QVERIFY2(m_coreController->m_serversRepository->defaultServerIndex() == 0, "Final default index should be 0");

        if (m_coreController->m_serversModel) {
            QVERIFY2(m_coreController->m_serversModel->rowCount() == 2, "Model should have 2 rows");
            QString modelDesc0 = m_coreController->m_serversModel->data(m_coreController->m_serversModel->index(0, 0), ServersModel::NameRole).toString();
            QVERIFY2(modelDesc0 == "Final Edited Server", "Model should reflect final edited name");
        }
    }
};

QTEST_MAIN(TestComplexOperations)
#include "testComplexOperations.moc"