The Tor Browser: storage/test/test_asyncStatementExecution_transaction.cpp@b8a032363ba2 (annotated)

storage/test/test_asyncStatementExecution_transaction.cpp@b8a032363ba2 (annotated)

storage/test/test_asyncStatementExecution_transaction.cpp

Thu, 22 Jan 2015 13:21:57 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 22 Jan 2015 13:21:57 +0100
branch: TOR_BUG_9701
changeset 15: b8a032363ba2
permissions: -rw-r--r--

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

 /* 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);
   nsRefPtr<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"

The Tor Browser / annotate

storage/test/test_asyncStatementExecution_transaction.cpp@b8a032363ba2 (annotated)

storage/test/test_asyncStatementExecution_transaction.cpp