MainWindowTest.cc 2.91 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
#include "QGCToolBar.h"
31 32
#include "MockLink.h"
#include "QGCMessageBox.h"
Don Gagne's avatar
Don Gagne committed
33 34 35 36 37 38 39 40 41 42 43

UT_REGISTER_TEST(MainWindowTest)

MainWindowTest::MainWindowTest(void)
{
    
}

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

    _mainWindow = MainWindow::_create(NULL, MainWindow::CUSTOM_MODE_PX4);
    Q_CHECK_PTR(_mainWindow);
Don Gagne's avatar
Don Gagne committed
47 48 49 50
}

void MainWindowTest::cleanup(void)
{
51 52 53
    _mainWindow->close();
    delete _mainWindow;
    
Don Gagne's avatar
Don Gagne committed
54 55 56
    UnitTest::cleanup();
}

57
void MainWindowTest::_clickThrough_test(void)
Don Gagne's avatar
Don Gagne committed
58
{
59 60
    QGCToolBar* toolbar = _mainWindow->findChild<QGCToolBar*>();
    Q_ASSERT(toolbar);
Don Gagne's avatar
Don Gagne committed
61
    
62 63 64 65 66 67 68
    QList<QToolButton*> buttons = toolbar->findChildren<QToolButton*>();
    foreach(QToolButton* button, buttons) {
        if (!button->menu()) {
            QTest::mouseClick(button, Qt::LeftButton);
            QTest::qWait(1000);
        }
    }
Don Gagne's avatar
Don Gagne committed
69 70
    
}
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95

void MainWindowTest::_connectWindowClose_test(void)
{
    LinkManager* linkMgr = LinkManager::instance();
    Q_CHECK_PTR(linkMgr);
    
    MockLink* link = new MockLink();
    Q_CHECK_PTR(link);
    // FIXME: LinkManager/MainWindow needs to be re-architected so that you don't have to addLink to MainWindow to get things to work
    _mainWindow->addLink(link);
    linkMgr->connectLink(link);
    QTest::qWait(5000); // Give enough time for UI to settle and heartbeats to go through
    
    // On MainWindow close we should get a message box telling the user to disconnect first
    setExpectedMessageBox(QGCMessageBox::Ok);
    _mainWindow->close();
    QTest::qWait(1000); // Need to allow signals to move between threads    
    checkExpectedMessageBox();

    // We are going to disconnect the link which is going to pop a save file dialog
    setExpectedFileDialog(getSaveFileName, QStringList());
    linkMgr->disconnectLink(link);
    QTest::qWait(1000); // Need to allow signals to move between threads
    checkExpectedFileDialog();
}