Revision control
Copy as Markdown
// Copyright (C) 2016 Kurt Pattyn <pattyn.kurt@gmail.com>.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#include <QtTest/QtTest>
#include <QtTest/qtestcase.h>
#include <QSignalSpy>
#include <QHostInfo>
#include <QDebug>
#include "QtWebSockets/QWebSocket"
// the test.
// wstest is part of the autobahn-testsuite:
class tst_WebSocketsTest : public QObject
{
Q_OBJECT
public:
tst_WebSocketsTest();
private Q_SLOTS:
void initTestCase();
void cleanupTestCase();
void init();
void cleanup();
/**
* @brief Test isValid() with an unoped socket
*/
void testInvalidWithUnopenedSocket();
/**
* @brief testTextMessage Tests sending and receiving a text message
*/
void testTextMessage();
void testBinaryMessage();
/**
* @brief Tests the method localAddress and localPort
*/
void testLocalAddress();
/**
* @brief Test the methods peerAddress, peerName and peerPort
*/
void testPeerAddress();
/**
* @brief Test the methods proxy() and setProxy() and check if it can be correctly set
*/
void testProxy();
/**
* @brief Runs the autobahn tests against our implementation
*/
//void autobahnTest();
private:
QWebSocket *m_pWebSocket;
QUrl m_url;
};
tst_WebSocketsTest::tst_WebSocketsTest() :
m_pWebSocket(0),
{
}
void tst_WebSocketsTest::initTestCase()
{
m_pWebSocket = new QWebSocket();
m_pWebSocket->open(m_url);
QTRY_VERIFY_WITH_TIMEOUT(m_pWebSocket->state() == QAbstractSocket::ConnectedState, 1000);
QVERIFY(m_pWebSocket->isValid());
}
void tst_WebSocketsTest::cleanupTestCase()
{
if (m_pWebSocket)
{
m_pWebSocket->close();
//QVERIFY2(m_pWebSocket->waitForDisconnected(1000), "Disconnection failed.");
delete m_pWebSocket;
m_pWebSocket = 0;
}
}
void tst_WebSocketsTest::init()
{
}
void tst_WebSocketsTest::cleanup()
{
}
void tst_WebSocketsTest::testTextMessage()
{
const QString message = QStringLiteral("Hello world!");
QSignalSpy spy(m_pWebSocket, SIGNAL(textMessageReceived(QString)));
QCOMPARE(m_pWebSocket->sendTextMessage(message), qint64(message.length()));
QTRY_VERIFY_WITH_TIMEOUT(spy.count() != 0, 1000);
QCOMPARE(spy.count(), 1);
QCOMPARE(spy.at(0).count(), 1);
QCOMPARE(spy.takeFirst().at(0).toString(), message);
}
void tst_WebSocketsTest::testBinaryMessage()
{
QSignalSpy spy(m_pWebSocket, SIGNAL(binaryMessageReceived(QByteArray)));
QByteArray data("Hello world!");
QCOMPARE(m_pWebSocket->sendBinaryMessage(data), qint64(data.size()));
QTRY_VERIFY_WITH_TIMEOUT(spy.count() != 0, 1000);
QCOMPARE(spy.count(), 1);
QCOMPARE(spy.at(0).count(), 1);
QCOMPARE(spy.takeFirst().at(0).toByteArray(), data);
}
void tst_WebSocketsTest::testLocalAddress()
{
QCOMPARE(m_pWebSocket->localAddress().toString(), QStringLiteral("127.0.0.1"));
quint16 localPort = m_pWebSocket->localPort();
QVERIFY2(localPort > 0, "Local port is invalid.");
}
void tst_WebSocketsTest::testPeerAddress()
{
QHostInfo hostInfo = QHostInfo::fromName(m_url.host());
const QList<QHostAddress> addresses = hostInfo.addresses();
QVERIFY(addresses.length() > 0);
QHostAddress peer = m_pWebSocket->peerAddress();
bool found = false;
for (const QHostAddress &a : addresses) {
if (a == peer)
{
found = true;
break;
}
}
if (!found)
{
QFAIL("PeerAddress is not found as a result of a reverse lookup");
}
QCOMPARE(m_pWebSocket->peerName(), m_url.host());
QCOMPARE(m_pWebSocket->peerPort(), (quint16)m_url.port(80));
}
void tst_WebSocketsTest::testProxy()
{
QNetworkProxy oldProxy = m_pWebSocket->proxy();
QNetworkProxy proxy(QNetworkProxy::HttpProxy, QStringLiteral("proxy.network.com"), 80);
m_pWebSocket->setProxy(proxy);
QCOMPARE(proxy, m_pWebSocket->proxy());
m_pWebSocket->setProxy(oldProxy);
QCOMPARE(oldProxy, m_pWebSocket->proxy());
}
void tst_WebSocketsTest::testInvalidWithUnopenedSocket()
{
QWebSocket qws;
QCOMPARE(qws.isValid(), false);
}
QTEST_MAIN(tst_WebSocketsTest)
#include "tst_websockets.moc"