101 lines
3.3 KiB
JavaScript
101 lines
3.3 KiB
JavaScript
if (typeof(classifierHelper) == "undefined") {
|
|
var classifierHelper = {};
|
|
}
|
|
|
|
const CLASSIFIER_COMMON_URL = SimpleTest.getTestFileURL("classifierCommon.js");
|
|
var classifierCommonScript = SpecialPowers.loadChromeScript(CLASSIFIER_COMMON_URL);
|
|
|
|
// addUrlToDB & removeUrlFromDB are asynchronous, queue the task to ensure
|
|
// the callback follow correct order.
|
|
classifierHelper._updates = [];
|
|
|
|
// Keep urls added to database, those urls should be automatically
|
|
// removed after test complete.
|
|
classifierHelper._updatesToCleanup = [];
|
|
|
|
// Pass { url: ..., db: ... } to add url to database,
|
|
// onsuccess/onerror will be called when update complete.
|
|
classifierHelper.addUrlToDB = function(updateData, onsuccess, onerror) {
|
|
var testUpdate = "";
|
|
for (var update of updateData) {
|
|
classifierHelper._updatesToCleanup.push(update);
|
|
testUpdate +=
|
|
"n:1000\ni:" + update.db + "\nad:1\n" +
|
|
"a:524:32:" + update.url.length + "\n" +
|
|
update.url;
|
|
}
|
|
|
|
classifierHelper._update(testUpdate, onsuccess, onerror);
|
|
}
|
|
|
|
// Pass { url: ..., db: ... } to remove url from database,
|
|
// onsuccess/onerror will be called when update complete.
|
|
classifierHelper.removeUrlFromDB = function(updateData, onsuccess, onerror) {
|
|
var testUpdate = "";
|
|
for (var update of updateData) {
|
|
testUpdate +=
|
|
"n:1000\ni:" + update.db + "\nsd:1\n" +
|
|
"s:524:32:" + (4 + update.url.length) + "\n" +
|
|
"524:" + update.url;
|
|
}
|
|
|
|
classifierHelper._updatesToCleanup =
|
|
classifierHelper._updatesToCleanup.filter((v) => {
|
|
return updateData.indexOf(v) == -1;
|
|
});
|
|
|
|
classifierHelper._update(testUpdate, onsuccess, onerror);
|
|
};
|
|
|
|
classifierHelper._update = function(testUpdate, onsuccess, onerror) {
|
|
// Queue the task if there is still an on-going update
|
|
classifierHelper._updates.push({"data": testUpdate,
|
|
"onsuccess": onsuccess,
|
|
"onerror": onerror});
|
|
if (classifierHelper._updates.length != 1) {
|
|
return;
|
|
}
|
|
|
|
classifierCommonScript.sendAsyncMessage("doUpdate", { testUpdate });
|
|
};
|
|
|
|
classifierHelper._updateSuccess = function() {
|
|
var update = classifierHelper._updates.shift();
|
|
update.onsuccess();
|
|
|
|
if (classifierHelper._updates.length) {
|
|
var testUpdate = classifierHelper._updates[0].data;
|
|
classifierCommonScript.sendAsyncMessage("doUpdate", { testUpdate });
|
|
}
|
|
};
|
|
|
|
classifierHelper._updateError = function(errorCode) {
|
|
var update = classifierHelper._updates.shift();
|
|
update.onerror(errorCode);
|
|
|
|
if (classifierHelper._updates.length) {
|
|
var testUpdate = classifierHelper._updates[0].data;
|
|
classifierCommonScript.sendAsyncMessage("doUpdate", { testUpdate });
|
|
}
|
|
};
|
|
|
|
classifierHelper._setup = function() {
|
|
classifierCommonScript.addMessageListener("updateSuccess", classifierHelper._updateSuccess);
|
|
classifierCommonScript.addMessageListener("updateError", classifierHelper._updateError);
|
|
|
|
// cleanup will be called at end of each testcase to remove all the urls added to database.
|
|
SimpleTest.registerCleanupFunction(classifierHelper._cleanup);
|
|
};
|
|
|
|
classifierHelper._cleanup = function() {
|
|
if (!classifierHelper._updatesToCleanup) {
|
|
return Promise.resolve();
|
|
}
|
|
|
|
return new Promise(function(resolve, reject) {
|
|
classifierHelper.removeUrlFromDB(classifierHelper._updatesToCleanup, resolve, reject);
|
|
});
|
|
};
|
|
|
|
classifierHelper._setup();
|