summaryrefslogtreecommitdiffstats
path: root/browser/components/sessionstore/test/browser_capabilities.js
diff options
context:
space:
mode:
Diffstat (limited to 'browser/components/sessionstore/test/browser_capabilities.js')
-rw-r--r--browser/components/sessionstore/test/browser_capabilities.js76
1 files changed, 76 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_capabilities.js b/browser/components/sessionstore/test/browser_capabilities.js
new file mode 100644
index 000000000..456e41882
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_capabilities.js
@@ -0,0 +1,76 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+/**
+ * These tests ensures that disabling features by flipping nsIDocShell.allow*
+ * properties are (re)stored as disabled. Disallowed features must be
+ * re-enabled when the tab is re-used by another tab restoration.
+ */
+add_task(function docshell_capabilities() {
+ let tab = yield createTab();
+ let browser = tab.linkedBrowser;
+ let docShell = browser.docShell;
+
+ // Get the list of capabilities for docShells.
+ let flags = Object.keys(docShell).filter(k => k.startsWith("allow"));
+
+ // Check that everything is allowed by default for new tabs.
+ let state = JSON.parse(ss.getTabState(tab));
+ ok(!("disallow" in state), "everything allowed by default");
+ ok(flags.every(f => docShell[f]), "all flags set to true");
+
+ // Flip a couple of allow* flags.
+ docShell.allowImages = false;
+ docShell.allowMetaRedirects = false;
+
+ // Now reload the document to ensure that these capabilities
+ // are taken into account.
+ browser.reload();
+ yield promiseBrowserLoaded(browser);
+
+ // Flush to make sure chrome received all data.
+ yield TabStateFlusher.flush(browser);
+
+ // Check that we correctly save disallowed features.
+ let disallowedState = JSON.parse(ss.getTabState(tab));
+ let disallow = new Set(disallowedState.disallow.split(","));
+ ok(disallow.has("Images"), "images not allowed");
+ ok(disallow.has("MetaRedirects"), "meta redirects not allowed");
+ is(disallow.size, 2, "two capabilities disallowed");
+
+ // Reuse the tab to restore a new, clean state into it.
+ yield promiseTabState(tab, {entries: [{url: "about:robots"}]});
+
+ // Flush to make sure chrome received all data.
+ yield TabStateFlusher.flush(browser);
+
+ // After restoring disallowed features must be available again.
+ state = JSON.parse(ss.getTabState(tab));
+ ok(!("disallow" in state), "everything allowed again");
+ ok(flags.every(f => docShell[f]), "all flags set to true");
+
+ // Restore the state with disallowed features.
+ yield promiseTabState(tab, disallowedState);
+
+ // Check that docShell flags are set.
+ ok(!docShell.allowImages, "images not allowed");
+ ok(!docShell.allowMetaRedirects, "meta redirects not allowed");
+
+ // Check that we correctly restored features as disabled.
+ state = JSON.parse(ss.getTabState(tab));
+ disallow = new Set(state.disallow.split(","));
+ ok(disallow.has("Images"), "images not allowed anymore");
+ ok(disallow.has("MetaRedirects"), "meta redirects not allowed anymore");
+ is(disallow.size, 2, "two capabilities disallowed");
+
+ // Clean up after ourselves.
+ gBrowser.removeTab(tab);
+});
+
+function createTab() {
+ let tab = gBrowser.addTab("about:mozilla");
+ let browser = tab.linkedBrowser;
+ return promiseBrowserLoaded(browser).then(() => tab);
+}