The Tor Browser: storage/test/test_async_callbacks_with_spun_event

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

storage/test/test_async_callbacks_with_spun_event_loops.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 "prthread.h"
 #include "nsIEventTarget.h"
 #include "nsIInterfaceRequestorUtils.h"
 #include "mozilla/Attributes.h"
 #include "sqlite3.h"
 ////////////////////////////////////////////////////////////////////////////////
 //// Async Helpers
 /**
  * Spins the events loop for current thread until aCondition is true.
  */
 void
 spin_events_loop_until_true(const bool* const aCondition)
 {
   nsCOMPtr<nsIThread> thread(::do_GetCurrentThread());
   nsresult rv = NS_OK;
   bool processed = true;
   while (!(*aCondition) && NS_SUCCEEDED(rv)) {
     rv = thread->ProcessNextEvent(true, &processed);
   }
 }
 ////////////////////////////////////////////////////////////////////////////////
 //// mozIStorageStatementCallback implementation
 class UnownedCallback MOZ_FINAL : public mozIStorageStatementCallback
 {
 public:
   NS_DECL_ISUPPORTS
   // Whether the object has been destroyed.
   static bool sAlive;
   // Whether the first result was received.
   static bool sResult;
   // Whether an error was received.
   static bool sError;
   UnownedCallback(mozIStorageConnection* aDBConn)
   : mDBConn(aDBConn)
   , mCompleted(false)
   {
     sAlive = true;
     sResult = false;
     sError = false;
   }
   ~UnownedCallback()
   {
     sAlive = false;
     blocking_async_close(mDBConn);
   }
   NS_IMETHOD HandleResult(mozIStorageResultSet* aResultSet)
   {
     sResult = true;
     spin_events_loop_until_true(&mCompleted);
     if (!sAlive) {
       NS_RUNTIMEABORT("The statement callback was destroyed prematurely.");
     }
     return NS_OK;
   }
   NS_IMETHOD HandleError(mozIStorageError* aError)
   {
     sError = true;
     spin_events_loop_until_true(&mCompleted);
     if (!sAlive) {
       NS_RUNTIMEABORT("The statement callback was destroyed prematurely.");
     }
     return NS_OK;
   }
   NS_IMETHOD HandleCompletion(uint16_t aReason)
   {
     mCompleted = true;
     return NS_OK;
   }
 protected:
   nsCOMPtr<mozIStorageConnection> mDBConn;
   bool mCompleted;
 };
 NS_IMPL_ISUPPORTS(UnownedCallback, mozIStorageStatementCallback)
 bool UnownedCallback::sAlive = false;
 bool UnownedCallback::sResult = false;
 bool UnownedCallback::sError = false;
 ////////////////////////////////////////////////////////////////////////////////
 //// Tests
 void
 test_SpinEventsLoopInHandleResult()
 {
   nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
   // Create a test table and populate it.
   nsCOMPtr<mozIStorageStatement> stmt;
   db->CreateStatement(NS_LITERAL_CSTRING(
     "CREATE TABLE test (id INTEGER PRIMARY KEY)"
   ), getter_AddRefs(stmt));
   stmt->Execute();
   stmt->Finalize();
   db->CreateStatement(NS_LITERAL_CSTRING(
     "INSERT INTO test (id) VALUES (?)"
   ), getter_AddRefs(stmt));
   for (int32_t i = 0; i < 30; ++i) {
     stmt->BindInt32ByIndex(0, i);
     stmt->Execute();
     stmt->Reset();
   }
   stmt->Finalize();
   db->CreateStatement(NS_LITERAL_CSTRING(
     "SELECT * FROM test"
   ), getter_AddRefs(stmt));
   nsCOMPtr<mozIStoragePendingStatement> ps;
   do_check_success(stmt->ExecuteAsync(new UnownedCallback(db),
                                       getter_AddRefs(ps)));
   stmt->Finalize();
   spin_events_loop_until_true(&UnownedCallback::sResult);
 }
 void
 test_SpinEventsLoopInHandleError()
 {
   nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());
   // Create a test table and populate it.
   nsCOMPtr<mozIStorageStatement> stmt;
   db->CreateStatement(NS_LITERAL_CSTRING(
     "CREATE TABLE test (id INTEGER PRIMARY KEY)"
   ), getter_AddRefs(stmt));
   stmt->Execute();
   stmt->Finalize();
   db->CreateStatement(NS_LITERAL_CSTRING(
     "INSERT INTO test (id) VALUES (1)"
   ), getter_AddRefs(stmt));
   stmt->Execute();
   stmt->Finalize();
   // This will cause a constraint error.
   db->CreateStatement(NS_LITERAL_CSTRING(
     "INSERT INTO test (id) VALUES (1)"
   ), getter_AddRefs(stmt));
   nsCOMPtr<mozIStoragePendingStatement> ps;
   do_check_success(stmt->ExecuteAsync(new UnownedCallback(db),
                                       getter_AddRefs(ps)));
   stmt->Finalize();
   spin_events_loop_until_true(&UnownedCallback::sError);
 }
 void (*gTests[])(void) = {
   test_SpinEventsLoopInHandleResult,
   test_SpinEventsLoopInHandleError,
 };
 const char *file = __FILE__;
 #define TEST_NAME "test async callbacks with spun event loops"
 #define TEST_FILE file
 #include "storage_test_harness_tail.h"

The Tor Browser / annotate

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

storage/test/test_async_callbacks_with_spun_event_loops.cpp