michael@0: /* Any copyright is dedicated to the Public Domain.
michael@0:  * http://creativecommons.org/publicdomain/zero/1.0/ */
michael@0: 
michael@0: Cu.import("resource://gre/modules/XPCOMUtils.jsm");
michael@0: Cu.import("resource://services-common/async.js");
michael@0: Cu.import("resource://services-common/utils.js");
michael@0: 
michael@0: _("Make sure querySpinningly will synchronously fetch rows for a query asyncly");
michael@0: 
michael@0: const SQLITE_CONSTRAINT_VIOLATION = 19;  // http://www.sqlite.org/c3ref/c_abort.html
michael@0: 
michael@0: let Svc = {};
michael@0: XPCOMUtils.defineLazyServiceGetter(Svc, "Form",
michael@0:                                    "@mozilla.org/satchel/form-history;1",
michael@0:                                    "nsIFormHistory2");
michael@0: 
michael@0: function querySpinningly(query, names) {
michael@0:   let q = Svc.Form.DBConnection.createStatement(query);
michael@0:   let r = Async.querySpinningly(q, names);
michael@0:   q.finalize();
michael@0:   return r;
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   initTestLogging("Trace");
michael@0: 
michael@0:   _("Make sure the call is async and allows other events to process");
michael@0:   let isAsync = false;
michael@0:   CommonUtils.nextTick(function() { isAsync = true; });
michael@0:   do_check_false(isAsync);
michael@0: 
michael@0:   _("Empty out the formhistory table");
michael@0:   let r0 = querySpinningly("DELETE FROM moz_formhistory");
michael@0:   do_check_eq(r0, null);
michael@0: 
michael@0:   _("Make sure there's nothing there");
michael@0:   let r1 = querySpinningly("SELECT 1 FROM moz_formhistory");
michael@0:   do_check_eq(r1, null);
michael@0: 
michael@0:   _("Insert a row");
michael@0:   let r2 = querySpinningly("INSERT INTO moz_formhistory (fieldname, value) VALUES ('foo', 'bar')");
michael@0:   do_check_eq(r2, null);
michael@0: 
michael@0:   _("Request a known value for the one row");
michael@0:   let r3 = querySpinningly("SELECT 42 num FROM moz_formhistory", ["num"]);
michael@0:   do_check_eq(r3.length, 1);
michael@0:   do_check_eq(r3[0].num, 42);
michael@0: 
michael@0:   _("Get multiple columns");
michael@0:   let r4 = querySpinningly("SELECT fieldname, value FROM moz_formhistory", ["fieldname", "value"]);
michael@0:   do_check_eq(r4.length, 1);
michael@0:   do_check_eq(r4[0].fieldname, "foo");
michael@0:   do_check_eq(r4[0].value, "bar");
michael@0: 
michael@0:   _("Get multiple columns with a different order");
michael@0:   let r5 = querySpinningly("SELECT fieldname, value FROM moz_formhistory", ["value", "fieldname"]);
michael@0:   do_check_eq(r5.length, 1);
michael@0:   do_check_eq(r5[0].fieldname, "foo");
michael@0:   do_check_eq(r5[0].value, "bar");
michael@0: 
michael@0:   _("Add multiple entries (sqlite doesn't support multiple VALUES)");
michael@0:   let r6 = querySpinningly("INSERT INTO moz_formhistory (fieldname, value) SELECT 'foo', 'baz' UNION SELECT 'more', 'values'");
michael@0:   do_check_eq(r6, null);
michael@0: 
michael@0:   _("Get multiple rows");
michael@0:   let r7 = querySpinningly("SELECT fieldname, value FROM moz_formhistory WHERE fieldname = 'foo'", ["fieldname", "value"]);
michael@0:   do_check_eq(r7.length, 2);
michael@0:   do_check_eq(r7[0].fieldname, "foo");
michael@0:   do_check_eq(r7[1].fieldname, "foo");
michael@0: 
michael@0:   _("Make sure updates work");
michael@0:   let r8 = querySpinningly("UPDATE moz_formhistory SET value = 'updated' WHERE fieldname = 'more'");
michael@0:   do_check_eq(r8, null);
michael@0: 
michael@0:   _("Get the updated");
michael@0:   let r9 = querySpinningly("SELECT value, fieldname FROM moz_formhistory WHERE fieldname = 'more'", ["fieldname", "value"]);
michael@0:   do_check_eq(r9.length, 1);
michael@0:   do_check_eq(r9[0].fieldname, "more");
michael@0:   do_check_eq(r9[0].value, "updated");
michael@0: 
michael@0:   _("Grabbing fewer fields than queried is fine");
michael@0:   let r10 = querySpinningly("SELECT value, fieldname FROM moz_formhistory", ["fieldname"]);
michael@0:   do_check_eq(r10.length, 3);
michael@0: 
michael@0:   _("Generate an execution error");
michael@0:   let query = "INSERT INTO moz_formhistory (fieldname, value) VALUES ('one', NULL)";
michael@0:   let stmt = Svc.Form.DBConnection.createStatement(query);
michael@0:   let r11, except; ;
michael@0:   try {
michael@0:     r11 = Async.querySpinningly(stmt);
michael@0:   } catch(e) {
michael@0:     except = e;
michael@0:   }
michael@0:   stmt.finalize()
michael@0:   do_check_true(!!except);
michael@0:   do_check_eq(except.result, SQLITE_CONSTRAINT_VIOLATION);
michael@0: 
michael@0:   _("Cleaning up");
michael@0:   querySpinningly("DELETE FROM moz_formhistory");
michael@0: 
michael@0:   _("Make sure the timeout got to run before this function ends");
michael@0:   do_check_true(isAsync);
michael@0: }