QGCUASFileManagerTest.h 3.19 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 25 26 27 28 29
/*=====================================================================
 
 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 QGCUASFILEMANAGERTEST_H
#define QGCUASFILEMANAGERTEST_H

#include <QObject>
#include <QtTest/QtTest>

Don Gagne's avatar
Don Gagne committed
30
#include "UnitTest.h"
Don Gagne's avatar
Don Gagne committed
31 32 33 34 35 36 37 38 39 40
#include "MockUAS.h"
#include "MockMavlinkFileServer.h"
#include "QGCUASFileManager.h"
#include "MultiSignalSpy.h"

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

Don Gagne's avatar
Don Gagne committed
41
class QGCUASFileManagerUnitTest : public UnitTest
Don Gagne's avatar
Don Gagne committed
42 43 44 45 46 47 48 49 50
{
    Q_OBJECT
    
public:
    QGCUASFileManagerUnitTest(void);
    
private slots:
    // Test case initialization
    void initTestCase(void);
51
    void cleanupTestCase(void);
Don Gagne's avatar
Don Gagne committed
52 53 54 55 56 57 58 59
    void init(void);
    void cleanup(void);
    
    // Test cases
    void _ackTest(void);
    void _noAckTest(void);
    void _resetTest(void);
    void _listTest(void);
60
    void _downloadTest(void);
Don Gagne's avatar
Don Gagne committed
61
    
62 63
    // Connected to QGCUASFileManager listEntry signal
    void listEntry(const QString& entry);
Don Gagne's avatar
Don Gagne committed
64 65
    
private:
66 67
    void _validateFileContents(const QString& filePath, uint8_t length);

Don Gagne's avatar
Don Gagne committed
68
    enum {
69
        listEntrySignalIndex = 0,
70
        listCompleteSignalIndex,
71 72 73
        downloadFileLengthSignalIndex,
        downloadFileCompleteSignalIndex,
        errorMessageSignalIndex,
Don Gagne's avatar
Don Gagne committed
74 75 76 77
        maxSignalIndex
    };
    
    enum {
78
        listEntrySignalMask =               1 << listEntrySignalIndex,
79
        listCompleteSignalMask =            1 << listCompleteSignalIndex,
80 81 82
        downloadFileLengthSignalMask =      1 << downloadFileLengthSignalIndex,
        downloadFileCompleteSignalMask =    1 << downloadFileCompleteSignalIndex,
        errorMessageSignalMask =            1 << errorMessageSignalIndex,
Don Gagne's avatar
Don Gagne committed
83
    };
84 85 86 87

    static const uint8_t    _systemIdQGC = 255;
    static const uint8_t    _systemIdServer = 128;

88
    MockUAS*                _mockUAS;
Don Gagne's avatar
Don Gagne committed
89 90 91 92 93 94 95 96
    MockMavlinkFileServer   _mockFileServer;
    
    QGCUASFileManager*  _fileManager;

    MultiSignalSpy*     _multiSpy;
    static const size_t _cSignals = maxSignalIndex;
    const char*         _rgSignals[_cSignals];
    
97 98 99 100
    /// @brief This is the amount of time to wait to allow the FileManager enough time to timeout waiting for an Ack.
    /// As such it must be larger than the Ack Timeout used by the FileManager.
    static const int _ackTimerTimeoutMsecs = QGCUASFileManager::ackTimerTimeoutMsecs * 2;
    
Don Gagne's avatar
Don Gagne committed
101 102 103 104
    QStringList _fileListReceived;
};

#endif