Don Gagne
12 years ago
2 changed files with 308 additions and 0 deletions
@ -0,0 +1,218 @@ |
|||||||
|
/*=====================================================================
|
||||||
|
|
||||||
|
QGroundControl Open Source Ground Control Station |
||||||
|
|
||||||
|
(c) 2009 - 2014 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
|
||||||
|
This file is part of the QGROUNDCONTROL project |
||||||
|
|
||||||
|
QGROUNDCONTROL is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
QGROUNDCONTROL is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU General Public License for more details. |
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License |
||||||
|
along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
======================================================================*/ |
||||||
|
|
||||||
|
#include "TCPLinkTest.h" |
||||||
|
#include <QTcpServer> |
||||||
|
|
||||||
|
/// @file
|
||||||
|
/// @brief TCPLink class unit test
|
||||||
|
///
|
||||||
|
/// @author Don Gagne <don@thegagnes.com>
|
||||||
|
|
||||||
|
TCPLinkUnitTest::TCPLinkUnitTest(void) : |
||||||
|
_link(NULL), |
||||||
|
_hostAddress(QHostAddress::LocalHost), |
||||||
|
_port(5760), |
||||||
|
_multiSpy(NULL) |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
// Called before every test
|
||||||
|
void TCPLinkUnitTest::init(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_link == NULL); |
||||||
|
Q_ASSERT(_multiSpy == NULL); |
||||||
|
|
||||||
|
_link = new TCPLink(_hostAddress, _port); |
||||||
|
Q_ASSERT(_link != NULL); |
||||||
|
|
||||||
|
_rgSignals[bytesReceivedSignalIndex] = SIGNAL(bytesReceived(LinkInterface*, QByteArray)); |
||||||
|
_rgSignals[connectedSignalIndex] = SIGNAL(connected(void)); |
||||||
|
_rgSignals[disconnectedSignalIndex] = SIGNAL(disconnected(void)); |
||||||
|
_rgSignals[connected2SignalIndex] = SIGNAL(connected(bool)); |
||||||
|
_rgSignals[nameChangedSignalIndex] = SIGNAL(nameChanged(QString)); |
||||||
|
_rgSignals[communicationErrorSignalIndex] = SIGNAL(communicationError(const QString&, const QString&)); |
||||||
|
_rgSignals[communicationUpdateSignalIndex] = SIGNAL(communicationUpdate(const QString&, const QString&)); |
||||||
|
_rgSignals[deleteLinkSignalIndex] = SIGNAL(deleteLink(LinkInterface* const)); |
||||||
|
|
||||||
|
_multiSpy = new MultiSignalSpy(); |
||||||
|
QCOMPARE(_multiSpy->init(_link, _rgSignals, _cSignals), true); |
||||||
|
} |
||||||
|
|
||||||
|
// Called after every test
|
||||||
|
void TCPLinkUnitTest::cleanup(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_multiSpy); |
||||||
|
Q_ASSERT(_link); |
||||||
|
|
||||||
|
delete _multiSpy; |
||||||
|
delete _link; |
||||||
|
|
||||||
|
_multiSpy = NULL; |
||||||
|
_link = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
void TCPLinkUnitTest::_properties_test(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_link); |
||||||
|
Q_ASSERT(_multiSpy); |
||||||
|
Q_ASSERT(_multiSpy->checkNoSignals() == true); |
||||||
|
|
||||||
|
// Make sure we get the right values back
|
||||||
|
QHostAddress hostAddressOut; |
||||||
|
quint16 portOut; |
||||||
|
|
||||||
|
hostAddressOut = _link->getHostAddress(); |
||||||
|
QCOMPARE(_hostAddress, hostAddressOut); |
||||||
|
|
||||||
|
portOut = _link->getPort(); |
||||||
|
QCOMPARE(_port, portOut); |
||||||
|
} |
||||||
|
|
||||||
|
void TCPLinkUnitTest::_nameChangedSignal_test(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_link); |
||||||
|
Q_ASSERT(_multiSpy); |
||||||
|
Q_ASSERT(_multiSpy->checkNoSignals() == true); |
||||||
|
|
||||||
|
_link->setHostAddress(QHostAddress("127.1.1.1")); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(nameChangedSignalMask), true); |
||||||
|
_multiSpy->clearSignalByIndex(nameChangedSignalIndex); |
||||||
|
|
||||||
|
_link->setPort(42); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(nameChangedSignalMask), true); |
||||||
|
_multiSpy->clearSignalByIndex(nameChangedSignalIndex);
|
||||||
|
} |
||||||
|
|
||||||
|
void TCPLinkUnitTest::_connectFail_test(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_link); |
||||||
|
Q_ASSERT(_multiSpy); |
||||||
|
Q_ASSERT(_multiSpy->checkNoSignals() == true); |
||||||
|
|
||||||
|
QCOMPARE(_link->connect(), false); |
||||||
|
|
||||||
|
// Make sure we get a linkError signal with the right link name
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(communicationErrorSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(communicationErrorSignalMask), true); |
||||||
|
QList<QVariant> arguments = _multiSpy->getSpyByIndex(communicationErrorSignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(0).toString(), _link->getName()); |
||||||
|
_multiSpy->clearSignalByIndex(communicationErrorSignalIndex); |
||||||
|
|
||||||
|
// Try to connect again to make sure everything was cleaned up correctly from previous failed connection
|
||||||
|
|
||||||
|
QCOMPARE(_link->connect(), false); |
||||||
|
|
||||||
|
// Make sure we get a linkError signal with the right link name
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(communicationErrorSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(communicationErrorSignalMask), true); |
||||||
|
arguments = _multiSpy->getSpyByIndex(communicationErrorSignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(0).toString(), _link->getName()); |
||||||
|
_multiSpy->clearSignalByIndex(communicationErrorSignalIndex); |
||||||
|
} |
||||||
|
|
||||||
|
void TCPLinkUnitTest::_connectSucceed_test(void) |
||||||
|
{ |
||||||
|
Q_ASSERT(_link); |
||||||
|
Q_ASSERT(_multiSpy); |
||||||
|
Q_ASSERT(_multiSpy->checkNoSignals() == true); |
||||||
|
|
||||||
|
// Start the server side
|
||||||
|
QTcpServer* server = new QTcpServer(this); |
||||||
|
QCOMPARE(server->listen(_hostAddress, _port), true); |
||||||
|
|
||||||
|
// Connect to the server
|
||||||
|
QCOMPARE(_link->connect(), true); |
||||||
|
|
||||||
|
// Wait for the connection to come through on server side
|
||||||
|
QCOMPARE(server->waitForNewConnection(1000), true); |
||||||
|
QTcpSocket* serverSocket = server->nextPendingConnection(); |
||||||
|
Q_ASSERT(serverSocket); |
||||||
|
|
||||||
|
// Make sure we get the two different connected signals
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(connectedSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(connectedSignalMask | connected2SignalMask), true); |
||||||
|
QList<QVariant> arguments = _multiSpy->getSpyByIndex(connected2SignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(0).toBool(), true); |
||||||
|
_multiSpy->clearSignalsByMask(connectedSignalMask); |
||||||
|
|
||||||
|
// Test server->link data path
|
||||||
|
|
||||||
|
QByteArray bytesOut("test"); |
||||||
|
|
||||||
|
// Write data from server to link
|
||||||
|
serverSocket->write(bytesOut); |
||||||
|
|
||||||
|
// Make sure we get the bytesReceived signal, with the correct data
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(bytesReceivedSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(bytesReceivedSignalMask), true); |
||||||
|
arguments = _multiSpy->getSpyByIndex(bytesReceivedSignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(1), QVariant(bytesOut)); |
||||||
|
_multiSpy->clearSignalByIndex(bytesReceivedSignalIndex); |
||||||
|
|
||||||
|
// Test link->server data path
|
||||||
|
|
||||||
|
// Write data from link to server
|
||||||
|
_link->writeBytes(bytesOut.data(), bytesOut.size()); |
||||||
|
QCOMPARE(_link->getSocket()->waitForBytesWritten(1000), true); |
||||||
|
|
||||||
|
// Make sure we get readyRead on server
|
||||||
|
QCOMPARE(serverSocket->waitForReadyRead(1000), true); |
||||||
|
|
||||||
|
// Read the data and make sure it matches
|
||||||
|
QByteArray bytesIn = serverSocket->read(bytesOut.size() + 100); |
||||||
|
QCOMPARE(bytesIn, bytesOut); |
||||||
|
|
||||||
|
// Disconnect the link
|
||||||
|
_link->disconnect(); |
||||||
|
|
||||||
|
// Make sure we get the disconnected signals on link side
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(disconnectedSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(disconnectedSignalMask | connected2SignalMask), true); |
||||||
|
arguments = _multiSpy->getSpyByIndex(connected2SignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(0).toBool(), false); |
||||||
|
_multiSpy->clearSignalsByMask(disconnectedSignalMask); |
||||||
|
|
||||||
|
// Make sure we get disconnected signals from the server side
|
||||||
|
QCOMPARE(serverSocket->waitForDisconnected(1000), true ); |
||||||
|
|
||||||
|
// Try to connect again to make sure everything was cleaned up correctly from previous connection
|
||||||
|
|
||||||
|
// Connect to the server
|
||||||
|
QCOMPARE(_link->connect(), true); |
||||||
|
|
||||||
|
// Wait for the connection to come through on server side
|
||||||
|
QCOMPARE(server->waitForNewConnection(1000), true); |
||||||
|
serverSocket = server->nextPendingConnection(); |
||||||
|
Q_ASSERT(serverSocket); |
||||||
|
|
||||||
|
// Make sure we get the two different connected signals
|
||||||
|
QCOMPARE(_multiSpy->waitForSignalByIndex(connectedSignalIndex, 1000), true); |
||||||
|
QCOMPARE(_multiSpy->checkOnlySignalByMask(connectedSignalMask | connected2SignalMask), true); |
||||||
|
arguments = _multiSpy->getSpyByIndex(connected2SignalIndex)->takeFirst(); |
||||||
|
QCOMPARE(arguments.at(0).toBool(), true); |
||||||
|
_multiSpy->clearSignalsByMask(connectedSignalMask); |
||||||
|
|
||||||
|
delete server; |
||||||
|
} |
||||||
@ -0,0 +1,90 @@ |
|||||||
|
/*=====================================================================
|
||||||
|
|
||||||
|
QGroundControl Open Source Ground Control Station |
||||||
|
|
||||||
|
(c) 2009 - 2014 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
|
||||||
|
This file is part of the QGROUNDCONTROL project |
||||||
|
|
||||||
|
QGROUNDCONTROL is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
QGROUNDCONTROL is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU General Public License for more details. |
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License |
||||||
|
along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
======================================================================*/ |
||||||
|
|
||||||
|
#ifndef TCPLINKTEST_H |
||||||
|
#define TCPLINKTEST_H |
||||||
|
|
||||||
|
#include <QObject> |
||||||
|
#include <QtTest/QtTest> |
||||||
|
#include <QApplication> |
||||||
|
|
||||||
|
#include "AutoTest.h" |
||||||
|
#include "TCPLink.h" |
||||||
|
#include "MultiSignalSpy.h" |
||||||
|
|
||||||
|
/// @file
|
||||||
|
/// @brief TCPLink class unit test
|
||||||
|
///
|
||||||
|
/// @author Don Gagne <don@thegagnes.com>
|
||||||
|
|
||||||
|
class TCPLinkUnitTest : public QObject |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
public: |
||||||
|
TCPLinkUnitTest(void); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void init(void); |
||||||
|
void cleanup(void); |
||||||
|
|
||||||
|
void _properties_test(void); |
||||||
|
void _nameChangedSignal_test(void); |
||||||
|
void _connectFail_test(void); |
||||||
|
void _connectSucceed_test(void); |
||||||
|
|
||||||
|
private: |
||||||
|
enum { |
||||||
|
bytesReceivedSignalIndex = 0, |
||||||
|
connectedSignalIndex, |
||||||
|
disconnectedSignalIndex, |
||||||
|
connected2SignalIndex, |
||||||
|
nameChangedSignalIndex, |
||||||
|
communicationErrorSignalIndex, |
||||||
|
communicationUpdateSignalIndex, |
||||||
|
deleteLinkSignalIndex, |
||||||
|
maxSignalIndex |
||||||
|
}; |
||||||
|
|
||||||
|
enum { |
||||||
|
bytesReceivedSignalMask = 1 << bytesReceivedSignalIndex, |
||||||
|
connectedSignalMask = 1 << connectedSignalIndex, |
||||||
|
disconnectedSignalMask = 1 << disconnectedSignalIndex, |
||||||
|
connected2SignalMask = 1 << connected2SignalIndex, |
||||||
|
nameChangedSignalMask = 1 << nameChangedSignalIndex, |
||||||
|
communicationErrorSignalMask = 1 << communicationErrorSignalIndex, |
||||||
|
communicationUpdateSignalMask = 1 << communicationUpdateSignalIndex, |
||||||
|
deleteLinkSignalMask = 1 << deleteLinkSignalIndex, |
||||||
|
}; |
||||||
|
|
||||||
|
TCPLink* _link; |
||||||
|
QHostAddress _hostAddress; |
||||||
|
quint16 _port; |
||||||
|
MultiSignalSpy* _multiSpy; |
||||||
|
static const size_t _cSignals = maxSignalIndex; |
||||||
|
const char* _rgSignals[_cSignals]; |
||||||
|
}; |
||||||
|
|
||||||
|
DECLARE_TEST(TCPLinkUnitTest) |
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue