1 files changed, 92 insertions, 0 deletions
diff --git a/tests/auto/network/access/qnetworkreply/tst_qnetworkreply.cpp b/tests/auto/network/access/qnetworkreply/tst_qnetworkreply.cpp
index 0509eb9a77..480eeecb63 100644
--- a/tests/auto/network/access/qnetworkreply/tst_qnetworkreply.cpp
+++ b/tests/auto/network/access/qnetworkreply/tst_qnetworkreply.cpp
@@ -461,6 +461,8 @@ private Q_SLOTS:
     void backgroundRequestConnectInBackground();
 #endif
 
+    void putWithRateLimiting();
+
     // NOTE: This test must be last!
     void parentingRepliesToTheApp();
 private:
@@ -7776,6 +7778,96 @@ void tst_QNetworkReply::backgroundRequestConnectInBackground()
 }
 #endif
 
+class RateLimitedUploadDevice : public QIODevice
+{
+    Q_OBJECT
+public:
+    QByteArray data;
+    QBuffer buffer;
+    qint64 read;
+    qint64 bandwidthQuota;
+    QTimer timer;
+
+    RateLimitedUploadDevice(QByteArray d) : QIODevice(),data(d),read(0),bandwidthQuota(0) {
+        buffer.setData(data);
+        buffer.open(QIODevice::ReadOnly);
+        timer.setInterval(200);
+        QObject::connect(&timer, SIGNAL(timeout()), this, SLOT(timeoutSlot()));
+        timer.start();
+    }
+
+    virtual qint64 writeData(const char* , qint64 ) {
+        Q_ASSERT(false);
+        return 0;
+    }
+
+    virtual qint64 readData(char* data, qint64 maxlen) {
+        //qDebug() << Q_FUNC_INFO << maxlen << bandwidthQuota;
+        maxlen = qMin(maxlen, buffer.bytesAvailable());
+        maxlen = qMin(maxlen, bandwidthQuota);
+        if (maxlen <= 0) {  // no quota or at end
+            return 0;
+        }
+        bandwidthQuota -= maxlen; // reduce quota
+
+        qint64 ret = buffer.read(data, maxlen);
+        if (ret == -1) {
+            return -1;
+        }
+        read += ret;
+        //qDebug() << Q_FUNC_INFO << maxlen << bandwidthQuota << read << ret << buffer.bytesAvailable();
+        return ret;
+    }
+    virtual bool atEnd() const {
+        return buffer.atEnd();
+    }
+    virtual qint64 size() const{
+        return data.length();
+    }
+    qint64 bytesAvailable() const
+    {
+        return buffer.bytesAvailable() + QIODevice::bytesAvailable();
+    }
+    virtual bool isSequential() const{ // random access, we can seek
+        return false;
+    }
+    virtual bool seek ( qint64 pos ) {
+        return buffer.seek(pos);
+    }
+protected slots:
+    void timeoutSlot() {
+        //qDebug() << Q_FUNC_INFO;
+        bandwidthQuota = 8*1024; // fill quota
+        emit readyRead();
+    }
+};
+
+void tst_QNetworkReply::putWithRateLimiting()
+{
+    QFile reference(testDataDir + "/rfc3252.txt");
+    reference.open(QIODevice::ReadOnly);
+    QByteArray data = reference.readAll();
+    QVERIFY(data.length() > 0);
+
+    QUrl url = QUrl::fromUserInput("http://" + QtNetworkSettings::serverName()+ "/qtest/cgi-bin/echo.cgi?");
+
+    QNetworkRequest request(url);
+    QNetworkReplyPtr reply;
+
+    RateLimitedUploadDevice rateLimitedUploadDevice(data);
+    rateLimitedUploadDevice.open(QIODevice::ReadOnly);
+
+    RUN_REQUEST(runCustomRequest(request, reply,QByteArray("POST"), &rateLimitedUploadDevice));
+    QCOMPARE(reply->error(), QNetworkReply::NoError);
+    QCOMPARE(reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt(), 200);
+
+    QByteArray uploadedData = reply->readAll();
+    QCOMPARE(uploadedData.length(), data.length());
+    QCOMPARE(uploadedData, data);
+}
+
+
+
 // NOTE: This test must be last testcase in tst_qnetworkreply!
 void tst_QNetworkReply::parentingRepliesToTheApp()
 {