gecko/toolkit/components/satchel/test/unit/perf_autocomplete.js

/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is Satchel Test Code.
 *
 * The Initial Developer of the Original Code is
 * Mozilla Corporation.
 * Portions created by the Initial Developer are Copyright (C) 2009
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *   Matthew Noorenberghe <mnoorenberghe@mozilla.com> (Original Author)
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */

var testnum = 0;
var fh;
var fac;
var prefs;

function countAllEntries() {
    let stmt = fh.DBConnection.createStatement("SELECT COUNT(*) as numEntries FROM moz_formhistory");
    do_check_true(stmt.executeStep());
    let numEntries = stmt.row.numEntries;
    stmt.finalize();
    return numEntries;
}

function do_AC_search(searchTerm, previousResult) {
    var duration = 0;
    var searchCount = 5;
    var tempPrevious = null;
    var startTime;
    for (var i = 0; i < searchCount; i++) {
        if (previousResult !== null)
            tempPrevious = fac.autoCompleteSearch("searchbar-history", previousResult, null, null);
        startTime = Date.now();
        results = fac.autoCompleteSearch("searchbar-history", searchTerm, null, tempPrevious);
        duration += Date.now() - startTime;
    }
    dump("[autoCompleteSearch][test " + testnum + "] for '" + searchTerm + "' ");
    if (previousResult !== null)
        dump("with '" + previousResult + "' previous result ");
    else
        dump("w/o previous result ");
    dump("took " + duration + " ms with " + results.matchCount + " matches.  ");
    dump("Average of " + Math.round(duration / searchCount) + " ms per search\n");
    return results;
}

function run_test() {
    try {

        // ===== test init =====
        var testfile = do_get_file("formhistory_1000.sqlite");
        var profileDir = dirSvc.get("ProfD", Ci.nsIFile);
        var results;

        // Cleanup from any previous tests or failures.
        var destFile = profileDir.clone();
        destFile.append("formhistory.sqlite");
        if (destFile.exists())
          destFile.remove(false);

        testfile.copyTo(profileDir, "formhistory.sqlite");

        fh = Cc["@mozilla.org/satchel/form-history;1"].
             getService(Ci.nsIFormHistory2);
        fac = Cc["@mozilla.org/satchel/form-autocomplete;1"].
              getService(Ci.nsIFormAutoComplete);
        prefs = Cc["@mozilla.org/preferences-service;1"].
                getService(Ci.nsIPrefBranch);

        timeGroupingSize = prefs.getIntPref("browser.formfill.timeGroupingSize") * 1000 * 1000;
        maxTimeGroupings = prefs.getIntPref("browser.formfill.maxTimeGroupings");
        bucketSize = prefs.getIntPref("browser.formfill.bucketSize");

        // ===== 1 =====
        // Check initial state is as expected
        testnum++;
        do_check_true(fh.hasEntries);
        do_check_eq(1000, countAllEntries());
        fac.autoCompleteSearch("searchbar-history", "zzzzzzzzzz", null, null); // warm-up search
        do_check_true(fh.nameExists("searchbar-history"));

        // ===== 2 =====
        // Search for '' with no previous result
        testnum++;
        results = do_AC_search("", null);
        do_check_true(results.matchCount > 0);

        // ===== 3 =====
        // Search for 'r' with no previous result
        testnum++;
        results = do_AC_search("r", null);
        do_check_true(results.matchCount > 0);

        // ===== 4 =====
        // Search for 'r' with '' previous result
        testnum++;
        results = do_AC_search("r", "");
        do_check_true(results.matchCount > 0);

        // ===== 5 =====
        // Search for 're' with no previous result
        testnum++;
        results = do_AC_search("re", null);
        do_check_true(results.matchCount > 0);

        // ===== 6 =====
        // Search for 're' with 'r' previous result
        testnum++;
        results = do_AC_search("re", "r");
        do_check_true(results.matchCount > 0);

        // ===== 7 =====
        // Search for 'rea' without previous result
        testnum++;
        results = do_AC_search("rea", null);
        let countREA = results.matchCount;

        // ===== 8 =====
        // Search for 'rea' with 're' previous result
        testnum++;
        results = do_AC_search("rea", "re");
        do_check_eq(countREA, results.matchCount);

        // ===== 9 =====
        // Search for 'real' with 'rea' previous result
        testnum++;
        results = do_AC_search("real", "rea");
        let countREAL = results.matchCount;
        do_check_true(results.matchCount <= countREA);

        // ===== 10 =====
        // Search for 'real' with 're' previous result
        testnum++;
        results = do_AC_search("real", "re");
        do_check_eq(countREAL, results.matchCount);

        // ===== 11 =====
        // Search for 'real' with no previous result
        testnum++;
        results = do_AC_search("real", null);
        do_check_eq(countREAL, results.matchCount);


    } catch (e) {
      throw "FAILED in test #" + testnum + " -- " + e;
    }
}