qtesteventloop.h

Go to the documentation of this file.

// Copyright (C) 2021 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
 
#ifndef QTESTEVENTLOOP_H
#define QTESTEVENTLOOP_H
 
#include <QtTest/qttestglobal.h>
#include <QtTest/qtestcase.h>
 
#include <QtCore/qbasictimer.h>
#include <QtCore/qcoreapplication.h>
#include <QtCore/qeventloop.h>
#include <QtCore/qobject.h>
#include <QtCore/qpointer.h>
#include <QtCore/qthread.h>
 
QT_BEGIN_NAMESPACE
 
 
class Q_TESTLIB_EXPORT QTestEventLoop : public QObject
{
    Q_OBJECT
 
public:
    QTestEventLoop(QObject *parent = nullptr)
        : QObject(parent), _timeout(false)
    {}
 
    void enterLoopMSecs(int ms) { enterLoop(std::chrono::milliseconds{ms}); }
    void enterLoop(int secs) { enterLoop(std::chrono::seconds{secs}); }
    inline void enterLoop(std::chrono::milliseconds msecs);
 
    inline void changeInterval(int secs)
    { changeInterval(std::chrono::seconds{secs}); }
 
    void changeInterval(std::chrono::nanoseconds nsecs)
    { timer.start(nsecs, this); }
 
    inline bool timeout() const
    { return _timeout; }
 
    inline static QTestEventLoop &instance()
    {
        Q_CONSTINIT static QPointer<QTestEventLoop> testLoop;
        if (testLoop.isNull())
            testLoop = new QTestEventLoop(QCoreApplication::instance());
        return *static_cast<QTestEventLoop *>(testLoop);
    }
 
public Q_SLOTS:
    inline void exitLoop();
 
protected:
    inline void timerEvent(QTimerEvent *e) override;
 
private:
    QEventLoop *loop = nullptr;
    QBasicTimer timer;
    uint _timeout   :1;
    Q_DECL_UNUSED_MEMBER uint reserved   :31;
};
 
inline void QTestEventLoop::enterLoop(std::chrono::milliseconds msecs)
{
    Q_ASSERT(!loop);
    _timeout = false;
 
    if (QTest::runningTest() && QTest::currentTestResolved())
        return;
 
    using namespace std::chrono_literals;
    QEventLoop l;
    // if tests want to measure sub-second precision, use a precise timer
    timer.start(msecs, msecs < 1s ? Qt::PreciseTimer : Qt::CoarseTimer, this);
 
    loop = &l;
    l.exec();
    loop = nullptr;
}
 
inline void QTestEventLoop::exitLoop()
{
    if (QThread *thr = thread(); thr && !thr->isCurrentThread()) {
        QMetaObject::invokeMethod(this, "exitLoop", Qt::QueuedConnection);
        return;
    }
 
    timer.stop();
 
    if (loop)
        loop->exit();
}
 
inline void QTestEventLoop::timerEvent(QTimerEvent *e)
{
    if (e->id() != timer.id())
        return;
    _timeout = true;
    exitLoop();
}
 
QT_END_NAMESPACE
 
#endif