1 files changed, 97 insertions, 0 deletions
diff --git a/toolkit/components/extensions/test/xpcshell/test_ext_storage_session.js b/toolkit/components/extensions/test/xpcshell/test_ext_storage_session.js
new file mode 100644
index 0000000000..9dc0aa5af9
--- /dev/null
+++ b/toolkit/components/extensions/test/xpcshell/test_ext_storage_session.js
@@ -0,0 +1,97 @@
+"use strict";
+
+AddonTestUtils.init(this);
+
+const server = createHttpServer({ hosts: ["example.com"] });
+server.registerDirectory("/data/", do_get_file("data"));
+
+add_setup(async function setup() {
+  await ExtensionTestUtils.startAddonManager();
+});
+
+add_task(async function test_storage_session() {
+  await test_background_page_storage("session");
+});
+
+add_task(async function test_storage_session_onChanged_event_page() {
+  await test_storage_change_event_page("session");
+});
+
+add_task(async function test_storage_session_persistance() {
+  await test_storage_after_reload("session", { expectPersistency: false });
+});
+
+add_task(async function test_storage_session_empty_events() {
+  await test_storage_empty_events("session");
+});
+
+add_task(async function test_storage_session_contentscript() {
+  let extensionData = {
+    manifest: {
+      content_scripts: [
+        {
+          matches: ["http://example.com/data/file_sample.html"],
+          js: ["content_script.js"],
+        },
+      ],
+      permissions: ["storage"],
+    },
+    background() {
+      let events = [];
+      browser.storage.onChanged.addListener((_, area) => {
+        events.push(area);
+      });
+      browser.test.onMessage.addListener(_msg => {
+        browser.test.sendMessage("bg-events", events.join());
+      });
+      browser.runtime.onMessage.addListener(async _msg => {
+        await browser.storage.local.set({ foo: "local" });
+        await browser.storage.session.set({ foo: "session" });
+        await browser.storage.sync.set({ foo: "sync" });
+        browser.test.sendMessage("done");
+      });
+    },
+    files: {
+      "content_script.js"() {
+        let events = [];
+        browser.storage.onChanged.addListener((_, area) => {
+          events.push(area);
+        });
+        browser.test.onMessage.addListener(_msg => {
+          browser.test.sendMessage("cs-events", events.join());
+        });
+
+        browser.test.assertEq(
+          typeof browser.storage.session,
+          "undefined",
+          "Expect storage.session to not be available in content scripts"
+        );
+        browser.runtime.sendMessage("ready");
+      },
+    },
+  };
+
+  let extension = ExtensionTestUtils.loadExtension(extensionData);
+  await extension.startup();
+
+  let contentPage = await ExtensionTestUtils.loadContentPage(
+    "http://example.com/data/file_sample.html"
+  );
+
+  await extension.awaitMessage("done");
+  extension.sendMessage("_getEvents");
+
+  equal(
+    "local,sync",
+    await extension.awaitMessage("cs-events"),
+    "Content script doesn't see storage.onChanged events from the session area."
+  );
+  equal(
+    "local,session,sync",
+    await extension.awaitMessage("bg-events"),
+    "Background receives onChanged events from all storage areas."
+  );
+
+  await extension.unload();
+  await contentPage.close();
+});