1 files changed, 90 insertions, 0 deletions
diff --git a/comm/mail/components/extensions/test/browser/browser_ext_sessions.js b/comm/mail/components/extensions/test/browser/browser_ext_sessions.js
new file mode 100644
index 0000000000..a77739c145
--- /dev/null
+++ b/comm/mail/components/extensions/test/browser/browser_ext_sessions.js
@@ -0,0 +1,90 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+add_task(async function test_sessions_data() {
+  let extension = ExtensionTestUtils.loadExtension({
+    background: async () => {
+      let [mailTab] = await browser.tabs.query({ mailTab: true });
+      let contentTab = await browser.tabs.create({
+        url: "https://www.example.com",
+      });
+
+      // Check that there is no data at the beginning.
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(mailTab.id, "aKey"),
+        undefined,
+        "Value for aKey should not exist"
+      );
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(contentTab.id, "aKey"),
+        undefined,
+        "Value for aKey should not exist"
+      );
+
+      // Set some data.
+      await browser.sessions.setTabValue(mailTab.id, "aKey", "1234");
+      await browser.sessions.setTabValue(contentTab.id, "aKey", "4321");
+
+      // Check the data is correct.
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(mailTab.id, "aKey"),
+        "1234",
+        "Value for aKey should exist"
+      );
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(contentTab.id, "aKey"),
+        "4321",
+        "Value for aKey should exist"
+      );
+
+      // Update data.
+      await browser.sessions.setTabValue(mailTab.id, "aKey", "12345");
+      await browser.sessions.setTabValue(contentTab.id, "aKey", "54321");
+
+      // Check the data is correct.
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(mailTab.id, "aKey"),
+        "12345",
+        "Value for aKey should exist"
+      );
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(contentTab.id, "aKey"),
+        "54321",
+        "Value for aKey should exist"
+      );
+
+      // Clear data.
+      await browser.sessions.removeTabValue(mailTab.id, "aKey");
+      await browser.sessions.removeTabValue(contentTab.id, "aKey");
+
+      // Check the data is removed.
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(mailTab.id, "aKey"),
+        undefined,
+        "Value for aKey should not exist"
+      );
+      browser.test.assertEq(
+        await browser.sessions.getTabValue(contentTab.id, "aKey"),
+        undefined,
+        "Value for aKey should not exist"
+      );
+
+      await browser.tabs.remove(contentTab.id);
+      browser.test.notifyPass();
+    },
+    manifest: {
+      manifest_version: 2,
+      browser_specific_settings: {
+        gecko: {
+          id: "sessions@mochi.test",
+        },
+      },
+      permissions: ["tabs"],
+    },
+  });
+
+  await extension.startup();
+  await extension.awaitFinish();
+  await extension.unload();
+});