The Tor Browser: storage/test/unit/test_storage_progresshandler.js@b8a032363ba2 (annotated)

storage/test/unit/test_storage_progresshandler.js@b8a032363ba2 (annotated)

storage/test/unit/test_storage_progresshandler.js

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

 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 // This file tests the custom progress handlers
 function setup()
 {
   var msc = getOpenedDatabase();
   msc.createTable("handler_tests", "id INTEGER PRIMARY KEY, num INTEGER");
   msc.beginTransaction();
   var stmt = createStatement("INSERT INTO handler_tests (id, num) VALUES(?1, ?2)");
   for(var i = 0; i < 100; ++i) {
     stmt.bindByIndex(0, i);
     stmt.bindByIndex(1, Math.floor(Math.random()*1000));
     stmt.execute();
   }
   stmt.reset();
   msc.commitTransaction();
   stmt.finalize();
 }
 var testProgressHandler = {
   calls: 0,
   abort: false,
   onProgress: function(comm) {
     ++this.calls;
     return this.abort;
   }
 };
 function test_handler_registration()
 {
   var msc = getOpenedDatabase();
   msc.setProgressHandler(10, testProgressHandler);
 }
 function test_handler_return()
 {
   var msc = getOpenedDatabase();
   var oldH = msc.setProgressHandler(5, testProgressHandler);
   do_check_true(oldH instanceof Ci.mozIStorageProgressHandler);
 }
 function test_handler_removal()
 {
   var msc = getOpenedDatabase();
   msc.removeProgressHandler();
   var oldH = msc.removeProgressHandler();
   do_check_eq(oldH, null);
 }
 function test_handler_call()
 {
   var msc = getOpenedDatabase();
   msc.setProgressHandler(50, testProgressHandler);
   // Some long-executing request
   var stmt = createStatement(
     "SELECT SUM(t1.num * t2.num) FROM handler_tests AS t1, handler_tests AS t2");
   while(stmt.executeStep());
   do_check_true(testProgressHandler.calls > 0);
   stmt.finalize();
 }
 function test_handler_abort()
 {
   var msc = getOpenedDatabase();
   testProgressHandler.abort = true;
   msc.setProgressHandler(50, testProgressHandler);
   // Some long-executing request
   var stmt = createStatement(
     "SELECT SUM(t1.num * t2.num) FROM handler_tests AS t1, handler_tests AS t2");
   const SQLITE_INTERRUPT = 9;
   try {
     while(stmt.executeStep());
     do_throw("We shouldn't get here!");
   } catch (e) {
     do_check_eq(Cr.NS_ERROR_ABORT, e.result);
     do_check_eq(SQLITE_INTERRUPT, msc.lastError);
   }
   try {
     stmt.finalize();
     do_throw("We shouldn't get here!");
   } catch (e) {
     // finalize should return the error code since we encountered an error
     do_check_eq(Cr.NS_ERROR_ABORT, e.result);
     do_check_eq(SQLITE_INTERRUPT, msc.lastError);
   }
 }
 var tests = [test_handler_registration, test_handler_return,
              test_handler_removal, test_handler_call,
              test_handler_abort];
 function run_test()
 {
   setup();
   for (var i = 0; i < tests.length; i++) {
     tests[i]();
   }
   cleanup();
 }