gecko-dev/browser/base/content/test/general/browser_bug817947.js

/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/ */

const URL = "http://mochi.test:8888/browser/";
const PREF = "browser.sessionstore.restore_on_demand";

function test() {
  waitForExplicitFinish();

  Services.prefs.setBoolPref(PREF, true);
  registerCleanupFunction(function() {
    Services.prefs.clearUserPref(PREF);
  });

  preparePendingTab(function(aTab) {
    let win = gBrowser.replaceTabWithWindow(aTab);

    whenDelayedStartupFinished(win, function() {
      let [tab] = win.gBrowser.tabs;

      whenLoaded(tab.linkedBrowser, function() {
        is(
          tab.linkedBrowser.currentURI.spec,
          URL,
          "correct url should be loaded"
        );
        ok(!tab.hasAttribute("pending"), "tab should not be pending");

        win.close();
        finish();
      });
    });
  });
}

function preparePendingTab(aCallback) {
  let tab = BrowserTestUtils.addTab(gBrowser, URL);

  whenLoaded(tab.linkedBrowser, function() {
    let sessionUpdatePromise = BrowserTestUtils.waitForSessionStoreUpdate(tab);
    BrowserTestUtils.removeTab(tab);
    sessionUpdatePromise.then(() => {
      let [{ state }] = JSON.parse(SessionStore.getClosedTabData(window));

      tab = BrowserTestUtils.addTab(gBrowser, "about:blank");
      whenLoaded(tab.linkedBrowser, function() {
        SessionStore.setTabState(tab, JSON.stringify(state));
        ok(tab.hasAttribute("pending"), "tab should be pending");
        aCallback(tab);
      });
    });
  });
}

function whenLoaded(aBrowser, aCallback) {
  BrowserTestUtils.browserLoaded(aBrowser).then(() => {
    executeSoon(aCallback);
  });
}