Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Improve QgsNetworkAccessManager tests
  • Loading branch information
nyalldawson committed Jan 28, 2019
1 parent 2e3a717 commit c707df7
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 11 deletions.
3 changes: 2 additions & 1 deletion src/core/qgsnetworkaccessmanager.cpp
Expand Up @@ -245,7 +245,8 @@ QNetworkReply *QgsNetworkAccessManager::createRequest( QNetworkAccessManager::Op
timer->setObjectName( QStringLiteral( "timeoutTimer" ) );
connect( timer, &QTimer::timeout, this, &QgsNetworkAccessManager::abortRequest );
timer->setSingleShot( true );
timer->start( s.value( QStringLiteral( "/qgis/networkAndProxy/networkTimeout" ), QStringLiteral( "60000" ) ).toInt() );
const int timeout = s.value( QStringLiteral( "/qgis/networkAndProxy/networkTimeout" ), QStringLiteral( "60000" ) ).toInt();
timer->start( timeout );

connect( reply, &QNetworkReply::downloadProgress, timer, [timer] { timer->start(); } );
connect( reply, &QNetworkReply::uploadProgress, timer, [timer] { timer->start(); } );
Expand Down
58 changes: 48 additions & 10 deletions tests/src/core/testqgsnetworkaccessmanager.cpp
Expand Up @@ -30,17 +30,31 @@ class BackgroundRequest : public QThread
Q_OBJECT

public:
BackgroundRequest( const QNetworkRequest &request )
BackgroundRequest( const QNetworkRequest &request, QNetworkAccessManager::Operation op = QNetworkAccessManager::GetOperation, const QByteArray &data = QByteArray() )
: mRequest( request )
{
moveToThread( this );
connect( this, &QThread::started, this, [ = ]
{
QVERIFY( QThread::currentThread() != QCoreApplication::instance()->thread() );
QgsNetworkAccessManager::instance()->get( mRequest );
switch ( op )
{
case QNetworkAccessManager::GetOperation:
mReply = QgsNetworkAccessManager::instance()->get( mRequest );
break;

case QNetworkAccessManager::PostOperation:
mReply = QgsNetworkAccessManager::instance()->post( mRequest, data );
break;

default:
break;
}

} );
}

QNetworkReply *mReply = nullptr;
private:
QNetworkRequest mRequest;
};
Expand Down Expand Up @@ -72,6 +86,7 @@ void TestQgsNetworkAccessManager::initTestCase()
QCoreApplication::setOrganizationDomain( QStringLiteral( "qgis.org" ) );
QCoreApplication::setApplicationName( QStringLiteral( "QGIS-TEST" ) );

QgsSettings().setValue( QStringLiteral( "/qgis/networkAndProxy/networkTimeout" ), 1000 );
QgsApplication::init();
QgsApplication::initQgis();
}
Expand Down Expand Up @@ -251,13 +266,13 @@ void TestQgsNetworkAccessManager::fetchPost()
bool loaded = false;
bool gotRequestAboutToBeCreatedSignal = false;
int requestId = -1;
QUrl u = QUrl::fromLocalFile( QStringLiteral( TEST_DATA_DIR ) + '/' + "encoded_html.html" );
QUrl u = QUrl( QStringLiteral( "http://httpbin.org/post" ) );
connect( QgsNetworkAccessManager::instance(), qgis::overload< QgsNetworkRequestParameters >::of( &QgsNetworkAccessManager::requestAboutToBeCreated ), &context, [&]( const QgsNetworkRequestParameters & params )
{
gotRequestAboutToBeCreatedSignal = true;
requestId = params.requestId();
QVERIFY( requestId > 0 );
QCOMPARE( params.operation(), QNetworkAccessManager::GetOperation );
QCOMPARE( params.operation(), QNetworkAccessManager::PostOperation );
QCOMPARE( params.request().url(), u );
QCOMPARE( params.content(), QByteArray( "a=b&c=d" ) );
} );
Expand All @@ -269,14 +284,39 @@ void TestQgsNetworkAccessManager::fetchPost()
QCOMPARE( reply.request().url(), u );
loaded = true;
} );
QgsNetworkAccessManager::instance()->post( QNetworkRequest( u ), QByteArray( "a=b&c=d" ) );
QNetworkRequest req( u );
req.setHeader( QNetworkRequest::ContentTypeHeader, QStringLiteral( "application/x-www-form-urlencoded" ) );
QNetworkReply *reply = QgsNetworkAccessManager::instance()->post( req, QByteArray( "a=b&c=d" ) );

while ( !loaded )
{
qApp->processEvents();
}

QVERIFY( gotRequestAboutToBeCreatedSignal );
QString replyContent = reply->readAll();
QVERIFY( replyContent.contains( QStringLiteral( "\"a\": \"b\"" ) ) );
QVERIFY( replyContent.contains( QStringLiteral( "\"c\": \"d\"" ) ) );

gotRequestAboutToBeCreatedSignal = false;
loaded = false;
req = QNetworkRequest( u );
req.setHeader( QNetworkRequest::ContentTypeHeader, QStringLiteral( "application/x-www-form-urlencoded" ) );
BackgroundRequest *thread = new BackgroundRequest( req, QNetworkAccessManager::PostOperation, QByteArray( "a=b&c=d" ) );

thread->start();

while ( !loaded )
{
qApp->processEvents();
}
QVERIFY( gotRequestAboutToBeCreatedSignal );
replyContent = thread->mReply->readAll();
QVERIFY( replyContent.contains( QStringLiteral( "\"a\": \"b\"" ) ) );
QVERIFY( replyContent.contains( QStringLiteral( "\"c\": \"d\"" ) ) );
thread->exit();
thread->wait();
thread->deleteLater();
}

void TestQgsNetworkAccessManager::fetchBadSsl()
Expand Down Expand Up @@ -328,15 +368,13 @@ void TestQgsNetworkAccessManager::fetchTimeout()
if ( QgsTest::isTravis() )
QSKIP( "This test is disabled on Travis CI environment" );

QgsSettings().setValue( QStringLiteral( "/qgis/networkAndProxy/networkTimeout" ), 5000 );

QObject context;
//test fetching from a blank url
bool gotRequestAboutToBeCreatedSignal = false;
bool gotTimeoutError = false;
bool finished = false;
int requestId = -1;
QUrl u = QUrl( QStringLiteral( "http://10.255.255.1" ) );
QUrl u = QUrl( QStringLiteral( "http://httpbin.org/delay/10" ) );
connect( QgsNetworkAccessManager::instance(), qgis::overload< QgsNetworkRequestParameters >::of( &QgsNetworkAccessManager::requestAboutToBeCreated ), &context, [&]( const QgsNetworkRequestParameters & params )
{
gotRequestAboutToBeCreatedSignal = true;
Expand All @@ -356,7 +394,7 @@ void TestQgsNetworkAccessManager::fetchTimeout()
} );
QgsNetworkAccessManager::instance()->get( QNetworkRequest( u ) );

while ( !gotTimeoutError && !finished )
while ( !gotTimeoutError )
{
qApp->processEvents();
}
Expand All @@ -372,7 +410,7 @@ void TestQgsNetworkAccessManager::fetchTimeout()

thread->start();

while ( !gotTimeoutError && !finished )
while ( !gotTimeoutError )
{
qApp->processEvents();
}
Expand Down

0 comments on commit c707df7

Please sign in to comment.