510 lines
13 KiB
C++
510 lines
13 KiB
C++
/* Any copyright is dedicated to the Public Domain.
|
|
http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
#include "storage_test_harness.h"
|
|
|
|
#include "nsIEventTarget.h"
|
|
#include "mozStorageConnection.h"
|
|
|
|
#include "sqlite3.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::storage;
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
//// Helpers
|
|
|
|
/**
|
|
* Commit hook to detect transactions.
|
|
*
|
|
* @param aArg
|
|
* An integer pointer that will be incremented for each commit.
|
|
*/
|
|
int commit_hook(void *aArg)
|
|
{
|
|
int *arg = static_cast<int *>(aArg);
|
|
(*arg)++;
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Executes the passed-in statements and checks if a transaction is created.
|
|
* When done statements are finalized and database connection is closed.
|
|
*
|
|
* @param aDB
|
|
* The database connection.
|
|
* @param aStmts
|
|
* Vector of statements.
|
|
* @param aStmtsLen
|
|
* Number of statements.
|
|
* @param aTransactionExpected
|
|
* Whether a transaction is expected or not.
|
|
*/
|
|
void
|
|
check_transaction(mozIStorageConnection *aDB,
|
|
mozIStorageBaseStatement **aStmts,
|
|
uint32_t aStmtsLen,
|
|
bool aTransactionExpected)
|
|
{
|
|
// -- install a transaction commit hook.
|
|
int commit = 0;
|
|
static_cast<Connection *>(aDB)->setCommitHook(commit_hook, &commit);
|
|
|
|
RefPtr<AsyncStatementSpinner> asyncSpin(new AsyncStatementSpinner());
|
|
nsCOMPtr<mozIStoragePendingStatement> asyncPend;
|
|
do_check_success(aDB->ExecuteAsync(aStmts, aStmtsLen, asyncSpin,
|
|
getter_AddRefs(asyncPend)));
|
|
do_check_true(asyncPend);
|
|
|
|
// -- complete the execution
|
|
asyncSpin->SpinUntilCompleted();
|
|
|
|
// -- uninstall the transaction commit hook.
|
|
static_cast<Connection *>(aDB)->setCommitHook(nullptr);
|
|
|
|
// -- check transaction
|
|
do_check_eq(aTransactionExpected, !!commit);
|
|
|
|
// -- check that only one transaction was created.
|
|
if (aTransactionExpected) {
|
|
do_check_eq(1, commit);
|
|
}
|
|
|
|
// -- cleanup
|
|
for (uint32_t i = 0; i < aStmtsLen; ++i) {
|
|
aStmts[i]->Finalize();
|
|
}
|
|
blocking_async_close(aDB);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
//// Tests
|
|
|
|
/**
|
|
* Test that executing multiple readonly AsyncStatements doesn't create a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_MultipleAsyncReadStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt1;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt2;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing multiple readonly Statements doesn't create a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_MultipleReadStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt1;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageStatement> stmt2;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing multiple AsyncStatements causing writes creates a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_MultipleAsyncReadWriteStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt1;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt2;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing multiple Statements causing writes creates a transaction.
|
|
*/
|
|
void
|
|
test_MultipleReadWriteStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt1;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageStatement> stmt2;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing multiple AsyncStatements causing writes creates a
|
|
* single transaction.
|
|
*/
|
|
void
|
|
test_MultipleAsyncWriteStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt1;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test1 (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt2;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test2 (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing multiple Statements causing writes creates a
|
|
* single transaction.
|
|
*/
|
|
void
|
|
test_MultipleWriteStatements()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt1;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test1 (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt1));
|
|
|
|
nsCOMPtr<mozIStorageStatement> stmt2;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test2 (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt2));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt1,
|
|
stmt2,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single read-only AsyncStatement doesn't create a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_SingleAsyncReadStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single read-only Statement doesn't create a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_SingleReadStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"SELECT * FROM sqlite_master"
|
|
), getter_AddRefs(stmt));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single AsyncStatement causing writes creates a
|
|
* transaction.
|
|
*/
|
|
void
|
|
test_SingleAsyncWriteStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single Statement causing writes creates a transaction.
|
|
*/
|
|
void
|
|
test_SingleWriteStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(stmt));
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single read-only AsyncStatement with multiple params
|
|
* doesn't create a transaction.
|
|
*/
|
|
void
|
|
test_MultipleParamsAsyncReadStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"SELECT :param FROM sqlite_master"
|
|
), getter_AddRefs(stmt));
|
|
|
|
// -- bind multiple BindingParams
|
|
nsCOMPtr<mozIStorageBindingParamsArray> paramsArray;
|
|
stmt->NewBindingParamsArray(getter_AddRefs(paramsArray));
|
|
for (int32_t i = 0; i < 2; i++) {
|
|
nsCOMPtr<mozIStorageBindingParams> params;
|
|
paramsArray->NewBindingParams(getter_AddRefs(params));
|
|
params->BindInt32ByName(NS_LITERAL_CSTRING("param"), 1);
|
|
paramsArray->AddParams(params);
|
|
}
|
|
stmt->BindParameters(paramsArray);
|
|
paramsArray = nullptr;
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single read-only Statement with multiple params
|
|
* doesn't create a transaction.
|
|
*/
|
|
void
|
|
test_MultipleParamsReadStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"SELECT :param FROM sqlite_master"
|
|
), getter_AddRefs(stmt));
|
|
|
|
// -- bind multiple BindingParams
|
|
nsCOMPtr<mozIStorageBindingParamsArray> paramsArray;
|
|
stmt->NewBindingParamsArray(getter_AddRefs(paramsArray));
|
|
for (int32_t i = 0; i < 2; i++) {
|
|
nsCOMPtr<mozIStorageBindingParams> params;
|
|
paramsArray->NewBindingParams(getter_AddRefs(params));
|
|
params->BindInt32ByName(NS_LITERAL_CSTRING("param"), 1);
|
|
paramsArray->AddParams(params);
|
|
}
|
|
stmt->BindParameters(paramsArray);
|
|
paramsArray = nullptr;
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), false);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single write AsyncStatement with multiple params
|
|
* creates a transaction.
|
|
*/
|
|
void
|
|
test_MultipleParamsAsyncWriteStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create a table for writes
|
|
nsCOMPtr<mozIStorageStatement> tableStmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(tableStmt));
|
|
tableStmt->Execute();
|
|
tableStmt->Finalize();
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageAsyncStatement> stmt;
|
|
db->CreateAsyncStatement(NS_LITERAL_CSTRING(
|
|
"DELETE FROM test WHERE id = :param"
|
|
), getter_AddRefs(stmt));
|
|
|
|
// -- bind multiple BindingParams
|
|
nsCOMPtr<mozIStorageBindingParamsArray> paramsArray;
|
|
stmt->NewBindingParamsArray(getter_AddRefs(paramsArray));
|
|
for (int32_t i = 0; i < 2; i++) {
|
|
nsCOMPtr<mozIStorageBindingParams> params;
|
|
paramsArray->NewBindingParams(getter_AddRefs(params));
|
|
params->BindInt32ByName(NS_LITERAL_CSTRING("param"), 1);
|
|
paramsArray->AddParams(params);
|
|
}
|
|
stmt->BindParameters(paramsArray);
|
|
paramsArray = nullptr;
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
/**
|
|
* Test that executing a single write Statement with multiple params
|
|
* creates a transaction.
|
|
*/
|
|
void
|
|
test_MultipleParamsWriteStatement()
|
|
{
|
|
nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
|
|
|
|
// -- create a table for writes
|
|
nsCOMPtr<mozIStorageStatement> tableStmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"CREATE TABLE test (id INTEGER PRIMARY KEY)"
|
|
), getter_AddRefs(tableStmt));
|
|
tableStmt->Execute();
|
|
tableStmt->Finalize();
|
|
|
|
// -- create statements and execute them
|
|
nsCOMPtr<mozIStorageStatement> stmt;
|
|
db->CreateStatement(NS_LITERAL_CSTRING(
|
|
"DELETE FROM test WHERE id = :param"
|
|
), getter_AddRefs(stmt));
|
|
|
|
// -- bind multiple BindingParams
|
|
nsCOMPtr<mozIStorageBindingParamsArray> paramsArray;
|
|
stmt->NewBindingParamsArray(getter_AddRefs(paramsArray));
|
|
for (int32_t i = 0; i < 2; i++) {
|
|
nsCOMPtr<mozIStorageBindingParams> params;
|
|
paramsArray->NewBindingParams(getter_AddRefs(params));
|
|
params->BindInt32ByName(NS_LITERAL_CSTRING("param"), 1);
|
|
paramsArray->AddParams(params);
|
|
}
|
|
stmt->BindParameters(paramsArray);
|
|
paramsArray = nullptr;
|
|
|
|
mozIStorageBaseStatement *stmts[] = {
|
|
stmt,
|
|
};
|
|
|
|
check_transaction(db, stmts, ArrayLength(stmts), true);
|
|
}
|
|
|
|
void (*gTests[])(void) = {
|
|
test_MultipleAsyncReadStatements,
|
|
test_MultipleReadStatements,
|
|
test_MultipleAsyncReadWriteStatements,
|
|
test_MultipleReadWriteStatements,
|
|
test_MultipleAsyncWriteStatements,
|
|
test_MultipleWriteStatements,
|
|
test_SingleAsyncReadStatement,
|
|
test_SingleReadStatement,
|
|
test_SingleAsyncWriteStatement,
|
|
test_SingleWriteStatement,
|
|
test_MultipleParamsAsyncReadStatement,
|
|
test_MultipleParamsReadStatement,
|
|
test_MultipleParamsAsyncWriteStatement,
|
|
test_MultipleParamsWriteStatement,
|
|
};
|
|
|
|
const char *file = __FILE__;
|
|
#define TEST_NAME "async statement execution transaction"
|
|
#define TEST_FILE file
|
|
#include "storage_test_harness_tail.h"
|