Add unit tests for QWebSocket

Added basic unit tests for QWebSocket. Connectivity is not tested yet.

Change-Id: I38d388d289733777e6ff0253e5b201aca99b0b10
Reviewed-by: Kurt Pattyn <pattyn.kurt@gmail.com>

3 files changed, 242 insertions, 0 deletions

diff --git a/tests/auto/auto.pro b/tests/auto/auto.pro
index c05ce29..c6d94fa 100644
--- a/tests/auto/auto.pro
+++ b/tests/auto/auto.pro
@@ -8,3 +8,6 @@ contains(QT_CONFIG, private_tests): SUBDIRS += \
    dataprocessor \
    websocketframe \
    handshakerequest
+
+SUBDIRS += \
+    qwebsocket
diff --git a/tests/auto/qwebsocket/qwebsocket.pro b/tests/auto/qwebsocket/qwebsocket.pro
new file mode 100644
index 0000000..0155d08
--- /dev/null
+++ b/tests/auto/qwebsocket/qwebsocket.pro
@@ -0,0 +1,13 @@
+CONFIG += console
+CONFIG += testcase
+CONFIG -= app_bundle
+
+QT       = core testlib websockets
+
+TARGET = tst_qwebsocket
+
+TEMPLATE = app
+
+SOURCES += tst_qwebsocket.cpp
+
+DEFINES += QT_DISABLE_DEPRECATED_BEFORE=0
diff --git a/tests/auto/qwebsocket/tst_qwebsocket.cpp b/tests/auto/qwebsocket/tst_qwebsocket.cpp
new file mode 100644
index 0000000..773e824
--- /dev/null
+++ b/tests/auto/qwebsocket/tst_qwebsocket.cpp
@@ -0,0 +1,226 @@
+/****************************************************************************
+**
+** Copyright (C) 2013 Digia Plc and/or its subsidiary(-ies).
+** Contact: http://www.qt-project.org/legal
+**
+** This file is part of the test suite of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and Digia.  For licensing terms and
+** conditions see http://qt.digia.com/licensing.  For further information
+** use the contact form at http://qt.digia.com/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Digia gives you certain additional
+** rights.  These rights are described in the Digia Qt LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3.0 as published by the Free Software
+** Foundation and appearing in the file LICENSE.GPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU General Public License version 3.0 requirements will be
+** met: http://www.gnu.org/copyleft/gpl.html.
+**
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+#include <QString>
+#include <QtTest>
+#include <QtWebSockets/QWebSocket>
+#include <QtWebSockets/qwebsocketprotocol.h>
+
+QT_USE_NAMESPACE
+
+Q_DECLARE_METATYPE(QWebSocketProtocol::Version)
+
+class tst_QWebSocket : public QObject
+{
+    Q_OBJECT
+
+public:
+    tst_QWebSocket();
+
+private Q_SLOTS:
+    void init();
+    void initTestCase();
+    void cleanupTestCase();
+    void tst_initialisation_data();
+    void tst_initialisation();
+    void tst_settersAndGetters();
+    void tst_invalidOpen();
+};
+
+tst_QWebSocket::tst_QWebSocket()
+{
+}
+
+void tst_QWebSocket::init()
+{
+    qRegisterMetaType<QWebSocketProtocol::Version>("QWebSocketProtocol::Version");
+}
+
+void tst_QWebSocket::initTestCase()
+{
+}
+
+void tst_QWebSocket::cleanupTestCase()
+{
+}
+
+void tst_QWebSocket::tst_initialisation_data()
+{
+    QTest::addColumn<QString>("origin");
+    QTest::addColumn<QString>("expectedOrigin");
+    QTest::addColumn<QWebSocketProtocol::Version>("version");
+    QTest::addColumn<QWebSocketProtocol::Version>("expectedVersion");
+
+    QTest::newRow("Default origin and version")
+            << QString() << QString()
+            << QWebSocketProtocol::VersionUnknown << QWebSocketProtocol::VersionLatest;
+    QTest::newRow("Specific origin and default version")
+            << QStringLiteral("qt-project.org") << QStringLiteral("qt-project.org")
+            << QWebSocketProtocol::VersionUnknown << QWebSocketProtocol::VersionLatest;
+    QTest::newRow("Specific origin and specific version")
+            << QStringLiteral("qt-project.org") << QStringLiteral("qt-project.org")
+            << QWebSocketProtocol::Version7 << QWebSocketProtocol::Version7;
+}
+
+void tst_QWebSocket::tst_initialisation()
+{
+    QFETCH(QString, origin);
+    QFETCH(QString, expectedOrigin);
+    QFETCH(QWebSocketProtocol::Version, version);
+    QFETCH(QWebSocketProtocol::Version, expectedVersion);
+
+    QScopedPointer<QWebSocket> socket;
+
+    if (origin.isEmpty() && (version == QWebSocketProtocol::VersionUnknown))
+        socket.reset(new QWebSocket);
+    else if (!origin.isEmpty() && (version == QWebSocketProtocol::VersionUnknown))
+        socket.reset(new QWebSocket(origin));
+    else
+        socket.reset(new QWebSocket(origin, version));
+
+    QCOMPARE(socket->origin(), expectedOrigin);
+    QCOMPARE(socket->version(), expectedVersion);
+    QCOMPARE(socket->error(), QAbstractSocket::UnknownSocketError);
+    //error string defaults to "Unknown error" (localised)
+    QVERIFY(!socket->errorString().isEmpty());
+    QVERIFY(!socket->isValid());
+    QVERIFY(socket->localAddress().isNull());
+    QCOMPARE(socket->localPort(), quint16(0));
+    QCOMPARE(socket->pauseMode(), QAbstractSocket::PauseNever);
+    QVERIFY(socket->peerAddress().isNull());
+    QCOMPARE(socket->peerPort(), quint16(0));
+    QVERIFY(socket->peerName().isEmpty());
+    QCOMPARE(socket->state(), QAbstractSocket::UnconnectedState);
+    QCOMPARE(socket->readBufferSize(), 0);
+    QVERIFY(socket->resourceName().isEmpty());
+    QVERIFY(!socket->requestUrl().isValid());
+    QCOMPARE(socket->closeCode(), QWebSocketProtocol::CloseCodeNormal);
+    QVERIFY(socket->closeReason().isEmpty());
+    QVERIFY(socket->flush());
+    QCOMPARE(socket->sendTextMessage(QStringLiteral("A text message")), 0);
+    QCOMPARE(socket->sendBinaryMessage(QByteArrayLiteral("A binary message")), 0);
+}
+
+void tst_QWebSocket::tst_settersAndGetters()
+{
+    QWebSocket socket;
+
+    socket.setPauseMode(QAbstractSocket::PauseNever);
+    QCOMPARE(socket.pauseMode(), QAbstractSocket::PauseNever);
+    socket.setPauseMode(QAbstractSocket::PauseOnSslErrors);
+    QCOMPARE(socket.pauseMode(), QAbstractSocket::PauseOnSslErrors);
+
+    socket.setReadBufferSize(0);
+    QCOMPARE(socket.readBufferSize(), 0);
+    socket.setReadBufferSize(128);
+    QCOMPARE(socket.readBufferSize(), 128);
+    socket.setReadBufferSize(-1);
+    QCOMPARE(socket.readBufferSize(), -1);
+}
+
+void tst_QWebSocket::tst_invalidOpen()
+{
+    QWebSocket socket;
+    QSignalSpy errorSpy(&socket, SIGNAL(error(QAbstractSocket::SocketError)));
+    QSignalSpy aboutToCloseSpy(&socket, SIGNAL(aboutToClose()));
+    QSignalSpy connectedSpy(&socket, SIGNAL(connected()));
+    QSignalSpy disconnectedSpy(&socket, SIGNAL(disconnected()));
+    QSignalSpy stateChangedSpy(&socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)));
+    QSignalSpy readChannelFinishedSpy(&socket, SIGNAL(readChannelFinished()));
+    QSignalSpy textFrameReceivedSpy(&socket, SIGNAL(textFrameReceived(QString,bool)));
+    QSignalSpy binaryFrameReceivedSpy(&socket, SIGNAL(binaryFrameReceived(QByteArray,bool)));
+    QSignalSpy textMessageReceivedSpy(&socket, SIGNAL(textMessageReceived(QString)));
+    QSignalSpy binaryMessageReceivedSpy(&socket, SIGNAL(binaryMessageReceived(QByteArray)));
+    QSignalSpy pongSpy(&socket, SIGNAL(pong(quint64,QByteArray)));
+    QSignalSpy bytesWrittenSpy(&socket, SIGNAL(bytesWritten(qint64)));
+
+    socket.open(QUrl(QStringLiteral("ws://SomeNonExistingWebSocketServer/")), true);
+
+    QVERIFY(socket.origin().isEmpty());
+    QCOMPARE(socket.version(), QWebSocketProtocol::VersionLatest);
+    //at this point the socket is in a connecting state
+    //so, there should no error at this point
+    QCOMPARE(socket.error(), QAbstractSocket::UnknownSocketError);
+    QVERIFY(!socket.errorString().isEmpty());
+    QVERIFY(!socket.isValid());
+    QVERIFY(socket.localAddress().isNull());
+    QCOMPARE(socket.localPort(), quint16(0));
+    QCOMPARE(socket.pauseMode(), QAbstractSocket::PauseNever);
+    QVERIFY(socket.peerAddress().isNull());
+    QCOMPARE(socket.peerPort(), quint16(0));
+    QCOMPARE(socket.peerName(), QStringLiteral("somenonexistingwebsocketserver"));
+    QCOMPARE(socket.state(), QAbstractSocket::ConnectingState);
+    QCOMPARE(socket.readBufferSize(), 0);
+    QCOMPARE(socket.resourceName(), QStringLiteral("/"));
+    QCOMPARE(socket.requestUrl(), QUrl(QStringLiteral("ws://SomeNonExistingWebSocketServer/")));
+    QCOMPARE(socket.closeCode(), QWebSocketProtocol::CloseCodeNormal);
+    QVERIFY(socket.closeReason().isEmpty());
+    QVERIFY(!socket.flush());   //flush should fail if socket is in connecting state
+    QCOMPARE(socket.sendTextMessage(QStringLiteral("A text message")), 0);
+    QCOMPARE(socket.sendBinaryMessage(QByteArrayLiteral("A text message")), 0);
+
+    errorSpy.wait();
+    QCOMPARE(errorSpy.count(), 1);
+    QList<QVariant> arguments = errorSpy.takeFirst();
+    QAbstractSocket::SocketError socketError =
+            qvariant_cast<QAbstractSocket::SocketError>(arguments.at(0));
+    QCOMPARE(socketError, QAbstractSocket::HostNotFoundError);
+    QCOMPARE(aboutToCloseSpy.count(), 0);
+    QCOMPARE(connectedSpy.count(), 0);
+    QCOMPARE(disconnectedSpy.count(), 1);
+    QCOMPARE(stateChangedSpy.count(), 2);   //connectingstate, unconnectedstate
+    arguments = stateChangedSpy.takeFirst();
+    QAbstractSocket::SocketState socketState =
+            qvariant_cast<QAbstractSocket::SocketState>(arguments.at(0));
+    arguments = stateChangedSpy.takeFirst();
+    socketState = qvariant_cast<QAbstractSocket::SocketState>(arguments.at(0));
+    QCOMPARE(socketState, QAbstractSocket::UnconnectedState);
+    QCOMPARE(readChannelFinishedSpy.count(), 0);
+    QCOMPARE(textFrameReceivedSpy.count(), 0);
+    QCOMPARE(binaryFrameReceivedSpy.count(), 0);
+    QCOMPARE(textMessageReceivedSpy.count(), 0);
+    QCOMPARE(binaryMessageReceivedSpy.count(), 0);
+    QCOMPARE(pongSpy.count(), 0);
+    QCOMPARE(bytesWrittenSpy.count(), 0);
+}
+
+QTEST_MAIN(tst_QWebSocket)
+
+#include "tst_qwebsocket.moc"