1 files changed, 89 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_pageStyle.js b/browser/components/sessionstore/test/browser_pageStyle.js
new file mode 100644
index 000000000..7abee5d9d
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_pageStyle.js
@@ -0,0 +1,89 @@
+/* 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)
+}