michael@0: /* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* vim:set ts=2 sw=2 sts=2 et: */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: // Get bookmark service
michael@0: try {
michael@0:   var bmsvc = Cc["@mozilla.org/browser/nav-bookmarks-service;1"].getService(Ci.nsINavBookmarksService);
michael@0: } catch(ex) {
michael@0:   do_throw("Could not get nav-bookmarks-service\n");
michael@0: }
michael@0: 
michael@0: // Get history service
michael@0: try {
michael@0:   var histsvc = Cc["@mozilla.org/browser/nav-history-service;1"].getService(Ci.nsINavHistoryService);
michael@0: } catch(ex) {
michael@0:   do_throw("Could not get history service\n");
michael@0: } 
michael@0: 
michael@0: // get bookmarks root id
michael@0: var root = bmsvc.bookmarksMenuFolder;
michael@0: 
michael@0: // main
michael@0: function run_test() {
michael@0:   // test querying for bookmarks in multiple folders
michael@0:   var testFolder1 = bmsvc.createFolder(root, "bug 384228 test folder 1",
michael@0:                                        bmsvc.DEFAULT_INDEX);
michael@0:   do_check_eq(bmsvc.getItemIndex(testFolder1), 0);
michael@0:   var testFolder2 = bmsvc.createFolder(root, "bug 384228 test folder 2",
michael@0:                                        bmsvc.DEFAULT_INDEX);
michael@0:   do_check_eq(bmsvc.getItemIndex(testFolder2), 1);
michael@0:   var testFolder3 = bmsvc.createFolder(root, "bug 384228 test folder 3",
michael@0:                                        bmsvc.DEFAULT_INDEX);
michael@0:   do_check_eq(bmsvc.getItemIndex(testFolder3), 2);
michael@0: 
michael@0:   var b1 = bmsvc.insertBookmark(testFolder1, uri("http://foo.tld/"),
michael@0:                                 bmsvc.DEFAULT_INDEX, "title b1 (folder 1)");
michael@0:   do_check_eq(bmsvc.getItemIndex(b1), 0);
michael@0:   var b2 = bmsvc.insertBookmark(testFolder1, uri("http://foo.tld/"),
michael@0:                                 bmsvc.DEFAULT_INDEX, "title b2 (folder 1)");
michael@0:   do_check_eq(bmsvc.getItemIndex(b2), 1);
michael@0:   var b3 = bmsvc.insertBookmark(testFolder2, uri("http://foo.tld/"),
michael@0:                                 bmsvc.DEFAULT_INDEX, "title b3 (folder 2)");
michael@0:   do_check_eq(bmsvc.getItemIndex(b3), 0);
michael@0:   var b4 = bmsvc.insertBookmark(testFolder3, uri("http://foo.tld/"),
michael@0:                                 bmsvc.DEFAULT_INDEX, "title b4 (folder 3)");
michael@0:   do_check_eq(bmsvc.getItemIndex(b4), 0);
michael@0:   // also test recursive search
michael@0:   var testFolder1_1 = bmsvc.createFolder(testFolder1, "bug 384228 test folder 1.1",
michael@0:                                          bmsvc.DEFAULT_INDEX);
michael@0:   do_check_eq(bmsvc.getItemIndex(testFolder1_1), 2);
michael@0:   var b5 = bmsvc.insertBookmark(testFolder1_1, uri("http://a1.com/"),
michael@0:                                 bmsvc.DEFAULT_INDEX, "title b5 (folder 1.1)");
michael@0:   do_check_eq(bmsvc.getItemIndex(b5), 0);
michael@0: 
michael@0:   var options = histsvc.getNewQueryOptions();
michael@0:   var query = histsvc.getNewQuery();
michael@0:   query.searchTerms = "title";
michael@0:   options.queryType = Ci.nsINavHistoryQueryOptions.QUERY_TYPE_BOOKMARKS;
michael@0:   query.setFolders([testFolder1, testFolder2], 2);
michael@0: 
michael@0:   var result = histsvc.executeQuery(query, options);
michael@0:   var rootNode = result.root;
michael@0:   rootNode.containerOpen = true;
michael@0: 
michael@0:   // should not match item from folder 3
michael@0:   do_check_eq(rootNode.childCount, 4);
michael@0: 
michael@0:   do_check_eq(rootNode.getChild(0).itemId, b1);
michael@0:   do_check_eq(rootNode.getChild(1).itemId, b2);
michael@0:   do_check_eq(rootNode.getChild(2).itemId, b3);
michael@0:   do_check_eq(rootNode.getChild(3).itemId, b5);
michael@0: 
michael@0:   rootNode.containerOpen = false;
michael@0: }