storage/test/unit/test_storage_function.js

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is Storage Test Code.
  15  *
  16  * The Initial Developer of the Original Code is
  17  *   Lev Serebryakov <lev@serebryakov.spb.ru>
  18  * Portions created by the Initial Developer are Copyright (C) 2007
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 2 or later (the "GPL"), or
  25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  26  * in which case the provisions of the GPL or the LGPL are applicable instead
  27  * of those above. If you wish to allow use of your version of this file only
  28  * under the terms of either the GPL or the LGPL, and not to allow others to
  29  * use your version of this file under the terms of the MPL, indicate your
  30  * decision by deleting the provisions above and replace them with the notice
  31  * and other provisions required by the GPL or the LGPL. If you do not delete
  32  * the provisions above, a recipient may use your version of this file under
  33  * the terms of any one of the MPL, the GPL or the LGPL.
  34  *
  35  * ***** END LICENSE BLOCK ***** */
  36
  37 // This file tests the custom functions
  38
  39 var testNums = [1, 2, 3, 4];
  40
  41 function setup()
  42 {
  43   getOpenedDatabase().createTable("function_tests", "id INTEGER PRIMARY KEY");
  44
  45   var stmt = createStatement("INSERT INTO function_tests (id) VALUES(?1)");
  46   for(var i = 0; i < testNums.length; ++i) {
  47     stmt.bindInt32Parameter(0, testNums[i]);
  48     stmt.execute();
  49   }
  50   stmt.reset();
  51   stmt.finalize();
  52 }
  53
  54 var testSquareFunction = {
  55   calls: 0,
  56
  57   onFunctionCall: function(val) {
  58     ++this.calls;
  59     return val.getInt32(0) * val.getInt32(0);
  60   }
  61 };
  62
  63 function test_function_registration()
  64 {
  65   var msc = getOpenedDatabase();
  66   msc.createFunction("test_square", 1, testSquareFunction);
  67 }
  68
  69 function test_function_no_double_registration()
  70 {
  71   var msc = getOpenedDatabase();
  72   try {
  73     msc.createFunction("test_square", 2, testSquareFunction);
  74     do_throw("We shouldn't get here!");
  75   } catch (e) {
  76     do_check_eq(Cr.NS_ERROR_FAILURE, e.result);
  77   }
  78 }
  79
  80 function test_function_removal()
  81 {
  82   var msc = getOpenedDatabase();
  83   msc.removeFunction("test_square");
  84   // Should be Ok now
  85   msc.createFunction("test_square", 1, testSquareFunction);
  86 }
  87
  88 function test_function_aliases()
  89 {
  90   var msc = getOpenedDatabase();
  91   msc.createFunction("test_square2", 1, testSquareFunction);
  92 }
  93
  94 function test_function_call()
  95 {
  96   var stmt = createStatement("SELECT test_square(id) FROM function_tests");
  97   while(stmt.executeStep());
  98   do_check_eq(testNums.length, testSquareFunction.calls);
  99   testSquareFunction.calls = 0;
 100   stmt.finalize();
 101 }
 102
 103 function test_function_result()
 104 {
 105   var stmt = createStatement("SELECT test_square(42) FROM function_tests");
 106   stmt.executeStep();
 107   do_check_eq(42*42, stmt.getInt32(0));
 108   testSquareFunction.calls = 0;
 109   stmt.finalize();
 110 }
 111
 112 var tests = [test_function_registration, test_function_no_double_registration,
 113              test_function_removal, test_function_aliases, test_function_call,
 114              test_function_result];
 115
 116 function run_test()
 117 {
 118   setup();
 119
 120   for (var i = 0; i < tests.length; i++) {
 121     tests[i]();
 122   }
 123
 124   cleanup();
 125 }