rint("Hello, World!")
+
+if __name__ == "__main__":
+ main()
+
+def main():
+ p)");
+
+ ASSERT_EQ(spy.count(), 1);
+ auto message = spy.takeFirst().at(0).value();
+ QJsonObject response = message.toJsonObject();
+
+ EXPECT_EQ(response["id"].toString(), "completion-1");
+ EXPECT_TRUE(response.contains("result"));
+
+ QJsonObject result = response["result"].toObject();
+ EXPECT_TRUE(result.contains("completions"));
+ EXPECT_FALSE(result["isIncomplete"].toBool());
+
+ QJsonArray completions = result["completions"].toArray();
+ ASSERT_EQ(completions.size(), 1);
+ EXPECT_EQ(completions[0].toObject()["text"].toString(), "test completion");
+}
+
+TEST_F(LLMClientInterfaceTest, completionChat)
+{
+ ON_CALL(m_promptProvider, getTemplateByName(_)).WillByDefault(Return(m_chatTemplate.get()));
+
+ m_documentReader.setDocumentInfo(
+ R"(
+def main():
+ print("Hello, World!")
+
+if __name__ == "__main__":
+ main()
+)",
+ "/path/to/file.py",
+ "text/python");
+
+ m_completeSettings.smartProcessInstuctText.setValue(true);
+
+ m_requestHandler->setFakeCompletion(
+ "Here's the code: ```cpp\nint main() {\n return 0;\n}\n```");
+
+ QSignalSpy spy(m_client.get(), &LanguageClient::BaseClientInterface::messageReceived);
+
+ QJsonObject request = createCompletionRequest();
+ m_client->sendData(QJsonDocument(request).toJson());
+
+ ASSERT_EQ(m_requestHandler->receivedRequests().size(), 1);
+
+ QJsonObject requestJson = m_requestHandler->receivedRequests().at(0).providerRequest;
+ auto messagesJson = requestJson["messages"].toArray();
+ ASSERT_EQ(messagesJson.size(), 1);
+ ASSERT_EQ(messagesJson.at(0).toObject()["content"].toString(), R"(user message template prefix:
+
+def main():
+ p
+suffix:
+rint("Hello, World!")
+
+if __name__ == "__main__":
+ main()
+
+)");
+
+ ASSERT_EQ(spy.count(), 1);
+ auto message = spy.takeFirst().at(0).value();
+ QJsonObject response = message.toJsonObject();
+
+ QJsonArray completions = response["result"].toObject()["completions"].toArray();
+ ASSERT_EQ(completions.size(), 1);
+
+ QString processedText = completions[0].toObject()["text"].toString();
+ EXPECT_TRUE(processedText.contains("# Here's the code:"));
+ EXPECT_TRUE(processedText.contains("int main()"));
+}
+
+TEST_F(LLMClientInterfaceTest, cancelRequest)
+{
+ QSignalSpy cancelSpy(m_requestHandler.get(), &LLMCore::RequestHandlerBase::requestCancelled);
+
+ QJsonObject cancelRequest = createCancelRequest("completion-1");
+ m_client->sendData(QJsonDocument(cancelRequest).toJson());
+
+ ASSERT_EQ(cancelSpy.count(), 1);
+ EXPECT_EQ(cancelSpy.takeFirst().at(0).toString(), "completion-1");
+}
+
+TEST_F(LLMClientInterfaceTest, ServerDeviceTemplate)
+{
+ EXPECT_EQ(m_client->serverDeviceTemplate().toFSPathString(), "Qode Assist");
+}
+
+} // namespace QodeAssist
diff --git a/test/MockDocumentReader.hpp b/test/MockDocumentReader.hpp
new file mode 100644
index 0000000..e3dc595
--- /dev/null
+++ b/test/MockDocumentReader.hpp
@@ -0,0 +1,53 @@
+/*
+ * Copyright (C) 2025 Povilas Kanapickas
+ *
+ * This file is part of QodeAssist.
+ *
+ * QodeAssist is free 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.
+ *
+ * QodeAssist 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 QodeAssist. If not, see .
+ */
+
+#pragma once
+
+#include "context/IDocumentReader.hpp"
+#include
+#include
+
+namespace QodeAssist {
+
+class MockDocumentReader : public Context::IDocumentReader
+{
+public:
+ MockDocumentReader() = default;
+
+ Context::DocumentInfo readDocument(const QString &path) const override
+ {
+ return m_documentInfo;
+ }
+
+ void setDocumentInfo(const QString &text, const QString &filePath, const QString &mimeType)
+ {
+ m_document = std::make_unique(text);
+ m_documentInfo.document = m_document.get();
+ m_documentInfo.filePath = filePath;
+ m_documentInfo.mimeType = mimeType;
+ }
+
+ ~MockDocumentReader() = default;
+
+private:
+ Context::DocumentInfo m_documentInfo;
+ std::unique_ptr m_document;
+};
+
+} // namespace QodeAssist
diff --git a/test/MockRequestHandler.hpp b/test/MockRequestHandler.hpp
new file mode 100644
index 0000000..94522cb
--- /dev/null
+++ b/test/MockRequestHandler.hpp
@@ -0,0 +1,59 @@
+/*
+ * Copyright (C) 2025 Povilas Kanapickas
+ *
+ * This file is part of QodeAssist.
+ *
+ * QodeAssist is free 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.
+ *
+ * QodeAssist 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 QodeAssist. If not, see .
+ */
+
+#pragma once
+
+#include
+
+namespace QodeAssist::LLMCore {
+
+class MockRequestHandler : public RequestHandlerBase
+{
+public:
+ explicit MockRequestHandler(QObject *parent = nullptr)
+ : RequestHandlerBase(parent)
+ , m_fakeCompletion("")
+ {}
+
+ void setFakeCompletion(const QString &completion) { m_fakeCompletion = completion; }
+
+ void sendLLMRequest(const LLMConfig &config, const QJsonObject &request) override
+ {
+ m_receivedRequests.append(config);
+
+ emit completionReceived(m_fakeCompletion, request, true);
+
+ QString requestId = request["id"].toString();
+ emit requestFinished(requestId, true, QString());
+ }
+
+ bool cancelRequest(const QString &id) override
+ {
+ emit requestCancelled(id);
+ return true;
+ }
+
+ const QVector &receivedRequests() const { return m_receivedRequests; }
+
+private:
+ QString m_fakeCompletion;
+ QVector m_receivedRequests;
+};
+
+} // namespace QodeAssist::LLMCore