qt5base-lts/tests/auto/concurrent/qtconcurrenttask/tst_qtconcurrenttask.cpp

197 lines
4.8 KiB
C++
Raw Normal View History

// Copyright (C) 2020 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include <qtconcurrenttask.h>
#include <QTest>
#include <QSemaphore>
#include <random>
class tst_QtConcurrentTask : public QObject
{
Q_OBJECT
private Q_SLOTS:
void taskWithFreeFunction();
void taskWithClassMethod();
void taskWithCallableObject();
void taskWithLambda();
void taskWithArguments();
void useCustomThreadPool();
void setPriority_data();
void setPriority();
void adjustAllSettings();
void ignoreFutureResult();
void withPromise();
};
using namespace QtConcurrent;
void tst_QtConcurrentTask::taskWithFreeFunction()
{
QVariant value(42);
auto result = task(&qvariant_cast<int>)
.withArguments(value)
.spawn()
.result();
QCOMPARE(result, 42);
}
void tst_QtConcurrentTask::taskWithClassMethod()
{
QString result("foobar");
task(&QString::chop).withArguments(&result, 3).spawn().waitForFinished();
QCOMPARE(result, "foo");
}
void tst_QtConcurrentTask::taskWithCallableObject()
{
QCOMPARE(task(std::plus<int>())
.withArguments(40, 2)
.spawn()
.result(),
42);
}
void tst_QtConcurrentTask::taskWithLambda()
{
QCOMPARE(task([]{ return 42; }).spawn().result(), 42);
}
void tst_QtConcurrentTask::taskWithArguments()
{
auto result = task([](int arg1, int arg2){ return arg1 + arg2; })
.withArguments(40, 2)
.spawn()
.result();
QCOMPARE(result, 42);
}
void tst_QtConcurrentTask::useCustomThreadPool()
{
QThreadPool pool;
int result = 0;
task([&]{ result = 42; }).onThreadPool(pool).spawn().waitForFinished();
QCOMPARE(result, 42);
}
void tst_QtConcurrentTask::setPriority_data()
{
QTest::addColumn<bool>("runWithPromise");
QTest::addRow("without promise") << false;
QTest::addRow("with promise") << true;
}
void tst_QtConcurrentTask::setPriority()
{
QFETCH(bool, runWithPromise);
QThreadPool pool;
pool.setMaxThreadCount(1);
QSemaphore sem;
QList<QFuture<void>> futureResults;
futureResults << task([&]{ sem.acquire(); })
.onThreadPool(pool)
.spawn();
const int tasksCount = 10;
QList<int> priorities(tasksCount);
std::iota(priorities.begin(), priorities.end(), 1);
auto seed = std::random_device {}();
std::shuffle(priorities.begin(), priorities.end(), std::default_random_engine(seed));
qDebug() << "Generated priorities list" << priorities << "using seed" << seed;
QList<int> actual;
for (int priority : priorities) {
if (runWithPromise) {
futureResults << task([priority, &actual] (QPromise<void> &) { actual << priority; })
.onThreadPool(pool)
.withPriority(priority)
.spawn();
} else {
futureResults << task([priority, &actual] { actual << priority; })
.onThreadPool(pool)
.withPriority(priority)
.spawn();
}
}
sem.release();
pool.waitForDone();
for (const auto &f : futureResults)
QVERIFY(f.isFinished());
QList<int> expected(priorities);
std::sort(expected.begin(), expected.end(), std::greater<>());
QCOMPARE(actual, expected);
}
void tst_QtConcurrentTask::adjustAllSettings()
{
QThreadPool pool;
pool.setMaxThreadCount(1);
const int priority = 10;
QList<int> result;
auto append = [&](auto &&...args){ (result << ... << args); };
task(std::move(append))
.withArguments(1, 2, 3)
.onThreadPool(pool)
.withPriority(priority)
.spawn()
.waitForFinished();
QCOMPARE(result, QList<int>({ 1, 2, 3 }));
}
void tst_QtConcurrentTask::ignoreFutureResult()
{
QThreadPool pool;
std::atomic_int value = 0;
for (std::size_t i = 0; i < 10; ++i)
task([&value]{ ++value; })
.onThreadPool(pool)
.spawn(FutureResult::Ignore);
pool.waitForDone();
QCOMPARE(value, 10);
}
void incrementWithPromise(QPromise<int> &promise, int i)
{
promise.addResult(i + 1);
}
void return0WithPromise(QPromise<int> &promise)
{
promise.addResult(0);
}
void tst_QtConcurrentTask::withPromise()
{
QCOMPARE(task(&return0WithPromise).spawn().result(), 0);
QCOMPARE(task(&return0WithPromise).withPriority(7).spawn().result(), 0);
QCOMPARE(task(&incrementWithPromise).withArguments(1).spawn().result(), 2);
QCOMPARE(task(&incrementWithPromise).withArguments(1).withPriority(7).spawn().result(), 2);
}
QTEST_MAIN(tst_QtConcurrentTask)
#include "tst_qtconcurrenttask.moc"