1 files changed, 81 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_cookies.js b/browser/components/sessionstore/test/browser_cookies.js
new file mode 100644
index 0000000000..f514efc777
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_cookies.js
@@ -0,0 +1,81 @@
+"use strict";
+
+function promiseSetCookie(cookie) {
+  info(`Set-Cookie: ${cookie}`);
+  return Promise.all([
+    waitForCookieChanged(),
+    SpecialPowers.spawn(
+      gBrowser.selectedBrowser,
+      [cookie],
+      passedCookie => (content.document.cookie = passedCookie)
+    ),
+  ]);
+}
+
+function waitForCookieChanged() {
+  return new Promise(resolve => {
+    Services.obs.addObserver(function observer(subj, topic, data) {
+      Services.obs.removeObserver(observer, topic);
+      resolve();
+    }, "session-cookie-changed");
+  });
+}
+
+function cookieExists(host, name, value) {
+  let {
+    cookies: [c],
+  } = JSON.parse(ss.getBrowserState());
+  return c && c.host == host && c.name == name && c.value == value;
+}
+
+// Setup and cleanup.
+add_task(async function test_setup() {
+  registerCleanupFunction(() => {
+    Services.cookies.removeAll();
+  });
+});
+
+// Test session cookie storage.
+add_task(async function test_run() {
+  Services.cookies.removeAll();
+
+  // Add a new tab for testing.
+  gBrowser.selectedTab = BrowserTestUtils.addTab(
+    gBrowser,
+    "http://example.com/"
+  );
+  await promiseBrowserLoaded(gBrowser.selectedBrowser);
+
+  // Add a session cookie.
+  await promiseSetCookie("foo=bar");
+  ok(cookieExists("example.com", "foo", "bar"), "cookie was added");
+
+  // Modify a session cookie.
+  await promiseSetCookie("foo=baz");
+  ok(cookieExists("example.com", "foo", "baz"), "cookie was modified");
+
+  // Turn the session cookie into a normal one.
+  let expiry = new Date();
+  expiry.setDate(expiry.getDate() + 2);
+  await promiseSetCookie(`foo=baz; Expires=${expiry.toUTCString()}`);
+  ok(!cookieExists("example.com", "foo", "baz"), "no longer a session cookie");
+
+  // Turn it back into a session cookie.
+  await promiseSetCookie("foo=bar");
+  ok(cookieExists("example.com", "foo", "bar"), "again a session cookie");
+
+  // Remove the session cookie.
+  await promiseSetCookie("foo=; Expires=Thu, 01 Jan 1970 00:00:00 GMT");
+  ok(!cookieExists("example.com", "foo", ""), "cookie was removed");
+
+  // Add a session cookie.
+  await promiseSetCookie("foo=bar");
+  ok(cookieExists("example.com", "foo", "bar"), "cookie was added");
+
+  // Clear all session cookies.
+  Services.cookies.removeAll();
+  ok(!cookieExists("example.com", "foo", "bar"), "cookies were cleared");
+
+  // Cleanup.
+  BrowserTestUtils.removeTab(gBrowser.selectedTab);
+});