Mypal68/browser/components/sessionstore/test/browser_394759_basic.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/. */

"use strict";

const TEST_URL =
  "data:text/html;charset=utf-8,<input%20id=txt>" +
  "<input%20type=checkbox%20id=chk>";

const { SessionStore } = ChromeUtils.import(
  "resource:///modules/sessionstore/SessionStore.jsm"
);

/**
 * This test ensures that closing a window is a reversible action. We will
 * close the the window, restore it and check that all data has been restored.
 * This includes window-specific data as well as form data for tabs.
 */
function test() {
  waitForExplicitFinish();

  let uniqueKey = "bug 394759";
  let uniqueValue = "unik" + Date.now();
  let uniqueText = "pi != " + Math.random();

  // Clear the list of closed windows.
  forgetClosedWindows();

  provideWindow(function onTestURLLoaded(newWin) {
    BrowserTestUtils.addTab(newWin.gBrowser).linkedBrowser.stop();

    // Mark the window with some unique data to be restored later on.
    ss.setCustomWindowValue(newWin, uniqueKey, uniqueValue);
    let [txt] = newWin.content.document.querySelectorAll("#txt");
    txt.value = uniqueText;

    let browser = newWin.gBrowser.selectedBrowser;
    setInputChecked(browser, { id: "chk", checked: true }).then(() => {
      BrowserTestUtils.closeWindow(newWin).then(() => {
        is(
          ss.getClosedWindowCount(),
          1,
          "The closed window was added to Recently Closed Windows"
        );

        let data = SessionStore.getClosedWindowData(false);

        // Verify that non JSON serialized data is the same as JSON serialized data.
        is(
          JSON.stringify(data),
          ss.getClosedWindowData(),
          "Non-serialized data is the same as serialized data"
        );

        ok(
          data[0].title == TEST_URL &&
            JSON.stringify(data[0]).indexOf(uniqueText) > -1,
          "The closed window data was stored correctly"
        );

        // Reopen the closed window and ensure its integrity.
        let newWin2 = ss.undoCloseWindow(0);

        ok(
          newWin2.isChromeWindow,
          "undoCloseWindow actually returned a window"
        );
        is(
          ss.getClosedWindowCount(),
          0,
          "The reopened window was removed from Recently Closed Windows"
        );

        // SSTabRestored will fire more than once, so we need to make sure we count them.
        let restoredTabs = 0;
        let expectedTabs = data[0].tabs.length;
        newWin2.addEventListener(
          "SSTabRestored",
          function sstabrestoredListener(aEvent) {
            ++restoredTabs;
            info("Restored tab " + restoredTabs + "/" + expectedTabs);
            if (restoredTabs < expectedTabs) {
              return;
            }

            is(restoredTabs, expectedTabs, "Correct number of tabs restored");
            newWin2.removeEventListener(
              "SSTabRestored",
              sstabrestoredListener,
              true
            );

            is(
              newWin2.gBrowser.tabs.length,
              2,
              "The window correctly restored 2 tabs"
            );
            is(
              newWin2.gBrowser.currentURI.spec,
              TEST_URL,
              "The window correctly restored the URL"
            );

            let chk;
            [txt, chk] = newWin2.content.document.querySelectorAll(
              "#txt, #chk"
            );
            ok(
              txt.value == uniqueText && chk.checked,
              "The window correctly restored the form"
            );
            is(
              ss.getCustomWindowValue(newWin2, uniqueKey),
              uniqueValue,
              "The window correctly restored the data associated with it"
            );

            // Clean up.
            BrowserTestUtils.closeWindow(newWin2).then(finish);
          },
          true
        );
      });
    });
  }, TEST_URL);
}

function setInputChecked(browser, data) {
  return sendMessage(browser, "ss-test:setInputChecked", data);
}