2 files changed, 16 insertions, 0 deletions
diff --git a/tests/auto/websockets/qwebsocket/tst_qwebsocket.cpp b/tests/auto/websockets/qwebsocket/tst_qwebsocket.cpp
index 211ea25..4dc60c7 100644
--- a/tests/auto/websockets/qwebsocket/tst_qwebsocket.cpp
+++ b/tests/auto/websockets/qwebsocket/tst_qwebsocket.cpp
@@ -435,9 +435,13 @@ void tst_QWebSocket::tst_sendTextMessage()
     QUrl urlConnected = arguments.at(0).toUrl();
     QCOMPARE(urlConnected, url);
 
+    QCOMPARE(socket.bytesToWrite(), 0);
     socket.sendTextMessage(QStringLiteral("Hello world!"));
+    QVERIFY(socket.bytesToWrite() > 12); // 12 + a few extra bytes for header
 
     QVERIFY(textMessageReceived.wait(500));
+    QCOMPARE(socket.bytesToWrite(), 0);
+
     QCOMPARE(textMessageReceived.count(), 1);
     QCOMPARE(binaryMessageReceived.count(), 0);
     QCOMPARE(binaryFrameReceived.count(), 0);
@@ -509,9 +513,13 @@ void tst_QWebSocket::tst_sendBinaryMessage()
     QTRY_COMPARE(socketConnectedSpy.count(), 1);
     QCOMPARE(socket.state(), QAbstractSocket::ConnectedState);
 
+    QCOMPARE(socket.bytesToWrite(), 0);
     socket.sendBinaryMessage(QByteArrayLiteral("Hello world!"));
+    QVERIFY(socket.bytesToWrite() > 12); // 12 + a few extra bytes for header
 
     QVERIFY(binaryMessageReceived.wait(500));
+    QCOMPARE(socket.bytesToWrite(), 0);
+
     QCOMPARE(textMessageReceived.count(), 0);
     QCOMPARE(textFrameReceived.count(), 0);
     QCOMPARE(binaryMessageReceived.count(), 1);
diff --git a/tests/auto/websockets/qwebsocketserver/tst_qwebsocketserver.cpp b/tests/auto/websockets/qwebsocketserver/tst_qwebsocketserver.cpp
index 8a3760d..a2f3c15 100644
--- a/tests/auto/websockets/qwebsocketserver/tst_qwebsocketserver.cpp
+++ b/tests/auto/websockets/qwebsocketserver/tst_qwebsocketserver.cpp
@@ -640,6 +640,14 @@ void tst_QWebSocketServer::tst_handleConnection()
     QTRY_COMPARE(wsMessageReceivedSpy.count(), 1);
     QList<QVariant> arguments = wsMessageReceivedSpy.takeFirst();
     QCOMPARE(arguments.first().toString(), QString("dummy"));
+
+    QSignalSpy clientMessageReceivedSpy(&webSocket, &QWebSocket::textMessageReceived);
+    webServerSocket->sendTextMessage("hello");
+    QVERIFY(webServerSocket->bytesToWrite() > 5); // 5 + a few extra bytes for header
+    QTRY_COMPARE(webServerSocket->bytesToWrite(), 0);
+    QTRY_COMPARE(clientMessageReceivedSpy.count(), 1);
+    arguments = clientMessageReceivedSpy.takeFirst();
+    QCOMPARE(arguments.first().toString(), QString("hello"));
 }
 
 QTEST_MAIN(tst_QWebSocketServer)