diff options
author | Lars Knoll <lars.knoll@nokia.com> | 2009-03-23 09:18:55 (GMT) |
---|---|---|
committer | Simon Hausmann <simon.hausmann@nokia.com> | 2009-03-23 09:18:55 (GMT) |
commit | e5fcad302d86d316390c6b0f62759a067313e8a9 (patch) | |
tree | c2afbf6f1066b6ce261f14341cf6d310e5595bc1 /tests/auto/qsqlthread | |
download | Qt-e5fcad302d86d316390c6b0f62759a067313e8a9.zip Qt-e5fcad302d86d316390c6b0f62759a067313e8a9.tar.gz Qt-e5fcad302d86d316390c6b0f62759a067313e8a9.tar.bz2 |
Long live Qt 4.5!
Diffstat (limited to 'tests/auto/qsqlthread')
-rw-r--r-- | tests/auto/qsqlthread/.gitignore | 1 | ||||
-rw-r--r-- | tests/auto/qsqlthread/qsqlthread.pro | 14 | ||||
-rw-r--r-- | tests/auto/qsqlthread/tst_qsqlthread.cpp | 524 |
3 files changed, 539 insertions, 0 deletions
diff --git a/tests/auto/qsqlthread/.gitignore b/tests/auto/qsqlthread/.gitignore new file mode 100644 index 0000000..4108739 --- /dev/null +++ b/tests/auto/qsqlthread/.gitignore @@ -0,0 +1 @@ +tst_qsqlthread diff --git a/tests/auto/qsqlthread/qsqlthread.pro b/tests/auto/qsqlthread/qsqlthread.pro new file mode 100644 index 0000000..23aa598 --- /dev/null +++ b/tests/auto/qsqlthread/qsqlthread.pro @@ -0,0 +1,14 @@ +load(qttest_p4) +SOURCES += tst_qsqlthread.cpp + +QT = core sql + + +wince*: { + plugFiles.sources = ../../../plugins/sqldrivers + plugFiles.path = . + DEPLOYMENT += plugFiles +} else { + win32:LIBS += -lws2_32 +} + diff --git a/tests/auto/qsqlthread/tst_qsqlthread.cpp b/tests/auto/qsqlthread/tst_qsqlthread.cpp new file mode 100644 index 0000000..94e0f98 --- /dev/null +++ b/tests/auto/qsqlthread/tst_qsqlthread.cpp @@ -0,0 +1,524 @@ +/**************************************************************************** +** +** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). +** Contact: Qt Software Information (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 either Technology Preview License Agreement or the +** Beta Release License Agreement. +** +** 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.0, included in the file LGPL_EXCEPTION.txt in this +** package. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 3.0 as published by the Free Software +** Foundation and appearing in the file LICENSE.GPL included in the +** packaging of this file. Please review the following information to +** ensure the GNU General Public License version 3.0 requirements will be +** met: http://www.gnu.org/copyleft/gpl.html. +** +** If you are unsure which license is appropriate for your use, please +** contact the sales department at qt-sales@nokia.com. +** $QT_END_LICENSE$ +** +****************************************************************************/ + + +#include <QtTest/QtTest> + + +#include "../qsqldatabase/tst_databases.h" + +#include <QtCore> +#include <QtSql> +#include "qdebug.h" + +#ifdef Q_OS_LINUX +#include <pthread.h> +#endif + +// set this define if Oracle is built with threading support +//#define QOCI_THREADED + +class tst_QSqlThread : public QObject +{ + Q_OBJECT + +public: + tst_QSqlThread(); + virtual ~tst_QSqlThread(); + + + void dropTestTables(); + void createTestTables(); + void recreateTestTables(); + void repopulateTestTables(); + + void generic_data(); + tst_Databases dbs; + +public slots: + void initTestCase(); + void cleanupTestCase(); + void init(); + void cleanup(); + +protected slots: + void threadFinished() { ++threadFinishedCount; } + +private slots: + void simpleThreading_data() { generic_data(); } + void simpleThreading(); + void readWriteThreading_data() { generic_data(); } + void readWriteThreading(); + void readFromSingleConnection_data() { generic_data(); } + void readFromSingleConnection(); + void readWriteFromSingleConnection_data() { generic_data(); } + void readWriteFromSingleConnection(); + void preparedReadWriteFromSingleConnection_data() { generic_data(); } + void preparedReadWriteFromSingleConnection(); + void transactionsFromSingleConnection_data() { generic_data(); } + void transactionsFromSingleConnection(); + +private: + int threadFinishedCount; +}; + +static QBasicAtomicInt counter; + +class QtTestSqlThread : public QThread +{ + Q_OBJECT +public: + QtTestSqlThread(const QSqlDatabase &aDb, QObject *parent = 0) + : QThread(parent), sourceDb(aDb) {} + + void runHelper(const QString &dbName) + { + QSqlDatabase db = QSqlDatabase::cloneDatabase(sourceDb, dbName); + QVERIFY_SQL(db, open()); + + int sum = 0; + QSqlQuery q("select id from " + qTableName("test"), db); + QVERIFY_SQL(q, isActive()); + while (q.next()) + sum += q.value(0).toInt(); + QCOMPARE(sum, 6); + q.clear(); + } + + void run() + { + QString dbName = QString("QThreadDb%1").arg((size_t)currentThreadId()); + runHelper(dbName); + + QSqlDatabase::database(dbName).close(); + QSqlDatabase::removeDatabase(dbName); + } + +private: + QSqlDatabase sourceDb; +}; + +enum { ProdConIterations = 10 }; + +class SqlProducer: public QThread +{ + Q_OBJECT +public: + SqlProducer(const QSqlDatabase &aDb, QObject *parent = 0) + : QThread(parent), sourceDb(aDb) {} + + void runHelper(const QString &dbName) + { + QSqlDatabase db = QSqlDatabase::cloneDatabase(sourceDb, dbName); + QVERIFY_SQL(db, open()); + QSqlQuery q(db); + QVERIFY_SQL(q, prepare("insert into " + qTableName("test") + " values (?, ?, ?)")); + int id = 10; + for (int i = 0; i < ProdConIterations; ++i) { + q.bindValue(0, ++id); + q.bindValue(1, "threaddy"); + q.bindValue(2, 10); + QVERIFY_SQL(q, exec()); +#ifdef Q_OS_LINUX + pthread_yield(); +#endif + } + } + + void run() + { + QString dbName = QString("Producer%1").arg((size_t)currentThreadId()); + runHelper(dbName); + QSqlDatabase::database(dbName).close(); + QSqlDatabase::removeDatabase(dbName); + } +private: + QSqlDatabase sourceDb; +}; + +class SqlConsumer: public QThread +{ + Q_OBJECT + +public: + SqlConsumer(const QSqlDatabase &aDb, QObject *parent = 0) + : QThread(parent), sourceDb(aDb) {} + + void runHelper(const QString &dbName) + { + QSqlDatabase db = QSqlDatabase::cloneDatabase(sourceDb, dbName); + QVERIFY_SQL(db, open()); + QSqlQuery q1(db), q2(db); + QVERIFY_SQL(q2, prepare("delete from " + qTableName("test") + " where id = :id")); + + for (int i = 0; i < ProdConIterations; ++i) { + QVERIFY_SQL(q1, exec("select max(id) from " + qTableName("test"))); + q1.first(); + q2.bindValue("id", q1.value(0)); + q1.clear(); + QVERIFY_SQL(q2, exec()); +#ifdef Q_OS_LINUX + pthread_yield(); +#endif + } + } + + void run() + { + QString dbName = QString("Consumer%1").arg((size_t)currentThreadId()); + runHelper(dbName); + QSqlDatabase::database(dbName).close(); + QSqlDatabase::removeDatabase(dbName); + } + +private: + QSqlDatabase sourceDb; +}; + +class SqlThread: public QThread +{ + Q_OBJECT + +public: + enum Mode { SimpleReading, PreparedReading, SimpleWriting, PreparedWriting }; + + SqlThread(Mode m, const QSqlDatabase &db, QObject *parent = 0) + : QThread(parent), sourceDb(db), mode(m) {} + + void run() + { + QSqlDatabase &db = sourceDb; + switch (mode) { + case SimpleReading: { + // Executes a Query for reading, iterates over the first 4 results + QSqlQuery q(sourceDb); + for (int j = 0; j < ProdConIterations; ++j) { + QVERIFY_SQL(q, exec("select id,name from " + qTableName("test") + " order by id")); + for (int i = 1; i < 4; ++i) { + QVERIFY_SQL(q, next()); + QCOMPARE(q.value(0).toInt(), i); + } + } + break; } + case SimpleWriting: { + // Executes a query for writing (appends a new row) + QSqlQuery q(sourceDb); + for (int j = 0; j < ProdConIterations; ++j) { + QVERIFY_SQL(q, exec(QString("insert into " + qTableName("test") + + " (id, name) values(%1, '%2')") + .arg(counter.fetchAndAddRelaxed(1)).arg("Robert"))); + } + break; } + case PreparedReading: { + // Prepares a query for reading and iterates over the results + QSqlQuery q(sourceDb); + QVERIFY_SQL(q, prepare("select id, name from " + qTableName("test") + " where id = ?")); + for (int j = 0; j < ProdConIterations; ++j) { + q.addBindValue(j % 3 + 1); + QVERIFY_SQL(q, exec()); + QVERIFY_SQL(q, next()); + QCOMPARE(q.value(0).toInt(), j % 3 + 1); + } + break; } + case PreparedWriting: { + QSqlQuery q(sourceDb); + QVERIFY_SQL(q, prepare("insert into " + qTableName("test") + " (id, name) " + "values(?, ?)")); + for (int i = 0; i < ProdConIterations; ++i) { + q.addBindValue(counter.fetchAndAddRelaxed(1)); + q.addBindValue("Robert"); + QVERIFY_SQL(q, exec()); + } + break; } + } + } + +private: + QSqlDatabase sourceDb; + Mode mode; +}; + + +tst_QSqlThread::tst_QSqlThread() + : threadFinishedCount(0) +{ +} + +tst_QSqlThread::~tst_QSqlThread() +{ +} + +void tst_QSqlThread::generic_data() +{ + if (dbs.fillTestTable() == 0) + QSKIP("No database drivers are available in this Qt configuration", SkipAll); +} + +void tst_QSqlThread::dropTestTables() +{ + for (int i = 0; i < dbs.dbNames.count(); ++i) { + QSqlDatabase db = QSqlDatabase::database(dbs.dbNames.at(i)); + QSqlQuery q(db); + + tst_Databases::safeDropTables(db, QStringList() << qTableName("test") << qTableName("test2") << qTableName("emptytable")); + } +} + +void tst_QSqlThread::createTestTables() +{ + for (int i = 0; i < dbs.dbNames.count(); ++i) { + QSqlDatabase db = QSqlDatabase::database(dbs.dbNames.at(i)); + QSqlQuery q(db); + + QVERIFY_SQL(q, exec("create table " + qTableName("test") + + "(id int NOT NULL primary key, name varchar(20), title int)")); + + QVERIFY_SQL(q, exec("create table " + qTableName("test2") + + "(id int NOT NULL primary key, title varchar(20))")); + + QVERIFY_SQL(q, exec("create table " + qTableName("emptytable") + + "(id int NOT NULL primary key)")); + } +} + +void tst_QSqlThread::repopulateTestTables() +{ + for (int i = 0; i < dbs.dbNames.count(); ++i) { + QSqlDatabase db = QSqlDatabase::database(dbs.dbNames.at(i)); + QSqlQuery q(db); + + QVERIFY_SQL(q, exec("delete from " + qTableName("test"))); + QVERIFY_SQL(q, exec("insert into " + qTableName("test") + " values(1, 'harry', 1)")); + QVERIFY_SQL(q, exec("insert into " + qTableName("test") + " values(2, 'trond', 2)")); + QVERIFY_SQL(q, exec("insert into " + qTableName("test") + " values(3, 'vohi', 3)")); + + QVERIFY_SQL(q, exec("delete from " + qTableName("test2"))); + QVERIFY_SQL(q, exec("insert into " + qTableName("test2") + " values(1, 'herr')")); + QVERIFY_SQL(q, exec("insert into " + qTableName("test2") + " values(2, 'mister')")); + } +} + +void tst_QSqlThread::recreateTestTables() +{ + dropTestTables(); + createTestTables(); + repopulateTestTables(); +} + +void tst_QSqlThread::initTestCase() +{ + dbs.open(); + recreateTestTables(); +} + +void tst_QSqlThread::cleanupTestCase() +{ + dropTestTables(); + dbs.close(); +} + +void tst_QSqlThread::init() +{ + threadFinishedCount = 0; + counter = 4; +} + +void tst_QSqlThread::cleanup() +{ +// repopulateTestTables(); +} + +// This test creates two threads that clone their db connection and read +// from it +void tst_QSqlThread::simpleThreading() +{ + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + QtTestSqlThread t1(db); + QtTestSqlThread t2(db); + + connect(&t1, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + connect(&t2, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + + t1.start(); + t2.start(); + + while (threadFinishedCount < 2) + QTest::qWait(100); +} + +// This test creates two threads that clone their db connection and read +// or write +void tst_QSqlThread::readWriteThreading() +{ + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + SqlProducer producer(db); + SqlConsumer consumer(db); + + connect(&producer, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + connect(&consumer, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + + producer.start(); + consumer.start(); + + while (threadFinishedCount < 2) + QTest::qWait(100); +} + +// run with n threads in parallel. Change this constant to hammer the poor DB server even more +static const int maxThreadCount = 4; + +void tst_QSqlThread::readFromSingleConnection() +{ +#ifdef QOCI_THREADED + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + QObject cleanupHelper; // make sure the threads die when we exit the scope + for (int i = 0; i < maxThreadCount; ++i) { + SqlThread *reader = new SqlThread(SqlThread::SimpleReading, db, &cleanupHelper); + connect(reader, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + reader->start(); + } + + while (threadFinishedCount < maxThreadCount) + QTest::qWait(100); +#endif +} + +void tst_QSqlThread::readWriteFromSingleConnection() +{ +#ifdef QOCI_THREADED + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + QObject cleanupHelper; + for (int i = 0; i < maxThreadCount; ++i) { + SqlThread *reader = new SqlThread(SqlThread::SimpleReading, db, &cleanupHelper); + connect(reader, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + reader->start(); + + SqlThread *writer = new SqlThread(SqlThread::SimpleWriting, db, &cleanupHelper); + connect(writer, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + writer->start(); + } + + while (threadFinishedCount < maxThreadCount * 2) + QTest::qWait(100); +#endif +} + +void tst_QSqlThread::preparedReadWriteFromSingleConnection() +{ +#ifdef QOCI_THREADED + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + QObject cleanupHelper; + for (int i = 0; i < maxThreadCount; ++i) { + SqlThread *reader = new SqlThread(SqlThread::PreparedReading, db, &cleanupHelper); + connect(reader, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + reader->start(); + + SqlThread *writer = new SqlThread(SqlThread::PreparedWriting, db, &cleanupHelper); + connect(writer, SIGNAL(finished()), this, SLOT(threadFinished()), Qt::QueuedConnection); + writer->start(); + } + + while (threadFinishedCount < maxThreadCount * 2) + QTest::qWait(100); +#endif +} + +void tst_QSqlThread::transactionsFromSingleConnection() +{ +#ifdef QOCI_THREADED + QFETCH(QString, dbName); + QSqlDatabase db = QSqlDatabase::database(dbName); + CHECK_DATABASE(db); + + if (db.databaseName() == ":memory:") + QSKIP("does not work with in-memory databases", SkipSingle); + + // start and commit a transaction + QVERIFY_SQL(db, db.transaction()); + preparedReadWriteFromSingleConnection(); // read and write from multiple threads + if (QTest::currentTestFailed()) + return; + QVERIFY_SQL(db, db.commit()); + + // reset test environment + threadFinishedCount = 0; + + // start and roll back a transaction + QVERIFY_SQL(db, db.transaction()); + preparedReadWriteFromSingleConnection(); // read and write from multiple threads + if (QTest::currentTestFailed()) + return; + QVERIFY_SQL(db, db.rollback()); +#endif +} + +QTEST_MAIN(tst_QSqlThread) +#include "tst_qsqlthread.moc" |