Use qScopeGuard() to take care of post-test tidy-up

Doing the tidy-up after the last check has the problem that it gets
skipped if any check fails, as the premature return skips the tidy-up.
In the process, added the missing tidy-up to prematureExec().

Repackage the code for one tidy-up as a named lambda so that the
duplicate can share it with the scope guard. Made some existing scope
guards const, while I was about it, so that the one that isn't -
because it gets dismiss()ed - stand out.

Change-Id: I96d6834d5d7675f15018169a7093b0211db6f8a9
Reviewed-by: Marc Mutz <marc.mutz@qt.io>
This commit is contained in:
Edward Welbourne 2022-02-01 15:17:10 +01:00
parent 9a0cfaf78d
commit 9030ffdffb

View File

@ -860,6 +860,7 @@ void tst_QSqlQuery::storedProceduresIBase()
" x = 42; " " x = 42; "
" y = 'Hello Anders'; " " y = 'Hello Anders'; "
"END" ) ); "END" ) );
const auto tidier = qScopeGuard([&]() { q.exec("drop procedure " + procName); });
QVERIFY_SQL(q, prepare("execute procedure " + procName)); QVERIFY_SQL(q, prepare("execute procedure " + procName));
QVERIFY_SQL( q, exec() ); QVERIFY_SQL( q, exec() );
@ -877,8 +878,6 @@ void tst_QSqlQuery::storedProceduresIBase()
// the second next shall fail // the second next shall fail
QVERIFY( !q.next() ); QVERIFY( !q.next() );
q.exec("drop procedure " + procName);
} }
void tst_QSqlQuery::outValuesDB2() void tst_QSqlQuery::outValuesDB2()
@ -1800,7 +1799,7 @@ void tst_QSqlQuery::writeNull()
}; };
for (const auto &nullableType : nullableTypes.keys()) { for (const auto &nullableType : nullableTypes.keys()) {
auto tableGuard = qScopeGuard([&]{ const auto tableGuard = qScopeGuard([&]{
q.exec("drop table " + tableName); q.exec("drop table " + tableName);
}); });
const QVariant nonNullValue = nullableTypes.value(nullableType); const QVariant nonNullValue = nullableTypes.value(nullableType);
@ -1947,7 +1946,10 @@ void tst_QSqlQuery::precision()
if (dbType == QSqlDriver::Interbase) if (dbType == QSqlDriver::Interbase)
QSKIP("DB unable to store high precision"); QSKIP("DB unable to store high precision");
const auto oldPrecision = db.driver()->numericalPrecisionPolicy(); const auto tidier = qScopeGuard([db, oldPrecision = db.driver()->numericalPrecisionPolicy()]() {
db.driver()->setNumericalPrecisionPolicy(oldPrecision);
});
db.driver()->setNumericalPrecisionPolicy(QSql::HighPrecision); db.driver()->setNumericalPrecisionPolicy(QSql::HighPrecision);
const QString qtest_precision(qTableName("qtest_precision", __FILE__, db)); const QString qtest_precision(qTableName("qtest_precision", __FILE__, db));
static const QLatin1String precStr("1.2345678901234567891"); static const QLatin1String precStr("1.2345678901234567891");
@ -1981,7 +1983,6 @@ void tst_QSqlQuery::precision()
} }
} }
} // SQLITE scope } // SQLITE scope
db.driver()->setNumericalPrecisionPolicy(oldPrecision);
} }
void tst_QSqlQuery::nullResult() void tst_QSqlQuery::nullResult()
@ -2884,6 +2885,8 @@ void tst_QSqlQuery::prematureExec()
if (cut >= 0) if (cut >= 0)
dbName = dbName.sliced(cut + 1); dbName = dbName.sliced(cut + 1);
const auto tidier = qScopeGuard([dbName]() { QSqlDatabase::removeDatabase(dbName); });
// Note: destruction of db needs to happen before we call removeDatabase.
auto db = QSqlDatabase::addDatabase(dbName); auto db = QSqlDatabase::addDatabase(dbName);
QSqlQuery q(db); QSqlQuery q(db);
@ -3031,6 +3034,10 @@ void tst_QSqlQuery::psql_specialFloatValues()
void tst_QSqlQuery::queryOnInvalidDatabase() void tst_QSqlQuery::queryOnInvalidDatabase()
{ {
{ {
const auto tidier = qScopeGuard([]() {
QSqlDatabase::removeDatabase("invalidConnection");
});
// Note: destruction of db needs to happen before we call removeDatabase.
QTest::ignoreMessage( QtWarningMsg, "QSqlDatabase: INVALID driver not loaded" ); QTest::ignoreMessage( QtWarningMsg, "QSqlDatabase: INVALID driver not loaded" );
QSqlDatabase db = QSqlDatabase::addDatabase( "INVALID", "invalidConnection" ); QSqlDatabase db = QSqlDatabase::addDatabase( "INVALID", "invalidConnection" );
QVERIFY2( db.lastError().isValid(), QVERIFY2( db.lastError().isValid(),
@ -3042,8 +3049,6 @@ void tst_QSqlQuery::queryOnInvalidDatabase()
qPrintable( QString( "query.lastError().isValid() should be true!" ) ) ); qPrintable( QString( "query.lastError().isValid() should be true!" ) ) );
} }
QSqlDatabase::removeDatabase( "invalidConnection" );
{ {
QSqlDatabase db = QSqlDatabase::database( "this connection does not exist" ); QSqlDatabase db = QSqlDatabase::database( "this connection does not exist" );
QTest::ignoreMessage( QtWarningMsg, "QSqlQuery::exec: database not open" ); QTest::ignoreMessage( QtWarningMsg, "QSqlQuery::exec: database not open" );
@ -3157,11 +3162,16 @@ void tst_QSqlQuery::sqlite_finish()
QSKIP( "This test requires a database on the filesystem, not in-memory"); QSKIP( "This test requires a database on the filesystem, not in-memory");
{ {
const auto tidier = qScopeGuard([]() {
QSqlDatabase::removeDatabase("sqlite_finish_sqlite");
});
// Note: destruction of db2 needs to happen before we call removeDatabase.
QSqlDatabase db2 = QSqlDatabase::addDatabase( "QSQLITE", "sqlite_finish_sqlite" ); QSqlDatabase db2 = QSqlDatabase::addDatabase( "QSQLITE", "sqlite_finish_sqlite" );
db2.setDatabaseName( db.databaseName() ); db2.setDatabaseName( db.databaseName() );
QVERIFY_SQL( db2, open() ); QVERIFY_SQL( db2, open() );
const QString tableName(qTableName("qtest_lockedtable", __FILE__, db)); const QString tableName(qTableName("qtest_lockedtable", __FILE__, db));
const auto wrapup = qScopeGuard([&]() { tst_Databases::safeDropTable(db, tableName); });
QSqlQuery q( db ); QSqlQuery q( db );
tst_Databases::safeDropTable( db, tableName ); tst_Databases::safeDropTable( db, tableName );
@ -3182,11 +3192,7 @@ void tst_QSqlQuery::sqlite_finish()
q.finish(); q.finish();
QVERIFY_SQL( q2, exec( "DELETE FROM " + tableName + " WHERE pk_id=2" ) ); QVERIFY_SQL( q2, exec( "DELETE FROM " + tableName + " WHERE pk_id=2" ) );
QCOMPARE( q2.numRowsAffected(), 1 ); QCOMPARE( q2.numRowsAffected(), 1 );
tst_Databases::safeDropTable( db, tableName );
} }
QSqlDatabase::removeDatabase( "sqlite_finish_sqlite" );
} }
void tst_QSqlQuery::nextResult() void tst_QSqlQuery::nextResult()
@ -3301,10 +3307,12 @@ void tst_QSqlQuery::nextResult()
// Stored procedure with multiple result sets // Stored procedure with multiple result sets
const QString procName(qTableName("proc_more_res", __FILE__, db)); const QString procName(qTableName("proc_more_res", __FILE__, db));
if (dbType == QSqlDriver::PostgreSQL) auto dropProc = [&]() {
q.exec(QString("DROP FUNCTION %1(refcursor, refcursor);").arg(procName)); q.exec(QLatin1String(dbType == QSqlDriver::PostgreSQL
else ? "DROP FUNCTION %1(refcursor, refcursor);"
q.exec(QString("DROP PROCEDURE %1;").arg(procName)); : "DROP PROCEDURE %1;").arg(procName));
};
dropProc(); // to make sure it's not there before we start
if (dbType == QSqlDriver::MySqlServer) if (dbType == QSqlDriver::MySqlServer)
QVERIFY_SQL( q, exec( QString( "CREATE PROCEDURE %1()" QVERIFY_SQL( q, exec( QString( "CREATE PROCEDURE %1()"
@ -3338,6 +3346,8 @@ void tst_QSqlQuery::nextResult()
"\nSELECT id, text FROM %2" "\nSELECT id, text FROM %2"
"\nSELECT empty, num, text, id FROM %3" ).arg( procName ).arg( tableName ).arg( tableName ) ) ); "\nSELECT empty, num, text, id FROM %3" ).arg( procName ).arg( tableName ).arg( tableName ) ) );
const auto tidier = qScopeGuard(dropProc);
if (dbType == QSqlDriver::MySqlServer || dbType == QSqlDriver::DB2) { if (dbType == QSqlDriver::MySqlServer || dbType == QSqlDriver::DB2) {
q.setForwardOnly( true ); q.setForwardOnly( true );
QVERIFY_SQL( q, exec( QString( "CALL %1()" ).arg( procName ) ) ); QVERIFY_SQL( q, exec( QString( "CALL %1()" ).arg( procName ) ) );
@ -3399,13 +3409,7 @@ void tst_QSqlQuery::nextResult()
} }
QVERIFY( !q.nextResult() ); QVERIFY( !q.nextResult() );
QVERIFY( !q.isActive() ); QVERIFY( !q.isActive() );
if (dbType == QSqlDriver::PostgreSQL)
q.exec(QString("DROP FUNCTION %1(refcursor, refcursor);").arg(procName));
else
q.exec(QString("DROP PROCEDURE %1;").arg(procName));
} }
@ -3583,13 +3587,14 @@ void tst_QSqlQuery::task_250026()
void tst_QSqlQuery::crashQueryOnCloseDatabase() void tst_QSqlQuery::crashQueryOnCloseDatabase()
{ {
for (const auto &dbName : qAsConst(dbs.dbNames)) { for (const auto &dbName : qAsConst(dbs.dbNames)) {
const auto tidier = qScopeGuard([]() { QSqlDatabase::removeDatabase("crashTest"); });
// Note: destruction of clonedDb needs to happen before we call removeDatabase.
QSqlDatabase clonedDb = QSqlDatabase::cloneDatabase( QSqlDatabase clonedDb = QSqlDatabase::cloneDatabase(
QSqlDatabase::database(dbName), "crashTest"); QSqlDatabase::database(dbName), "crashTest");
qDebug() << "Testing crash in sqlquery dtor for driver" << clonedDb.driverName(); qDebug() << "Testing crash in sqlquery dtor for driver" << clonedDb.driverName();
QVERIFY(clonedDb.open()); QVERIFY(clonedDb.open());
QSqlQuery q(clonedDb); QSqlQuery q(clonedDb);
clonedDb.close(); clonedDb.close();
QSqlDatabase::removeDatabase("crashTest");
} }
} }