Differential D6879 Diff 17454 src/kmailtransport/plugins/smtp/autotests/fakeserver.cpp

Changeset View

Standalone View

View Options

src/kmailtransport/plugins/smtp/autotests/fakeserver.cpp

  • This file was added.
1​/*
2​ Copyright 2010 BetterInbox <contact@betterinbox.com>
3​ Author: Christophe Laveault <christophe@betterinbox.com>
4​ Gregory Schlomoff <gregory.schlomoff@gmail.com>
5​
6​ This library is free software; you can redistribute it and/or
7​ modify it under the terms of the GNU Lesser General Public
8​ License as published by the Free Software Foundation; either
9​ version 2.1 of the License, or (at your option) any later version.
10​
11​ This library is distributed in the hope that it will be useful,
12​ but WITHOUT ANY WARRANTY; without even the implied warranty of
13​ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14​ Lesser General Public License for more details.
15​
16​ You should have received a copy of the GNU Lesser General Public
17​ License along with this library. If not, see <http://www.gnu.org/licenses/>.
18​*/
19​
20​#include "fakeserver.h"
21​
22​#include <QTest>
23​#include <QFile>
24​#include <QDebug>
25​
26​FakeServer::FakeServer(QObject *parent) :
27​ QThread(parent),
28​ m_tcpServer(nullptr)
mlaurentUnsubmitted
Not Done

initialize m_tcpServer to nullptr

mlaurent: initialize m_tcpServer to nullptr
29​{
30​ moveToThread(this);
31​}
32​
33​QByteArray FakeServer::greeting()
34​{
35​ return "S: 220 localhost ESMTP xx777xx";
36​}
37​
38​QList<QByteArray> FakeServer::greetingAndEhlo(bool multiline)
39​{
40​ return QList<QByteArray>() << greeting()
41​ << "C: EHLO 127.0.0.1"
42​ << QByteArray("S: 250") + (multiline ? '-' : ' ') + "Localhost ready to roll";
43​}
44​
45​QList<QByteArray> FakeServer::bye()
46​{
47​ return { "C: QUIT",
48​ "S: 221 So long, and thanks for all the fish",
49​ "X: "
50​ };
51​}
52​
53​FakeServer::~FakeServer()
54​{
55​ quit();
56​ wait();
57​}
58​
59​void FakeServer::startAndWait()
60​{
61​ start();
62​ // this will block until the event queue starts
63​ QMetaObject::invokeMethod(this, "started", Qt::BlockingQueuedConnection);
64​}
65​
66​void FakeServer::dataAvailable()
67​{
68​ QMutexLocker locker(&m_mutex);
69​
70​ QTcpSocket *socket = qobject_cast<QTcpSocket *>(sender());
71​ Q_ASSERT(socket != nullptr);
72​
73​ int scenarioNumber = m_clientSockets.indexOf(socket);
74​
75​ QVERIFY(!m_scenarios[scenarioNumber].isEmpty());
76​
77​ readClientPart(scenarioNumber);
78​ writeServerPart(scenarioNumber);
79​}
80​
81​void FakeServer::newConnection()
82​{
83​ QMutexLocker locker(&m_mutex);
84​
85​ m_clientSockets << m_tcpServer->nextPendingConnection();
86​ connect(m_clientSockets.last(), SIGNAL(readyRead()), this, SLOT(dataAvailable()));
87​ //m_clientParsers << new KIMAP::ImapStreamParser( m_clientSockets.last(), true );
88​
89​ QVERIFY(m_clientSockets.size() <= m_scenarios.size());
90​
91​ writeServerPart(m_clientSockets.size() - 1);
92​}
93​
94​void FakeServer::run()
95​{
96​ m_tcpServer = new QTcpServer();
97​ if (!m_tcpServer->listen(QHostAddress(QHostAddress::LocalHost), 5989)) {
98​ qFatal("Unable to start the server");
99​ return;
100​ }
101​
102​ connect(m_tcpServer, SIGNAL(newConnection()), this, SLOT(newConnection()));
103​
104​ exec();
105​
106​ qDeleteAll(m_clientSockets);
107​
108​ delete m_tcpServer;
109​}
110​
111​void FakeServer::started()
112​{
113​ // do nothing: this is a dummy slot used by startAndWait()
114​}
115​
116​void FakeServer::setScenario(const QList<QByteArray> &scenario)
117​{
118​ QMutexLocker locker(&m_mutex);
119​
120​ m_scenarios.clear();
121​ m_scenarios << scenario;
122​}
123​
124​void FakeServer::addScenario(const QList<QByteArray> &scenario)
125​{
126​ QMutexLocker locker(&m_mutex);
127​
128​ m_scenarios << scenario;
129​}
130​
131​void FakeServer::addScenarioFromFile(const QString &fileName)
132​{
133​ QFile file(fileName);
134​ file.open(QFile::ReadOnly);
135​
136​ QList<QByteArray> scenario;
137​
138​ // When loading from files we never have the authentication phase
139​ // force jumping directly to authenticated state.
140​ //scenario << preauth();
141​
142​ while (!file.atEnd()) {
143​ scenario << file.readLine().trimmed();
144​ }
145​
146​ file.close();
147​
148​ addScenario(scenario);
149​}
150​
151​bool FakeServer::isScenarioDone(int scenarioNumber) const
152​{
153​ QMutexLocker locker(&m_mutex);
154​
155​ if (scenarioNumber < m_scenarios.size()) {
156​ return m_scenarios[scenarioNumber].isEmpty();
157​ } else {
158​ return true; // Non existent hence empty, right?
159​ }
160​}
161​
162​bool FakeServer::isAllScenarioDone() const
163​{
164​ QMutexLocker locker(&m_mutex);
165​
166​ for (const auto &scenario : qAsConst(m_scenarios)) {
mlaurentUnsubmitted
Not Done

for(...: qAsConst(...))

mlaurent: for(...: qAsConst(...))
167​ if (!scenario.isEmpty()) {
168​ qDebug() << scenario;
169​ return false;
170​ }
171​ }
172​
173​ return true;
174​}
175​
176​void FakeServer::writeServerPart(int scenarioNumber)
177​{
178​ QList<QByteArray> scenario = m_scenarios[scenarioNumber];
179​ QTcpSocket *clientSocket = m_clientSockets[scenarioNumber];
180​
181​ while (!scenario.isEmpty()
182​ && (scenario.first().startsWith("S: ") || scenario.first().startsWith("W: "))) {
183​ QByteArray rule = scenario.takeFirst();
184​
185​ if (rule.startsWith("S: ")) {
186​ QByteArray payload = rule.mid(3);
187​ clientSocket->write(payload + "\r\n");
188​ } else {
189​ int timeout = rule.mid(3).toInt();
190​ QTest::qWait(timeout);
191​ }
192​ }
193​
194​ if (!scenario.isEmpty() && scenario.first().startsWith('X')) {
195​ scenario.takeFirst();
196​ clientSocket->close();
197​ }
198​
199​ if (!scenario.isEmpty()) {
200​ QVERIFY(scenario.first().startsWith("C: "));
201​ }
202​
203​ m_scenarios[scenarioNumber] = scenario;
204​}
205​
206​void FakeServer::readClientPart(int scenarioNumber)
207​{
208​ QList<QByteArray> scenario = m_scenarios[scenarioNumber];
209​ QTcpSocket *clientSocket = m_clientSockets[scenarioNumber];
210​
211​ while (!scenario.isEmpty() && scenario.first().startsWith("C: ")) {
212​ QByteArray line = clientSocket->readLine();
213​ QByteArray received = "C: " + line.trimmed();
214​ QByteArray expected = scenario.takeFirst();
215​
216​ if (expected == "C: SKIP" && !scenario.isEmpty()) {
217​ expected = scenario.takeFirst();
218​ while (received != expected) {
219​ received = "C: " + clientSocket->readLine().trimmed();
220​ }
221​ }
222​
223​ QCOMPARE(QString::fromUtf8(received), QString::fromUtf8(expected));
224​ QCOMPARE(received, expected);
225​ }
226​
227​ if (!scenario.isEmpty()) {
228​ QVERIFY(scenario.first().startsWith("S: "));
229​ }
230​
231​ m_scenarios[scenarioNumber] = scenario;
232​}