TCPLinkTest.cc 6.78 KB
Newer Older
Don Gagne's avatar
Don Gagne committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
/*=====================================================================
 
 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"
25
#include "TCPLoopBackServer.h"
Don Gagne's avatar
Don Gagne committed
26 27 28 29 30 31

/// @file
///     @brief TCPLink class unit test
///
///     @author Don Gagne <don@thegagnes.com>

32 33 34 35
TCPLinkUnitTest::TCPLinkUnitTest(void)
    : _config(NULL)
    , _link(NULL)
    , _multiSpy(NULL)
Don Gagne's avatar
Don Gagne committed
36
{
37 38 39
    _config = new TCPConfiguration("MockTCP");
    _config->setAddress(QHostAddress::LocalHost);
    _config->setPort(5760);
Don Gagne's avatar
Don Gagne committed
40 41 42 43 44
}

// Called before every test
void TCPLinkUnitTest::init(void)
{
Don Gagne's avatar
Don Gagne committed
45 46
    UnitTest::init();
    
Don Gagne's avatar
Don Gagne committed
47 48 49
    Q_ASSERT(_link == NULL);
    Q_ASSERT(_multiSpy == NULL);

50
    _link = new TCPLink(_config);
Don Gagne's avatar
Don Gagne committed
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
    Q_ASSERT(_link != NULL);

    _rgSignals[bytesReceivedSignalIndex] = SIGNAL(bytesReceived(LinkInterface*, QByteArray));
    _rgSignals[connectedSignalIndex] = SIGNAL(connected(void));
    _rgSignals[disconnectedSignalIndex] = SIGNAL(disconnected(void));
    _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);
69
    Q_ASSERT(_config);
Don Gagne's avatar
Don Gagne committed
70 71 72

    delete _multiSpy;
    delete _link;
73
    delete _config;
Don Gagne's avatar
Don Gagne committed
74 75

    _multiSpy = NULL;
76 77
    _link     = NULL;
    _config   = NULL;
Don Gagne's avatar
Don Gagne committed
78
    UnitTest::cleanup();
Don Gagne's avatar
Don Gagne committed
79 80 81 82
}

void TCPLinkUnitTest::_properties_test(void)
{
83
    Q_ASSERT(_config);
Don Gagne's avatar
Don Gagne committed
84 85 86 87
    Q_ASSERT(_link);
    Q_ASSERT(_multiSpy);
    Q_ASSERT(_multiSpy->checkNoSignals() == true);
    
88
    // Test no longer valid
Don Gagne's avatar
Don Gagne committed
89 90 91 92
}

void TCPLinkUnitTest::_nameChangedSignal_test(void)
{
93 94
    // Test no longer valid
    Q_ASSERT(_config);
Don Gagne's avatar
Don Gagne committed
95 96 97 98 99 100
    Q_ASSERT(_link);
    Q_ASSERT(_multiSpy);
}

void TCPLinkUnitTest::_connectFail_test(void)
{
101
    Q_ASSERT(_config);
Don Gagne's avatar
Don Gagne committed
102 103 104 105
    Q_ASSERT(_link);
    Q_ASSERT(_multiSpy);
    Q_ASSERT(_multiSpy->checkNoSignals() == true);
    
106 107
    // With the new threading model connect will always succeed. We only get an error signal
    // for a failed connected.
108
    QCOMPARE(_link->_connect(), true);
Don Gagne's avatar
Don Gagne committed
109 110 111 112 113 114 115

    // 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);
116
    
117
    _link->_disconnect();
Don Gagne's avatar
Don Gagne committed
118 119 120

    // Try to connect again to make sure everything was cleaned up correctly from previous failed connection
    
121
    QCOMPARE(_link->_connect(), true);
Don Gagne's avatar
Don Gagne committed
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
    
    // 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
138
    TCPLoopBackServer* server = new TCPLoopBackServer(_config->address(), _config->port());
139
    Q_CHECK_PTR(server);
Don Gagne's avatar
Don Gagne committed
140 141
    
    // Connect to the server
142
    QCOMPARE(_link->_connect(), true);
Don Gagne's avatar
Don Gagne committed
143
    
144
    // Make sure we get the connected signals
Don Gagne's avatar
Don Gagne committed
145
    QCOMPARE(_multiSpy->waitForSignalByIndex(connectedSignalIndex, 10000), true);
146
    QCOMPARE(_multiSpy->checkOnlySignalByMask(connectedSignalMask), true);
147
    _multiSpy->clearAllSignals();
Don Gagne's avatar
Don Gagne committed
148 149 150
    
    // Test link->server data path
    
151 152
    QByteArray bytesOut("test");

Don Gagne's avatar
Don Gagne committed
153
    // Write data from link to server
154 155 156
    const char* bytesWrittenSignal = SIGNAL(bytesWritten(qint64));
    MultiSignalSpy bytesWrittenSpy;
    QCOMPARE(bytesWrittenSpy.init(_link->getSocket(), &bytesWrittenSignal, 1), true);
Don Gagne's avatar
Don Gagne committed
157
    _link->writeBytes(bytesOut.data(), bytesOut.size());
158 159 160 161 162 163
    _multiSpy->clearAllSignals();
    
    // We emit this signal such that it will be queued and run on the TCPLink thread. This in turn
    // allows the TCPLink object to pump the bytes through.
    connect(this, SIGNAL(waitForBytesWritten(int)), _link, SLOT(waitForBytesWritten(int)));
    emit waitForBytesWritten(1000);
Don Gagne's avatar
Don Gagne committed
164

165
    // Check for loopback, both from signal received and actual bytes returned
Don Gagne's avatar
Don Gagne committed
166

167 168 169
    QCOMPARE(_multiSpy->waitForSignalByIndex(bytesReceivedSignalIndex, 1000), true);
    QCOMPARE(_multiSpy->checkOnlySignalByMask(bytesReceivedSignalMask), true);
    
Don Gagne's avatar
Don Gagne committed
170
    // Read the data and make sure it matches
171
    QList<QVariant> arguments = _multiSpy->getSpyByIndex(bytesReceivedSignalIndex)->takeFirst();
172
    QVERIFY(arguments.at(1).toByteArray() == bytesOut);
Don Gagne's avatar
Don Gagne committed
173
    
174 175
    _multiSpy->clearAllSignals();

Don Gagne's avatar
Don Gagne committed
176
    // Disconnect the link
177
    _link->_disconnect();
Don Gagne's avatar
Don Gagne committed
178
    
179
    // Make sure we get the disconnected signal on link side
Don Gagne's avatar
Don Gagne committed
180
    QCOMPARE(_multiSpy->waitForSignalByIndex(disconnectedSignalIndex, 1000), true);
181
    QCOMPARE(_multiSpy->checkOnlySignalByMask(disconnectedSignalMask), true);
182
    _multiSpy->clearAllSignals();
Don Gagne's avatar
Don Gagne committed
183 184 185 186
    
    // Try to connect again to make sure everything was cleaned up correctly from previous connection
    
    // Connect to the server
187
    QCOMPARE(_link->_connect(), true);
Don Gagne's avatar
Don Gagne committed
188
    
189
    // Make sure we get the connected signal
Don Gagne's avatar
Don Gagne committed
190
    QCOMPARE(_multiSpy->waitForSignalByIndex(connectedSignalIndex, 1000), true);
191
    QCOMPARE(_multiSpy->checkOnlySignalByMask(connectedSignalMask), true);
192 193 194 195
    _multiSpy->clearAllSignals();
    
    server->quit();
    QTest::qWait(500);  // Wait a little for server thread to terminate
Don Gagne's avatar
Don Gagne committed
196
    delete server;
Don Gagne's avatar
Don Gagne committed
197
}