/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the test suite of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtTest/QtTest>

#include <qcoreapplication.h>

#include <qdatetime.h>

#include <qmutex.h>

#include <qthread.h>

#include <qtimer.h>

#include <qwaitcondition.h>

#include <qdebug.h>

#ifdef Q_OS_UNIX

#include <pthread.h>

#endif

#if defined(Q_OS_WINCE)

#include <windows.h>

#elif defined(Q_OS_WIN)

#include <process.h>

#include <windows.h>

#endif

//TESTED_CLASS=

//TESTED_FILES=

class tst_QThread : public QObject

{

    Q_OBJECT

public:

    tst_QThread();

    virtual ~tst_QThread();

private slots:

    void currentThreadId();

    void currentThread();

    void idealThreadCount();

    void isFinished();

    void isRunning();

    void setPriority();

    void priority();

    void setStackSize();

    void stackSize();

    void exit();

    void start();

    void terminate();

    void quit();

    void wait();

    void started();

    void finished();

    void terminated();

    void run();

    void exec();

    void setTerminationEnabled();

    void sleep();

    void msleep();

    void usleep();

    void nativeThreadAdoption();

    void adoptedThreadAffinity();

    void adoptedThreadSetPriority();

    void adoptedThreadExit();

    void adoptedThreadExec();

    void adoptedThreadFinished();

    void adoptMultipleThreads();

    void stressTest();

};

enum { one_minute = 60 * 1000, five_minutes = 5 * one_minute };

class SignalRecorder : public QObject

{

    Q_OBJECT

public:

    QAtomicInt activationCount;

    inline SignalRecorder()

    { activationCount = 0; }

    bool wasActivated()

    { return activationCount > 0; }

public slots:

    void slot();

};

void SignalRecorder::slot()

{ activationCount.ref(); }

class Current_Thread : public QThread

{

public:

    Qt::HANDLE id;

    QThread *thread;

    void run()

    {

        id = QThread::currentThreadId();

        thread = QThread::currentThread();

    }

};

class Simple_Thread : public QThread

{

public:

    QMutex mutex;

    QWaitCondition cond;

    void run()

    {

        QMutexLocker locker(&mutex);

        cond.wakeOne();

    }

};

class Exit_Object : public QObject

{

    Q_OBJECT

public:

    QThread *thread;

    int code;

public slots:

    void slot()

    { thread->exit(code); }

};

class Exit_Thread : public Simple_Thread

{

public:

    Exit_Object *object;

    int code;

    int result;

    void run()

    {

        Simple_Thread::run();

        if (object) {

            object->thread = this;

            object->code = code;

            QTimer::singleShot(100, object, SLOT(slot()));

        }

        result = exec();

    }

};

class Terminate_Thread : public Simple_Thread

{

public:

    void run()

    {

        setTerminationEnabled(false);

        {

            QMutexLocker locker(&mutex);

            cond.wakeOne();

            cond.wait(&mutex, five_minutes);

        }

        setTerminationEnabled(true);

        Q_ASSERT_X(false, "tst_QThread", "test case hung");

    }

};

class Quit_Object : public QObject

{

    Q_OBJECT

public:

    QThread *thread;

public slots:

    void slot()

    { thread->quit(); }

};

class Quit_Thread : public Simple_Thread

{

public:

    Quit_Object *object;

    int result;

    void run()

    {

        Simple_Thread::run();

        if (object) {

            object->thread = this;

            QTimer::singleShot(100, object, SLOT(slot()));

        }

        result = exec();

    }

};

class Sleep_Thread : public Simple_Thread

{

public:

    enum SleepType { Second, Millisecond, Microsecond };

    SleepType sleepType;

    int interval;

    int elapsed; // result, in *MILLISECONDS*

    void run()

    {

        QMutexLocker locker(&mutex);

        elapsed = 0;

        QTime time;

        time.start();

        switch (sleepType) {

        case Second:

            sleep(interval);

            break;

        case Millisecond:

            msleep(interval);

            break;

        case Microsecond:

            usleep(interval);

            break;

        }

        elapsed = time.elapsed();

        cond.wakeOne();

    }

};

tst_QThread::tst_QThread()

{

}

tst_QThread::~tst_QThread()

{

}

void tst_QThread::currentThreadId()

{

    Current_Thread thread;

    thread.id = 0;

    thread.thread = 0;

    thread.start();

    QVERIFY(thread.wait(five_minutes));

    QVERIFY(thread.id != 0);

    QVERIFY(thread.id != QThread::currentThreadId());

}

void tst_QThread::currentThread()

{

    QVERIFY(QThread::currentThread() != 0);

    QCOMPARE(QThread::currentThread(), thread());

    Current_Thread thread;

    thread.id = 0;

    thread.thread = 0;

    thread.start();

    QVERIFY(thread.wait(five_minutes));

    QCOMPARE(thread.thread, (QThread *)&thread);

}

void tst_QThread::idealThreadCount()

{

    QVERIFY(QThread::idealThreadCount() > 0);

    qDebug() << "Available cpu cores:" << QThread::idealThreadCount();

}

void tst_QThread::isFinished()

{

    Simple_Thread thread;

    QVERIFY(!thread.isFinished());

    QMutexLocker locker(&thread.mutex);

    thread.start();

    QVERIFY(!thread.isFinished());

    thread.cond.wait(locker.mutex());

    QVERIFY(thread.wait(five_minutes));

    QVERIFY(thread.isFinished());

}

void tst_QThread::isRunning()

{

    Simple_Thread thread;

    QVERIFY(!thread.isRunning());

    QMutexLocker locker(&thread.mutex);

    thread.start();

    QVERIFY(thread.isRunning());

    thread.cond.wait(locker.mutex());

    QVERIFY(thread.wait(five_minutes));

    QVERIFY(!thread.isRunning());

}

void tst_QThread::setPriority()

{

    Simple_Thread thread;

    // cannot change the priority, since the thread is not running

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::IdlePriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::LowestPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::LowPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::NormalPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::HighPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::HighestPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::TimeCriticalPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QMutexLocker locker(&thread.mutex);

    thread.start();

    // change the priority of a running thread

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    thread.setPriority(QThread::IdlePriority);

    QCOMPARE(thread.priority(), QThread::IdlePriority);

    thread.setPriority(QThread::LowestPriority);

    QCOMPARE(thread.priority(), QThread::LowestPriority);

    thread.setPriority(QThread::LowPriority);

    QCOMPARE(thread.priority(), QThread::LowPriority);

    thread.setPriority(QThread::NormalPriority);

    QCOMPARE(thread.priority(), QThread::NormalPriority);

    thread.setPriority(QThread::HighPriority);

    QCOMPARE(thread.priority(), QThread::HighPriority);

    thread.setPriority(QThread::HighestPriority);

    QCOMPARE(thread.priority(), QThread::HighestPriority);

    thread.setPriority(QThread::TimeCriticalPriority);

    QCOMPARE(thread.priority(), QThread::TimeCriticalPriority);

    thread.cond.wait(locker.mutex());

    QVERIFY(thread.wait(five_minutes));

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::IdlePriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::LowestPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::LowPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::NormalPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::HighPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::HighestPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

    QTest::ignoreMessage(QtWarningMsg, "QThread::setPriority: Cannot set priority, thread is not running");

    thread.setPriority(QThread::TimeCriticalPriority);

    QCOMPARE(thread.priority(), QThread::InheritPriority);

}

void tst_QThread::priority()

{ DEPENDS_ON("setPriority"); }

void tst_QThread::setStackSize()

{

    Simple_Thread thread;

    QCOMPARE(thread.stackSize(), 0u);

    thread.setStackSize(8192u);

    QCOMPARE(thread.stackSize(), 8192u);

    thread.setStackSize(0u);

    QCOMPARE(thread.stackSize(), 0u);

}

void tst_QThread::stackSize()

{

    DEPENDS_ON("setStackSize");

}

void tst_QThread::exit()

{

    Exit_Thread thread;

    thread.object = new Exit_Object;

    thread.object->moveToThread(&thread);

    thread.code = 42;

    thread.result = 0;

    QVERIFY(!thread.isFinished());

    QVERIFY(!thread.isRunning());

    QMutexLocker locker(&thread.mutex);

    thread.start();

    QVERIFY(thread.isRunning());

    QVERIFY(!thread.isFinished());

    thread.cond.wait(locker.mutex());

    QVERIFY(thread.wait(five_minutes));

    QVERIFY(thread.isFinished());

    QVERIFY(!thread.isRunning());

    QCOMPARE(thread.result, thread.code);

    delete thread.object;

    Exit_Thread thread2;

    thread2.object = 0;

    thread2.code = 53;

    thread2.result = 0;

    QMutexLocker locker2(&thread2.mutex);

    thread2.start();

    thread2.exit(thread2.code);

    thread2.cond.wait(locker2.mutex());

    QVERIFY(thread2.wait(five_minutes));

    QCOMPARE(thread2.result, thread2.code);

}

void tst_QThread::start()

{

    QThread::Priority priorities[] = {

        QThread::IdlePriority,

        QThread::LowestPriority,

        QThread::LowPriority,

        QThread::NormalPriority,

        QThread::HighPriority,

        QThread::HighestPriority,

        QThread::TimeCriticalPriority,

        QThread::InheritPriority

    };

    const int prio_count = sizeof(priorities) / sizeof(QThread::Priority);

    for (int i = 0; i < prio_count; ++i) {

        Simple_Thread thread;

        QVERIFY(!thread.isFinished());

        QVERIFY(!thread.isRunning());

        QMutexLocker locker(&thread.mutex);

        thread.start(priorities[i]);

        QVERIFY(thread.isRunning());

        QVERIFY(!thread.isFinished());

        thread.cond.wait(locker.mutex());

        QVERIFY(thread.wait(five_minutes));

        QVERIFY(thread.isFinished());

        QVERIFY(!thread.isRunning());

    }

}

void tst_QThread::terminate()

{

    Terminate_Thread thread;

    {

        QMutexLocker locker(&thread.mutex);

        thread.start();

        QVERIFY(thread.cond.wait(locker.mutex(), five_minutes));

        thread.terminate();

        thread.cond.wakeOne();

    }

    QVERIFY(thread.wait(five_minutes));

}

void tst_QThread::quit()

{

    Quit_Thread thread;

    thread.object = new Quit_Object;

    thread.object->moveToThread(&thread);

    thread.result = -1;

    QVERIFY(!thread.isFinished());

    QVERIFY(!thread.isRunning());

    QMutexLocker locker(&thread.mutex);

    thread.start();

    QVERIFY(thread.isRunning());

    QVERIFY(!thread.isFinished());

    thread.cond.wait(locker.mutex());

    QVERIFY(thread.wait(five_minutes));

    QVERIFY(thread.isFinished());

    QVERIFY(!thread.isRunning());

    QCOMPARE(thread.result, 0);

    delete thread.object;

    Quit_Thread thread2;

    thread2.object = 0;

    thread2.result = -1;

    QMutexLocker locker2(&thread2.mutex);

    thread2.start();

    thread2.quit();

    thread2.cond.wait(locker2.mutex());

    QVERIFY(thread2.wait(five_minutes));

    QCOMPARE(thread2.result, 0);

}