Split integration test file to several files.

2026-08-06 00:43:19 +02:00 · 2018-02-13 20:53:05 +00:00
parent e3a1308a04
commit 70dabc07dc
12 changed files with 1053 additions and 827 deletions
@@ -0,0 +1,177 @@
+/*
+ * Copyright (C) 2004-2016 ZNC, see the NOTICE file for details.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "znctest.h"
+
+#ifndef ZNC_BIN_DIR
+#define ZNC_BIN_DIR ""
+#endif
+
+namespace znc_inttest {
+
+void WriteConfig(QString path) {
+    // clang-format off
+    Process p(ZNC_BIN_DIR "/znc", QStringList() << "--debug"
+                                                << "--datadir" << path
+                                                << "--makeconf");
+    p.ReadUntil("Listen on port");          p.Write("12345");
+    p.ReadUntil("Listen using SSL");        p.Write();
+    p.ReadUntil("IPv6");                    p.Write();
+    p.ReadUntil("Username");                p.Write("user");
+    p.ReadUntil("password");                p.Write("hunter2", false);
+    p.ReadUntil("Confirm");                 p.Write("hunter2", false);
+    p.ReadUntil("Nick [user]");             p.Write();
+    p.ReadUntil("Alternate nick [user_]");  p.Write();
+    p.ReadUntil("Ident [user]");            p.Write();
+    p.ReadUntil("Real name");               p.Write();
+    p.ReadUntil("Bind host");               p.Write();
+    p.ReadUntil("Set up a network?");       p.Write();
+    p.ReadUntil("Name [freenode]");         p.Write("test");
+    p.ReadUntil("Server host (host only)"); p.Write("127.0.0.1");
+    p.ReadUntil("Server uses SSL?");        p.Write();
+    p.ReadUntil("6667");                    p.Write();
+    p.ReadUntil("password");                p.Write();
+    p.ReadUntil("channels");                p.Write();
+    p.ReadUntil("Launch ZNC now?");         p.Write("no");
+    p.ShouldFinishItself();
+    // clang-format on
+}
+
+void ZNCTest::SetUp() {
+    WriteConfig(m_dir.path());
+    ASSERT_TRUE(m_server.listen(QHostAddress::LocalHost, 6667))
+        << m_server.errorString().toStdString();
+}
+
+Socket ZNCTest::ConnectIRCd() {
+    [this] {
+        ASSERT_TRUE(m_server.waitForNewConnection(30000 /* msec */));
+    }();
+    return WrapIO(m_server.nextPendingConnection());
+}
+
+Socket ZNCTest::ConnectClient() {
+    m_clients.emplace_back();
+    QTcpSocket& sock = m_clients.back();
+    sock.connectToHost("127.0.0.1", 12345);
+    [&] {
+        ASSERT_TRUE(sock.waitForConnected())
+            << sock.errorString().toStdString();
+    }();
+    return WrapIO(&sock);
+}
+
+Socket ZNCTest::LoginClient() {
+    auto client = ConnectClient();
+    client.Write("PASS :hunter2");
+    client.Write("NICK nick");
+    client.Write("USER user/test x x :x");
+    return client;
+}
+
+std::unique_ptr<Process> ZNCTest::Run() {
+    return std::unique_ptr<Process>(new Process(
+        ZNC_BIN_DIR "/znc", QStringList() << "--debug"
+                                          << "--datadir" << m_dir.path(),
+        [](QProcess* proc) {
+            proc->setProcessChannelMode(QProcess::ForwardedChannels);
+        }));
+}
+
+std::unique_ptr<QNetworkReply> ZNCTest::HttpGet(QNetworkRequest request) {
+    return HandleHttp(m_network.get(request));
+}
+std::unique_ptr<QNetworkReply> ZNCTest::HttpPost(
+    QNetworkRequest request, QList<QPair<QString, QString>> data) {
+    request.setHeader(QNetworkRequest::ContentTypeHeader,
+                      "application/x-www-form-urlencoded");
+    QUrlQuery q;
+    q.setQueryItems(data);
+    return HandleHttp(m_network.post(request, q.toString().toUtf8()));
+}
+std::unique_ptr<QNetworkReply> ZNCTest::HandleHttp(QNetworkReply* reply) {
+    QEventLoop loop;
+    QObject::connect(reply, &QNetworkReply::finished, [&]() {
+        std::cout << "Got HTTP reply" << std::endl;
+        loop.quit();
+    });
+    QObject::connect(
+        reply,
+        static_cast<void (QNetworkReply::*)(QNetworkReply::NetworkError)>(
+            &QNetworkReply::error),
+        [&](QNetworkReply::NetworkError e) {
+            ADD_FAILURE() << reply->errorString().toStdString();
+        });
+    QTimer::singleShot(30000 /* msec */, &loop, [&]() {
+        ADD_FAILURE() << "connection timeout";
+        loop.quit();
+    });
+    std::cout << "Start HTTP loop.exec()" << std::endl;
+    loop.exec();
+    std::cout << "Finished HTTP loop.exec()" << std::endl;
+    return std::unique_ptr<QNetworkReply>(reply);
+}
+
+void ZNCTest::InstallModule(QString name, QString content) {
+    QDir dir(m_dir.path());
+    ASSERT_TRUE(dir.mkpath("modules"));
+    ASSERT_TRUE(dir.cd("modules"));
+    if (name.endsWith(".cpp")) {
+        // Compile
+        QTemporaryDir srcdir;
+        QFile file(QDir(srcdir.path()).filePath(name));
+        ASSERT_TRUE(file.open(QIODevice::WriteOnly | QIODevice::Text));
+        QTextStream out(&file);
+        out << content;
+        file.close();
+        Process p(
+            ZNC_BIN_DIR "/znc-buildmod", QStringList() << file.fileName(),
+            [&](QProcess* proc) {
+                proc->setWorkingDirectory(dir.absolutePath());
+                proc->setProcessChannelMode(QProcess::ForwardedChannels);
+            });
+        p.ShouldFinishItself();
+    } else if (name.endsWith(".py")) {
+        // Dedent
+        QStringList lines = content.split("\n");
+        int maxoffset = -1;
+        for (const QString& line : lines) {
+            int nonspace = line.indexOf(QRegExp("\\S"));
+            if (nonspace == -1) continue;
+            if (nonspace < maxoffset || maxoffset == -1)
+                maxoffset = nonspace;
+        }
+        if (maxoffset == -1) maxoffset = 0;
+        QFile file(dir.filePath(name));
+        ASSERT_TRUE(file.open(QIODevice::WriteOnly | QIODevice::Text));
+        QTextStream out(&file);
+        for (const QString& line : lines) {
+            // QTextStream::operator<<(const QStringRef &string) was
+            // introduced in Qt 5.6; let's keep minimum required version
+            // less than that for now.
+            out << line.mid(maxoffset) << "\n";
+        }
+    } else {
+        // Write as is
+        QFile file(dir.filePath(name));
+        ASSERT_TRUE(file.open(QIODevice::WriteOnly | QIODevice::Text));
+        QTextStream out(&file);
+        out << content;
+    }
+}
+
+
+}  // namespace znc_inttest