mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
411 lines
17 KiB
JavaScript
411 lines
17 KiB
JavaScript
/* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et: */
|
|
/* 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/. */
|
|
|
|
const Ci = Components.interfaces;
|
|
const Cc = Components.classes;
|
|
const Cr = Components.results;
|
|
const Cu = Components.utils;
|
|
|
|
Cu.import("resource://gre/modules/Services.jsm");
|
|
|
|
// Import common head.
|
|
let (commonFile = do_get_file("../head_common.js", false)) {
|
|
let uri = Services.io.newFileURI(commonFile);
|
|
Services.scriptloader.loadSubScript(uri.spec, this);
|
|
}
|
|
|
|
// Put any other stuff relative to this test folder below.
|
|
|
|
|
|
// Some Useful Date constants - PRTime uses microseconds, so convert
|
|
const DAY_MICROSEC = 86400000000;
|
|
const today = Date.now() * 1000;
|
|
const yesterday = today - DAY_MICROSEC;
|
|
const lastweek = today - (DAY_MICROSEC * 7);
|
|
const daybefore = today - (DAY_MICROSEC * 2);
|
|
const tomorrow = today + DAY_MICROSEC;
|
|
const old = today - (DAY_MICROSEC * 3);
|
|
const futureday = today + (DAY_MICROSEC * 3);
|
|
const olderthansixmonths = today - (DAY_MICROSEC * 31 * 7);
|
|
|
|
|
|
/**
|
|
* Generalized function to pull in an array of objects of data and push it into
|
|
* the database. It does NOT do any checking to see that the input is
|
|
* appropriate.
|
|
*/
|
|
function populateDB(aArray) {
|
|
PlacesUtils.history.runInBatchMode({
|
|
runBatched: function (aUserData)
|
|
{
|
|
aArray.forEach(function (data)
|
|
{
|
|
try {
|
|
// make the data object into a query data object in order to create proper
|
|
// default values for anything left unspecified
|
|
var qdata = new queryData(data);
|
|
if (qdata.isVisit) {
|
|
// Then we should add a visit for this node
|
|
var referrer = qdata.referrer ? uri(qdata.referrer) : null;
|
|
var visitId = PlacesUtils.history.addVisit(uri(qdata.uri), qdata.lastVisit,
|
|
referrer, qdata.transType,
|
|
qdata.isRedirect, qdata.sessionID);
|
|
if (qdata.title && !qdata.isDetails) {
|
|
PlacesUtils.history.setPageTitle(uri(qdata.uri), qdata.title);
|
|
}
|
|
if (qdata.visitCount && !qdata.isDetails) {
|
|
// Set a fake visit_count, this is not a real count but can be used
|
|
// to test sorting by visit_count.
|
|
let stmt = DBConn().createStatement(
|
|
"UPDATE moz_places SET visit_count = :vc WHERE url = :url");
|
|
stmt.params.vc = qdata.visitCount;
|
|
stmt.params.url = qdata.uri;
|
|
try {
|
|
stmt.execute();
|
|
}
|
|
catch (ex) {
|
|
print("Error while setting visit_count.");
|
|
}
|
|
finally {
|
|
stmt.finalize();
|
|
}
|
|
}
|
|
}
|
|
|
|
if (qdata.isRedirect) {
|
|
// Redirect sources added through the docshell are properly marked
|
|
// as redirects and get hidden state, the API doesn't have that
|
|
// power (And actually doesn't make much sense to add redirects
|
|
// through the API).
|
|
// This must be async cause otherwise the updateFrecency call
|
|
// done by addVisits may randomly happen after it, overwriting the
|
|
// value.
|
|
let stmt = DBConn().createAsyncStatement(
|
|
"UPDATE moz_places SET hidden = 1 WHERE url = :url");
|
|
stmt.params.url = qdata.uri;
|
|
try {
|
|
stmt.executeAsync();
|
|
}
|
|
catch (ex) {
|
|
print("Error while setting hidden.");
|
|
}
|
|
finally {
|
|
stmt.finalize();
|
|
}
|
|
}
|
|
|
|
if (qdata.isDetails) {
|
|
// Then we add extraneous page details for testing
|
|
PlacesUtils.history.addVisit(uri(qdata.uri),
|
|
qdata.lastVisit,
|
|
null,
|
|
TRANSITION_LINK,
|
|
false,
|
|
0);
|
|
PlacesUtils.ghistory2.setPageTitle(uri(qdata.uri),
|
|
qdata.title);
|
|
}
|
|
|
|
if (qdata.markPageAsTyped){
|
|
PlacesUtils.bhistory.markPageAsTyped(uri(qdata.uri));
|
|
}
|
|
|
|
if (qdata.isPageAnnotation) {
|
|
if (qdata.removeAnnotation)
|
|
PlacesUtils.annotations.removePageAnnotation(uri(qdata.uri),
|
|
qdata.annoName);
|
|
else {
|
|
PlacesUtils.annotations.setPageAnnotation(uri(qdata.uri),
|
|
qdata.annoName,
|
|
qdata.annoVal,
|
|
qdata.annoFlags,
|
|
qdata.annoExpiration);
|
|
}
|
|
}
|
|
|
|
if (qdata.isItemAnnotation) {
|
|
if (qdata.removeAnnotation)
|
|
PlacesUtils.annotations.removeItemAnnotation(qdata.itemId,
|
|
qdata.annoName);
|
|
else {
|
|
PlacesUtils.annotations.setItemAnnotation(qdata.itemId,
|
|
qdata.annoName,
|
|
qdata.annoVal,
|
|
qdata.annoFlags,
|
|
qdata.annoExpiration);
|
|
}
|
|
}
|
|
|
|
if (qdata.isPageBinaryAnnotation) {
|
|
if (qdata.removeAnnotation)
|
|
PlacesUtils.annotations.removePageAnnotation(uri(qdata.uri),
|
|
qdata.annoName);
|
|
else {
|
|
PlacesUtils.annotations.setPageAnnotationBinary(uri(qdata.uri),
|
|
qdata.annoName,
|
|
qdata.binarydata,
|
|
qdata.binaryDataLength,
|
|
qdata.annoMimeType,
|
|
qdata.annoFlags,
|
|
qdata.annoExpiration);
|
|
}
|
|
}
|
|
|
|
if (qdata.isItemBinaryAnnotation) {
|
|
if (qdata.removeAnnotation)
|
|
PlacesUtils.annotations.removeItemAnnotation(qdata.itemId,
|
|
qdata.annoName);
|
|
else {
|
|
PlacesUtils.annotations.setItemAnnotationBinary(qdata.itemId,
|
|
qdata.annoName,
|
|
qdata.binaryData,
|
|
qdata.binaryDataLength,
|
|
qdata.annoMimeType,
|
|
qdata.annoFlags,
|
|
qdata.annoExpiration);
|
|
}
|
|
}
|
|
|
|
if (qdata.isFolder) {
|
|
let folderId = PlacesUtils.bookmarks.createFolder(qdata.parentFolder,
|
|
qdata.title,
|
|
qdata.index);
|
|
if (qdata.readOnly)
|
|
PlacesUtils.bookmarks.setFolderReadonly(folderId, true);
|
|
}
|
|
|
|
if (qdata.isLivemark) {
|
|
PlacesUtils.livemarks.addLivemark({ title: qdata.title
|
|
, parentId: qdata.parentFolder
|
|
, index: qdata.index
|
|
, feedURI: uri(qdata.feedURI)
|
|
, siteURI: uri(qdata.uri)
|
|
});
|
|
}
|
|
|
|
if (qdata.isBookmark) {
|
|
let itemId = PlacesUtils.bookmarks.insertBookmark(qdata.parentFolder,
|
|
uri(qdata.uri),
|
|
qdata.index,
|
|
qdata.title);
|
|
if (qdata.keyword)
|
|
PlacesUtils.bookmarks.setKeywordForBookmark(itemId, qdata.keyword);
|
|
if (qdata.dateAdded)
|
|
PlacesUtils.bookmarks.setItemDateAdded(itemId, qdata.dateAdded);
|
|
if (qdata.lastModified)
|
|
PlacesUtils.bookmarks.setItemLastModified(itemId, qdata.lastModified);
|
|
}
|
|
|
|
if (qdata.isTag) {
|
|
PlacesUtils.tagging.tagURI(uri(qdata.uri), qdata.tagArray);
|
|
}
|
|
|
|
if (qdata.isDynContainer) {
|
|
PlacesUtils.bookmarks.createDynamicContainer(qdata.parentFolder,
|
|
qdata.title,
|
|
qdata.contractId,
|
|
qdata.index);
|
|
}
|
|
|
|
if (qdata.isSeparator)
|
|
PlacesUtils.bookmarks.insertSeparator(qdata.parentFolder, qdata.index);
|
|
} catch (ex) {
|
|
// use the data object here in case instantiation of qdata failed
|
|
LOG("Problem with this URI: " + data.uri);
|
|
do_throw("Error creating database: " + ex + "\n");
|
|
}
|
|
}); // End of function and array
|
|
}
|
|
}, null);
|
|
}
|
|
|
|
|
|
/**
|
|
* The Query Data Object - this object encapsulates data for our queries and is
|
|
* used to parameterize our calls to the Places APIs to put data into the
|
|
* database. It also has some interesting meta functions to determine which APIs
|
|
* should be called, and to determine if this object should show up in the
|
|
* resulting query.
|
|
* Its parameter is an object specifying which attributes you want to set.
|
|
* For ex:
|
|
* var myobj = new queryData({isVisit: true, uri:"http://mozilla.com", title="foo"});
|
|
* Note that it doesn't do any input checking on that object.
|
|
*/
|
|
function queryData(obj) {
|
|
this.isVisit = obj.isVisit ? obj.isVisit : false;
|
|
this.isBookmark = obj.isBookmark ? obj.isBookmark: false;
|
|
this.uri = obj.uri ? obj.uri : "";
|
|
this.lastVisit = obj.lastVisit ? obj.lastVisit : today;
|
|
this.referrer = obj.referrer ? obj.referrer : null;
|
|
this.transType = obj.transType ? obj.transType : Ci.nsINavHistoryService.TRANSITION_TYPED;
|
|
this.isRedirect = obj.isRedirect ? obj.isRedirect : false;
|
|
this.sessionID = obj.sessionID ? obj.sessionID : 0;
|
|
this.isDetails = obj.isDetails ? obj.isDetails : false;
|
|
this.title = obj.title ? obj.title : "";
|
|
this.markPageAsTyped = obj.markPageAsTyped ? obj.markPageAsTyped : false;
|
|
this.isPageAnnotation = obj.isPageAnnotation ? obj.isPageAnnotation : false;
|
|
this.removeAnnotation= obj.removeAnnotation ? true : false;
|
|
this.annoName = obj.annoName ? obj.annoName : "";
|
|
this.annoVal = obj.annoVal ? obj.annoVal : "";
|
|
this.annoFlags = obj.annoFlags ? obj.annoFlags : 0;
|
|
this.annoExpiration = obj.annoExpiration ? obj.annoExpiration : 0;
|
|
this.isItemAnnotation = obj.isItemAnnotation ? obj.isItemAnnotation : false;
|
|
this.itemId = obj.itemId ? obj.itemId : 0;
|
|
this.isPageBinaryAnnotation = obj.isPageBinaryAnnotation ?
|
|
obj.isPageBinaryAnnotation : false;
|
|
this.isItemBinaryAnnotation = obj.isItemBinaryAnnotation ?
|
|
obj.isItemBinaryAnnotation : false;
|
|
this.binaryData = obj.binaryData ? obj.binaryData : null;
|
|
this.binaryDataLength = obj.binaryDataLength ? obj.binaryDataLength : 0;
|
|
this.annoMimeType = obj.annoMimeType ? obj.annoMimeType : "";
|
|
this.isTag = obj.isTag ? obj.isTag : false;
|
|
this.tagArray = obj.tagArray ? obj.tagArray : null;
|
|
this.isLivemark = obj.isLivemark ? obj.isLivemark : false;
|
|
this.parentFolder = obj.parentFolder ? obj.parentFolder
|
|
: PlacesUtils.placesRootId;
|
|
this.feedURI = obj.feedURI ? obj.feedURI : "";
|
|
this.index = obj.index ? obj.index : PlacesUtils.bookmarks.DEFAULT_INDEX;
|
|
this.isFolder = obj.isFolder ? obj.isFolder : false;
|
|
this.contractId = obj.contractId ? obj.contractId : "";
|
|
this.lastModified = obj.lastModified ? obj.lastModified : today;
|
|
this.dateAdded = obj.dateAdded ? obj.dateAdded : today;
|
|
this.keyword = obj.keyword ? obj.keyword : "";
|
|
this.visitCount = obj.visitCount ? obj.visitCount : 0;
|
|
this.readOnly = obj.readOnly ? obj.readOnly : false;
|
|
this.isSeparator = obj.hasOwnProperty("isSeparator") && obj.isSeparator;
|
|
|
|
// And now, the attribute for whether or not this object should appear in the
|
|
// resulting query
|
|
this.isInQuery = obj.isInQuery ? obj.isInQuery : false;
|
|
}
|
|
|
|
// All attributes are set in the constructor above
|
|
queryData.prototype = { }
|
|
|
|
|
|
/**
|
|
* Helper function to compare an array of query objects with a result set.
|
|
* It assumes the array of query objects contains the SAME SORT as the result
|
|
* set. It checks the the uri, title, time, and bookmarkIndex properties of
|
|
* the results, where appropriate.
|
|
*/
|
|
function compareArrayToResult(aArray, aRoot) {
|
|
LOG("Comparing Array to Results");
|
|
|
|
var wasOpen = aRoot.containerOpen;
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = true;
|
|
|
|
// check expected number of results against actual
|
|
var expectedResultCount = aArray.filter(function(aEl) { return aEl.isInQuery; }).length;
|
|
if (expectedResultCount != aRoot.childCount) {
|
|
// Debugging code for failures.
|
|
dump_table("moz_places");
|
|
dump_table("moz_historyvisits");
|
|
LOG("Found children:");
|
|
for (let i = 0; i < aRoot.childCount; i++) {
|
|
LOG(aRoot.getChild(i).uri);
|
|
}
|
|
LOG("Expected:");
|
|
for (let i = 0; i < aArray.length; i++) {
|
|
if (aArray[i].isInQuery)
|
|
LOG(aArray[i].uri);
|
|
}
|
|
}
|
|
do_check_eq(expectedResultCount, aRoot.childCount);
|
|
|
|
var inQueryIndex = 0;
|
|
for (var i = 0; i < aArray.length; i++) {
|
|
if (aArray[i].isInQuery) {
|
|
var child = aRoot.getChild(inQueryIndex);
|
|
//LOG("testing testData[" + i + "] vs result[" + inQueryIndex + "]");
|
|
if (!aArray[i].isFolder && !aArray[i].isSeparator) {
|
|
LOG("testing testData[" + aArray[i].uri + "] vs result[" + child.uri + "]");
|
|
if (aArray[i].uri != child.uri) {
|
|
dump_table("moz_places");
|
|
do_throw("Expected " + aArray[i].uri + " found " + child.uri);
|
|
}
|
|
}
|
|
if (!aArray[i].isSeparator && aArray[i].title != child.title)
|
|
do_throw("Expected " + aArray[i].title + " found " + child.title);
|
|
if (aArray[i].hasOwnProperty("lastVisit") &&
|
|
aArray[i].lastVisit != child.time)
|
|
do_throw("Expected " + aArray[i].lastVisit + " found " + child.time);
|
|
if (aArray[i].hasOwnProperty("index") &&
|
|
aArray[i].index != PlacesUtils.bookmarks.DEFAULT_INDEX &&
|
|
aArray[i].index != child.bookmarkIndex)
|
|
do_throw("Expected " + aArray[i].index + " found " + child.bookmarkIndex);
|
|
|
|
inQueryIndex++;
|
|
}
|
|
}
|
|
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = false;
|
|
LOG("Comparing Array to Results passes");
|
|
}
|
|
|
|
|
|
/**
|
|
* Helper function to check to see if one object either is or is not in the
|
|
* result set. It can accept either a queryData object or an array of queryData
|
|
* objects. If it gets an array, it only compares the first object in the array
|
|
* to see if it is in the result set.
|
|
* Returns: True if item is in query set, and false if item is not in query set
|
|
* If input is an array, returns True if FIRST object in array is in
|
|
* query set. To compare entire array, use the function above.
|
|
*/
|
|
function isInResult(aQueryData, aRoot) {
|
|
var rv = false;
|
|
var uri;
|
|
var wasOpen = aRoot.containerOpen;
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = true;
|
|
|
|
// If we have an array, pluck out the first item. If an object, pluc out the
|
|
// URI, we just compare URI's here.
|
|
if ("uri" in aQueryData) {
|
|
uri = aQueryData.uri;
|
|
} else {
|
|
uri = aQueryData[0].uri;
|
|
}
|
|
|
|
for (var i=0; i < aRoot.childCount; i++) {
|
|
if (uri == aRoot.getChild(i).uri) {
|
|
rv = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = false;
|
|
return rv;
|
|
}
|
|
|
|
|
|
/**
|
|
* A nice helper function for debugging things. It LOGs the contents of a
|
|
* result set.
|
|
*/
|
|
function displayResultSet(aRoot) {
|
|
|
|
var wasOpen = aRoot.containerOpen;
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = true;
|
|
|
|
if (!aRoot.hasChildren) {
|
|
// Something wrong? Empty result set?
|
|
LOG("Result Set Empty");
|
|
return;
|
|
}
|
|
|
|
for (var i=0; i < aRoot.childCount; ++i) {
|
|
LOG("Result Set URI: " + aRoot.getChild(i).uri + " Title: " +
|
|
aRoot.getChild(i).title + " Visit Time: " + aRoot.getChild(i).time);
|
|
}
|
|
if (!wasOpen)
|
|
aRoot.containerOpen = false;
|
|
}
|