// SPDX-FileCopyrightText: 2021 Carl Schwan <carlschwan@kde.org>
//
// SPDX-License-Identifier: LGPL-2.1-or-later

#ifdef Quotient_E2EE_ENABLED
#include "testgroupsession.h"
#include "olm/qolminboundsession.h"
#include "olm/qolmoutboundsession.h"
#include "olm/utils.h"

using namespace Quotient;

void TestOlmSession::groupSessionPicklingValid()
{
    auto ogs = QOlmOutboundGroupSession::create();
    const auto ogsId = std::get<QByteArray>(ogs->sessionId());
    QVERIFY(QByteArray::fromBase64Encoding(ogsId).decodingStatus == QByteArray::Base64DecodingStatus::Ok);
    QCOMPARE(0, ogs->sessionMessageIndex());

    auto ogsPickled = std::get<QByteArray>(ogs->pickle(Unencrypted {}));
    auto ogs2 = std::get<std::unique_ptr<QOlmOutboundGroupSession>>(QOlmOutboundGroupSession::unpickle(ogsPickled, Unencrypted {}));
    QCOMPARE(ogsId, std::get<QByteArray>(ogs2->sessionId()));

    auto igs = QOlmInboundGroupSession::create(std::get<QByteArray>(ogs->sessionKey()));
    const auto igsId = igs->sessionId();
    // ID is valid base64?
    QVERIFY(QByteArray::fromBase64Encoding(igsId).decodingStatus == QByteArray::Base64DecodingStatus::Ok);

    //// no messages have been sent yet
    QCOMPARE(0, igs->firstKnownIndex());

    auto igsPickled = igs->pickle(Unencrypted {});
    igs = std::get<std::unique_ptr<QOlmInboundGroupSession>>(QOlmInboundGroupSession::unpickle(igsPickled, Unencrypted {}));
    QCOMPARE(igsId, igs->sessionId());
}

QTEST_MAIN(TestOlmSession)
#endif