SetupViewTest.cc 3.42 KB
Newer Older
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
///     @author Don Gagne <don@thegagnes.com>

#include "SetupViewTest.h"
#include "MockLink.h"
#include "QGCMessageBox.h"
30 31 32
#include "SetupView.h"
#include "UASManager.h"
#include "AutoPilotPluginManager.h"
33

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

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

void SetupViewTest::init(void)
{
    UnitTest::init();

    _mainWindow = MainWindow::_create(NULL);
    Q_CHECK_PTR(_mainWindow);
Don Gagne's avatar
Don Gagne committed
49 50 51
    
    _mainToolBar = _mainWindow->getMainToolBar();
    Q_ASSERT(_mainToolBar);
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
}

void SetupViewTest::cleanup(void)
{
    _mainWindow->close();
    delete _mainWindow;
    
    UnitTest::cleanup();
}

void SetupViewTest::_clickThrough_test(void)
{
    LinkManager* linkMgr = LinkManager::instance();
    Q_CHECK_PTR(linkMgr);
    
    MockLink* link = new MockLink();
    Q_CHECK_PTR(link);
    link->setAutopilotType(MAV_AUTOPILOT_PX4);
70
    LinkManager::instance()->_addLink(link);
71 72 73
    linkMgr->connectLink(link);
    QTest::qWait(5000); // Give enough time for UI to settle and heartbeats to go through
    
Don Gagne's avatar
Don Gagne committed
74 75
    // Switch to the Setup view
    _mainToolBar->onSetupView();
76
    QTest::qWait(1000);
Don Gagne's avatar
Don Gagne committed
77
    
78 79 80 81 82 83 84 85 86 87 88 89 90 91
    MainWindow* mainWindow = MainWindow::instance();
    Q_ASSERT(mainWindow);
    QWidget* setupViewWidget = mainWindow->getCurrentViewWidget();
    Q_ASSERT(setupViewWidget);
    SetupView* setupView = qobject_cast<SetupView*>(setupViewWidget);
    Q_ASSERT(setupView);

    // Click through fixed buttons
    setupView->firmwareButtonClicked();
    QTest::qWait(1000);
    setupView->parametersButtonClicked();
    QTest::qWait(1000);
    setupView->summaryButtonClicked();
    QTest::qWait(1000);
92
    
93 94 95 96 97 98 99 100 101 102
    // Click through component buttons
    UASInterface* uas = UASManager::instance()->getActiveUAS();
    Q_ASSERT(uas);
    AutoPilotPlugin* autopilot = AutoPilotPluginManager::instance()->getInstanceForAutoPilotPlugin(uas);
    Q_ASSERT(autopilot);
    const QVariantList& components = autopilot->vehicleComponents();
    foreach(QVariant varComponent, components) {
        setupView->setupButtonClicked(varComponent);
    }

103 104 105 106 107 108 109 110 111 112 113
    // On MainWindow close we should get a message box telling the user to disconnect first. Disconnect will then pop
    // the log file save dialog.
    
    setExpectedMessageBox(QGCMessageBox::Yes);
    setExpectedFileDialog(getSaveFileName, QStringList());
    
    _mainWindow->close();
    QTest::qWait(1000); // Need to allow signals to move between threads
    checkExpectedMessageBox();
    checkExpectedFileDialog();
}