MainWindowTest.cc 3.06 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/>.
 
 ======================================================================*/

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

#include "MainWindowTest.h"
30 31
#include "MockLink.h"
#include "QGCMessageBox.h"
32
#include "MultiVehicleManager.h"
Don Gagne's avatar
Don Gagne committed
33

Don Gagne's avatar
Don Gagne committed
34
UT_REGISTER_TEST(MainWindowTest)
Don Gagne's avatar
Don Gagne committed
35

Don Gagne's avatar
Don Gagne committed
36 37 38
MainWindowTest::MainWindowTest(void) :
    _mainWindow(NULL),
    _mainToolBar(NULL)
Don Gagne's avatar
Don Gagne committed
39 40 41 42 43 44 45
{
    
}

void MainWindowTest::init(void)
{
    UnitTest::init();
46

47
    _mainWindow = MainWindow::_create(NULL);
48
    Q_CHECK_PTR(_mainWindow);
Don Gagne's avatar
Don Gagne committed
49 50 51
    
    _mainToolBar = _mainWindow->getMainToolBar();
    Q_ASSERT(_mainToolBar);
Don Gagne's avatar
Don Gagne committed
52 53 54 55
}

void MainWindowTest::cleanup(void)
{
56 57 58
    _mainWindow->close();
    delete _mainWindow;
    
Don Gagne's avatar
Don Gagne committed
59 60 61
    UnitTest::cleanup();
}

62
void MainWindowTest::_connectWindowClose_test(MAV_AUTOPILOT autopilot)
63 64 65 66 67 68
{
    LinkManager* linkMgr = LinkManager::instance();
    Q_CHECK_PTR(linkMgr);
    
    MockLink* link = new MockLink();
    Q_CHECK_PTR(link);
69
    link->setAutopilotType(autopilot);
70
    LinkManager::instance()->_addLink(link);
71
    
72
    linkMgr->connectLink(link);
Don Gagne's avatar
Don Gagne committed
73
    
74
    // Wait for the Vehicle to work it's way through the various threads
Don Gagne's avatar
Don Gagne committed
75
    
76 77
    QSignalSpy spyVehicle(MultiVehicleManager::instance(), SIGNAL(activeVehicleChanged(Vehicle*)));
    QCOMPARE(spyVehicle.wait(5000), true);
78
    
Don Gagne's avatar
Don Gagne committed
79 80 81
    // Cycle through all the top level views
    
    _mainToolBar->onSetupView();
Don Gagne's avatar
Don Gagne committed
82
    QTest::qWait(200);
Don Gagne's avatar
Don Gagne committed
83
    _mainToolBar->onPlanView();
Don Gagne's avatar
Don Gagne committed
84
    QTest::qWait(200);
Don Gagne's avatar
Don Gagne committed
85
    _mainToolBar->onFlyView();
Don Gagne's avatar
Don Gagne committed
86
    QTest::qWait(200);
Don Gagne's avatar
Don Gagne committed
87
    _mainToolBar->onAnalyzeView();
Don Gagne's avatar
Don Gagne committed
88
    QTest::qWait(200);
89
    
90 91
    // On MainWindow close we should get a message box telling the user to disconnect first. Cancel should do nothing.
    setExpectedMessageBox(QGCMessageBox::Cancel);
92 93 94 95 96 97 98
    _mainWindow->close();
    QTest::qWait(1000); // Need to allow signals to move between threads    
    checkExpectedMessageBox();

    linkMgr->disconnectLink(link);
    QTest::qWait(1000); // Need to allow signals to move between threads
}
99 100 101 102 103 104 105 106

void MainWindowTest::_connectWindowClosePX4_test(void) {
    _connectWindowClose_test(MAV_AUTOPILOT_PX4);
}

void MainWindowTest::_connectWindowCloseGeneric_test(void) {
    _connectWindowClose_test(MAV_AUTOPILOT_ARDUPILOTMEGA);
}