c65e29db34
This commit adds a helper module for doing common tasks related to site data, such as adding dummy data and getting usage. There are many places that would potentially need to be cleaned up to use this module instead, but I consider that work (and the likely try failure fallout) out of scope for this bug. MozReview-Commit-ID: 5eMDgHhClsO
79 lines
2.5 KiB
JavaScript
79 lines
2.5 KiB
JavaScript
"use strict";
|
|
|
|
var EXPORTED_SYMBOLS = [
|
|
"SiteDataTestUtils",
|
|
];
|
|
|
|
ChromeUtils.import("resource://gre/modules/Services.jsm");
|
|
const {Sanitizer} = ChromeUtils.import("resource:///modules/Sanitizer.jsm", {});
|
|
|
|
/**
|
|
* This module assists with tasks around testing functionality that shows
|
|
* or clears site data.
|
|
*
|
|
* Please note that you will have to clean up changes made manually, for
|
|
* example using SiteDataTestUtils.clear().
|
|
*/
|
|
var SiteDataTestUtils = {
|
|
|
|
/**
|
|
* Adds a new entry to a dummy indexedDB database for the specified origin.
|
|
*
|
|
* @param {String} origin - the origin of the site to add test data for
|
|
*
|
|
* @returns a Promise that resolves when the data was added successfully.
|
|
*/
|
|
addToIndexedDB(origin) {
|
|
return new Promise(resolve => {
|
|
let uri = Services.io.newURI(origin);
|
|
let principal = Services.scriptSecurityManager.createCodebasePrincipal(uri, {});
|
|
let request = indexedDB.openForPrincipal(principal, "TestDatabase", 1);
|
|
request.onupgradeneeded = function(e) {
|
|
let db = e.target.result;
|
|
db.createObjectStore("TestStore", { keyPath: "id" });
|
|
};
|
|
request.onsuccess = function(e) {
|
|
let db = e.target.result;
|
|
let tx = db.transaction("TestStore", "readwrite");
|
|
let store = tx.objectStore("TestStore");
|
|
tx.oncomplete = resolve;
|
|
store.put({ id: performance.now().toString(), description: "IndexedDB Test"});
|
|
};
|
|
});
|
|
},
|
|
|
|
/**
|
|
* Adds a new cookie for the specified origin, with the specified contents.
|
|
* The cookie will be valid for one day.
|
|
*
|
|
* @param {String} name - the cookie name
|
|
* @param {String} value - the cookie value
|
|
*/
|
|
addToCookies(origin, name, value) {
|
|
let uri = Services.io.newURI(origin);
|
|
Services.cookies.add(uri.host, uri.pathQueryRef, name, value,
|
|
false, false, false, Date.now() + 24000 * 60 * 60);
|
|
},
|
|
|
|
/**
|
|
* Gets the current quota usage for the specified origin.
|
|
*
|
|
* @returns a Promise that resolves to an integer with the total
|
|
* amount of disk usage by a given origin.
|
|
*/
|
|
getQuotaUsage(origin) {
|
|
return new Promise(resolve => {
|
|
let uri = Services.io.newURI(origin);
|
|
let principal = Services.scriptSecurityManager.createCodebasePrincipal(uri, {});
|
|
Services.qms.getUsageForPrincipal(principal, request => resolve(request.result.usage));
|
|
});
|
|
},
|
|
|
|
/**
|
|
* Cleans up all site data.
|
|
*/
|
|
clear() {
|
|
return Sanitizer.sanitize(["cookies", "cache", "siteSettings", "offlineApps"]);
|
|
},
|
|
};
|