106 lines
3.2 KiB
C++
106 lines
3.2 KiB
C++
/*************************************************************************************
|
|
* Copyright (C) 2014 by Albert Vaca Cintora <albertvaka@gmail.com> *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Lesser General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2.1 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library 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 *
|
|
* Lesser General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Lesser General Public *
|
|
* License along with this library; if not, write to the Free Software *
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA *
|
|
*************************************************************************************/
|
|
|
|
#include "../core/backends/lan/downloadjob.h"
|
|
|
|
#include <QTest>
|
|
#include <QTcpServer>
|
|
#include <QTcpSocket>
|
|
#include <QProcess>
|
|
#include <QEventLoop>
|
|
#include <QTimer>
|
|
#include <QHostAddress>
|
|
#include <KJob>
|
|
#include <iostream>
|
|
|
|
class DownloadJobTest : public QObject
|
|
{
|
|
Q_OBJECT
|
|
|
|
private Q_SLOTS:
|
|
void failToConnectShouldDestroyTheJob();
|
|
void closingTheConnectionShouldDestroyTheJob();
|
|
private:
|
|
|
|
void initServer();
|
|
void initDownloadJob();
|
|
void awaitToBeDestroyedOrTimeOut();
|
|
void stopServer();
|
|
|
|
QTimer mTimer;
|
|
QEventLoop mLoop;
|
|
DownloadJob* test;
|
|
QTcpServer *mServer;
|
|
};
|
|
|
|
void DownloadJobTest::initServer()
|
|
{
|
|
mServer = new QTcpServer(this);
|
|
QVERIFY2(mServer->listen(QHostAddress::LocalHost, 8694), "Failed to create local tcp server");
|
|
}
|
|
|
|
void DownloadJobTest::stopServer()
|
|
{
|
|
mServer->close();
|
|
}
|
|
|
|
void DownloadJobTest::initDownloadJob()
|
|
{
|
|
QVariantMap transferInfo;
|
|
transferInfo["port"]= 8694;
|
|
test = new DownloadJob(QHostAddress::LocalHost, transferInfo);
|
|
test->start();
|
|
}
|
|
|
|
void DownloadJobTest::awaitToBeDestroyedOrTimeOut()
|
|
{
|
|
//Either the job is destroyed
|
|
connect(test, &QObject::destroyed, &mLoop, &QEventLoop::quit);
|
|
|
|
//Or we time out
|
|
mTimer.setInterval(2000);
|
|
mTimer.setSingleShot(true);
|
|
connect(&mTimer, &QTimer::timeout, [this]() {
|
|
mLoop.quit();
|
|
QFAIL("Test timed out");
|
|
});
|
|
mTimer.start();
|
|
|
|
//We wait
|
|
mLoop.exec();
|
|
|
|
mTimer.stop();
|
|
}
|
|
|
|
void DownloadJobTest::failToConnectShouldDestroyTheJob()
|
|
{
|
|
initDownloadJob();
|
|
awaitToBeDestroyedOrTimeOut();
|
|
}
|
|
|
|
void DownloadJobTest::closingTheConnectionShouldDestroyTheJob()
|
|
{
|
|
initServer();
|
|
initDownloadJob();
|
|
stopServer();
|
|
awaitToBeDestroyedOrTimeOut();
|
|
}
|
|
|
|
QTEST_GUILESS_MAIN(DownloadJobTest)
|
|
|
|
#include "downloadjobtest.moc"
|