105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
|
/**
|
||
|
|
* @file Transaction_test.cpp
|
||
|
|
* @ingroup tests
|
||
|
|
* @brief Test of a SQLite Transaction.
|
||
|
|
*
|
||
|
|
* Copyright (c) 2012-2019 Sebastien Rombauts (sebastien.rombauts@gmail.com)
|
||
|
|
*
|
||
|
|
* Distributed under the MIT License (MIT) (See accompanying file LICENSE.txt
|
||
|
|
* or copy at http://opensource.org/licenses/MIT)
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <SQLiteCpp/Transaction.h>
|
||
|
|
#include <SQLiteCpp/Database.h>
|
||
|
|
#include <SQLiteCpp/Statement.h>
|
||
|
|
#include <SQLiteCpp/Exception.h>
|
||
|
|
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
#include <cstdio>
|
||
|
|
|
||
|
|
TEST(Transaction, commitRollback)
|
||
|
|
{
|
||
|
|
// Create a new database
|
||
|
|
SQLite::Database db(":memory:", SQLite::OPEN_READWRITE|SQLite::OPEN_CREATE);
|
||
|
|
EXPECT_EQ(SQLite::OK, db.getErrorCode());
|
||
|
|
|
||
|
|
{
|
||
|
|
// Begin transaction
|
||
|
|
SQLite::Transaction transaction(db);
|
||
|
|
|
||
|
|
EXPECT_EQ(0, db.exec("CREATE TABLE test (id INTEGER PRIMARY KEY, value TEXT)"));
|
||
|
|
EXPECT_EQ(SQLite::OK, db.getErrorCode());
|
||
|
|
|
||
|
|
// Insert a first value
|
||
|
|
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"first\")"));
|
||
|
|
EXPECT_EQ(1, db.getLastInsertRowid());
|
||
|
|
|
||
|
|
// Commit transaction
|
||
|
|
transaction.commit();
|
||
|
|
|
||
|
|
// Commit again throw an exception
|
||
|
|
EXPECT_THROW(transaction.commit(), SQLite::Exception);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Auto rollback if no commit() before the end of scope
|
||
|
|
{
|
||
|
|
// Begin transaction
|
||
|
|
SQLite::Transaction transaction(db);
|
||
|
|
|
||
|
|
// Insert a second value (that will be rollbacked)
|
||
|
|
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"third\")"));
|
||
|
|
EXPECT_EQ(2, db.getLastInsertRowid());
|
||
|
|
|
||
|
|
// end of scope: automatic rollback
|
||
|
|
}
|
||
|
|
|
||
|
|
// Auto rollback of a transaction on error/exception
|
||
|
|
try
|
||
|
|
{
|
||
|
|
// Begin transaction
|
||
|
|
SQLite::Transaction transaction(db);
|
||
|
|
|
||
|
|
// Insert a second value (that will be rollbacked)
|
||
|
|
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"second\")"));
|
||
|
|
EXPECT_EQ(2, db.getLastInsertRowid());
|
||
|
|
|
||
|
|
// Execute with an error => exception with auto-rollback
|
||
|
|
db.exec("DesiredSyntaxError to raise an exception to rollback the transaction");
|
||
|
|
|
||
|
|
GTEST_FATAL_FAILURE_("we should never get there");
|
||
|
|
transaction.commit(); // We should never get there
|
||
|
|
}
|
||
|
|
catch (std::exception& e)
|
||
|
|
{
|
||
|
|
std::cout << "SQLite exception: " << e.what() << std::endl;
|
||
|
|
// expected error, see above
|
||
|
|
}
|
||
|
|
|
||
|
|
// Double rollback with a manual command before the end of scope
|
||
|
|
{
|
||
|
|
// Begin transaction
|
||
|
|
SQLite::Transaction transaction(db);
|
||
|
|
|
||
|
|
// Insert a second value (that will be rollbacked)
|
||
|
|
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"third\")"));
|
||
|
|
EXPECT_EQ(2, db.getLastInsertRowid());
|
||
|
|
|
||
|
|
// Execute a manual rollback (no real use case I can think of, so no rollback() method)
|
||
|
|
db.exec("ROLLBACK");
|
||
|
|
|
||
|
|
// end of scope: the automatic rollback should not raise an error because it is harmless
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check the results (expect only one row of result, as all other one have been rollbacked)
|
||
|
|
SQLite::Statement query(db, "SELECT * FROM test");
|
||
|
|
int nbRows = 0;
|
||
|
|
while (query.executeStep())
|
||
|
|
{
|
||
|
|
nbRows++;
|
||
|
|
EXPECT_EQ(1, query.getColumn(0).getInt());
|
||
|
|
EXPECT_STREQ("first", query.getColumn(1).getText());
|
||
|
|
}
|
||
|
|
EXPECT_EQ(1, nbRows);
|
||
|
|
}
|