--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/auto/qsqlthread/tst_qsqlthread.cpp Mon Jan 11 14:00:40 2010 +0000
@@ -0,0 +1,530 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 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 "../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(const QString &engine=QString());
+ 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(const QString& engine)
+{
+ if ( dbs.fillTestTable(engine) == 0 ) {
+ if(engine.isEmpty())
+ QSKIP( "No database drivers are available in this Qt configuration", SkipAll );
+ else
+ QSKIP( (QString("No database drivers of type %1 are available in this Qt configuration").arg(engine)).toLocal8Bit(), 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);
+ else if (tst_Databases::isMSAccess(db))
+ QSKIP("does not work with MS Access 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"