mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-07-23 19:51:05 -04:00
77 lines
2.5 KiB
C++
77 lines
2.5 KiB
C++
// Copyright (C) 2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
|
|
|
#include "ChatFileStoreTest.hpp"
|
|
|
|
#include <QDir>
|
|
#include <QFile>
|
|
#include <QJsonArray>
|
|
#include <QJsonObject>
|
|
#include <QTemporaryDir>
|
|
#include <QTest>
|
|
|
|
#include "ChatView/ChatFileStore.hpp"
|
|
#include "acp/AgentBinding.hpp"
|
|
#include "session/HistorySerializer.hpp"
|
|
#include "session/Session.hpp"
|
|
|
|
namespace QodeAssist {
|
|
|
|
void ChatFileStoreTest::testChatFileStoreRoundTripsStoredContent()
|
|
{
|
|
QTemporaryDir dir;
|
|
QVERIFY(dir.isValid());
|
|
|
|
const QString chatFilePath = dir.filePath("chat.json");
|
|
const QByteArray original = QByteArrayLiteral("int main() { return 0; }\n");
|
|
|
|
QString storedPath;
|
|
QVERIFY(Chat::ChatFileStore::saveContentToStorage(
|
|
chatFilePath, "main.cpp", QString::fromLatin1(original.toBase64()), storedPath));
|
|
QVERIFY(!storedPath.isEmpty());
|
|
|
|
QCOMPARE(Chat::ChatFileStore::loadRawContentFromStorage(chatFilePath, storedPath), original);
|
|
QCOMPARE(
|
|
Chat::ChatFileStore::loadContentFromStorage(chatFilePath, storedPath),
|
|
QString::fromLatin1(original.toBase64()));
|
|
|
|
const QString contentFolder = Chat::ChatFileStore::getChatContentFolder(chatFilePath);
|
|
QVERIFY(QFileInfo(QDir(contentFolder).filePath(storedPath)).exists());
|
|
}
|
|
|
|
void ChatFileStoreTest::testLegacyChatFileLoadsThroughTheFileStore()
|
|
{
|
|
QTemporaryDir dir;
|
|
QVERIFY(dir.isValid());
|
|
|
|
QJsonObject root;
|
|
root["version"] = "0.2";
|
|
root["messages"] = QJsonArray{
|
|
QJsonObject{{"role", 1}, {"content", "explain this"}, {"id", "u1"}},
|
|
QJsonObject{{"role", 2}, {"content", "here is the answer"}, {"id", "r1"}}};
|
|
|
|
const QString filePath = dir.filePath("legacy.json");
|
|
QFile file(filePath);
|
|
QVERIFY(file.open(QIODevice::WriteOnly));
|
|
file.write(QJsonDocument(root).toJson());
|
|
file.close();
|
|
|
|
Session::ConversationHistory history;
|
|
Acp::AgentBinding binding;
|
|
const Chat::SerializationResult result
|
|
= Chat::ChatFileStore::loadFromFile(history, binding, filePath);
|
|
|
|
QVERIFY(result.success);
|
|
QVERIFY(result.warningMessage.isEmpty());
|
|
QVERIFY(binding.isEmpty());
|
|
QCOMPARE(history.size(), 2);
|
|
QCOMPARE(history.at(0).role, Session::MessageRole::User);
|
|
QCOMPARE(
|
|
std::get<Session::TextBlock>(history.at(0).blocks.at(0)),
|
|
(Session::TextBlock{"explain this"}));
|
|
QCOMPARE(history.at(1).role, Session::MessageRole::Assistant);
|
|
}
|
|
|
|
} // namespace QodeAssist
|