tenfourfox/browser/components/sessionstore/test/browser_pageStyle.js
Cameron Kaiser c9b2922b70 hello FPR
2017-04-19 00:56:45 -07:00

90 lines
2.7 KiB
JavaScript

/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
const URL = getRootDirectory(gTestPath) + "browser_pageStyle_sample.html";
const URL_NESTED = getRootDirectory(gTestPath) + "browser_pageStyle_sample_nested.html";
/**
* This test ensures that page style information is correctly persisted.
*/
add_task(function page_style() {
let tab = gBrowser.addTab(URL);
let browser = tab.linkedBrowser;
yield promiseBrowserLoaded(browser);
let sheets = yield getStyleSheets(browser);
// Enable all style sheets one by one.
for (let [title, disabled] of sheets) {
yield enableStyleSheetsForSet(browser, title);
let tab2 = gBrowser.duplicateTab(tab);
yield promiseTabRestored(tab2);
let sheets = yield getStyleSheets(tab2.linkedBrowser);
let enabled = sheets.filter(([title, disabled]) => !disabled);
if (title.startsWith("fail_")) {
ok(!enabled.length, "didn't restore " + title);
} else {
is(enabled.length, 1, "restored one style sheet");
is(enabled[0][0], title, "restored correct sheet");
}
gBrowser.removeTab(tab2);
}
// Disable all styles and verify that this is correctly persisted.
yield setAuthorStyleDisabled(browser, true);
let tab2 = gBrowser.duplicateTab(tab);
yield promiseTabRestored(tab2);
let authorStyleDisabled = yield getAuthorStyleDisabled(tab2.linkedBrowser);
ok(authorStyleDisabled, "disabled all stylesheets");
// Clean up.
gBrowser.removeTab(tab);
gBrowser.removeTab(tab2);
});
/**
* This test ensures that page style notification from nested documents are
* received and the page style is persisted correctly.
*/
add_task(function nested_page_style() {
let tab = gBrowser.addTab(URL_NESTED);
let browser = tab.linkedBrowser;
yield promiseBrowserLoaded(browser);
yield enableSubDocumentStyleSheetsForSet(browser, "alternate");
yield promiseRemoveTab(tab);
let [{state: {pageStyle}}] = JSON.parse(ss.getClosedTabData(window));
let expected = JSON.stringify({children: [{pageStyle: "alternate"}]});
is(JSON.stringify(pageStyle), expected, "correct pageStyle persisted");
});
function getStyleSheets(browser) {
return sendMessage(browser, "ss-test:getStyleSheets");
}
function enableStyleSheetsForSet(browser, name) {
return sendMessage(browser, "ss-test:enableStyleSheetsForSet", name);
}
function enableSubDocumentStyleSheetsForSet(browser, name) {
return sendMessage(browser, "ss-test:enableSubDocumentStyleSheetsForSet", {
id: "iframe", set: name
});
}
function getAuthorStyleDisabled(browser) {
return sendMessage(browser, "ss-test:getAuthorStyleDisabled");
}
function setAuthorStyleDisabled(browser, val) {
return sendMessage(browser, "ss-test:setAuthorStyleDisabled", val)
}