tenfourfox/browser/components/sessionstore/test/browser_423132.js

/* 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/. */

function test() {
  // test that cookies are stored and restored correctly by sessionstore,
  // bug 423132.

  waitForExplicitFinish();

  let cs = Cc["@mozilla.org/cookiemanager;1"].getService(Ci.nsICookieManager2);
  cs.removeAll();

  // make sure that sessionstore.js can be forced to be created by setting
  // the interval pref to 0
  gPrefService.setIntPref("browser.sessionstore.interval", 0);

  const testURL = "http://mochi.test:8888/browser/" +
    "browser/components/sessionstore/test/browser_423132_sample.html";

  // open a new window
  let newWin = openDialog(location, "_blank", "chrome,all,dialog=no", "about:blank");

  // make sure sessionstore saves the cookie data, then close the window
  newWin.addEventListener("load", function (aEvent) {
    newWin.removeEventListener("load", arguments.callee, false);

    // Wait for sessionstore to be ready to restore this window
    executeSoon(function() {
      newWin.gBrowser.loadURI(testURL, null, null);

      promiseBrowserLoaded(newWin.gBrowser.selectedBrowser).then(() => {
        let ready = () => {
          // get the sessionstore state for the window
          let state = ss.getWindowState(newWin);

          // verify our cookie got set during pageload
          let e = cs.enumerator;
          let cookie;
          let i = 0;
          while (e.hasMoreElements()) {
            cookie = e.getNext().QueryInterface(Ci.nsICookie);
            i++;
          }
          is(i, 1, "expected one cookie");

          // remove the cookie
          cs.removeAll();

          // restore the window state
          ss.setWindowState(newWin, state, true);

          // at this point, the cookie should be restored...
          e = cs.enumerator;
          let cookie2;
          while (e.hasMoreElements()) {
            cookie2 = e.getNext().QueryInterface(Ci.nsICookie);
            if (cookie.name == cookie2.name)
              break;
          }
          is(cookie.name, cookie2.name, "cookie name successfully restored");
          is(cookie.value, cookie2.value, "cookie value successfully restored");
          is(cookie.path, cookie2.path, "cookie path successfully restored");

          // clean up
          if (gPrefService.prefHasUserValue("browser.sessionstore.interval"))
            gPrefService.clearUserPref("browser.sessionstore.interval");
          cs.removeAll();
          BrowserTestUtils.closeWindow(newWin).then(finish);
        };

        function flushAndReady() {
          TabStateFlusher.flush(newWin.gBrowser.selectedBrowser).then(ready);
        }

        if (newWin.gMultiProcessBrowser) {
          let tab = newWin.gBrowser.selectedTab;
          promiseTabRestored(tab).then(flushAndReady);
        } else {
          flushAndReady();
        }
      }, true, testURL);
    });
  }, false);
}