fune/browser/components/sessionstore/test/browser_undoCloseById.js

/* eslint-disable mozilla/no-arbitrary-setTimeout */
"use strict";

/**
 * This test is for the undoCloseById function.
 */

async function openWindow(url) {
  let win = await promiseNewWindowLoaded();
  let flags = Ci.nsIWebNavigation.LOAD_FLAGS_REPLACE_HISTORY;
  BrowserTestUtils.loadURIString(win.gBrowser.selectedBrowser, url, { flags });
  await promiseBrowserLoaded(win.gBrowser.selectedBrowser, true, url);
  return win;
}

async function closeWindow(win) {
  await BrowserTestUtils.closeWindow(win);
  // Wait 20 ms to allow SessionStorage a chance to register the closed window.
  await new Promise(resolve => setTimeout(resolve, 20));
}

function getLastClosedTabData(win) {
  const closedTabs = SessionStore.getClosedTabData(win);
  return closedTabs[closedTabs.length - 1];
}

add_task(async function test_undoCloseById() {
  // Clear the lists of closed windows and tabs.
  forgetClosedWindows();
  for (const win of SessionStore.getWindows()) {
    while (SessionStore.getClosedTabCountForWindow(win)) {
      SessionStore.forgetClosedTab(win, 0);
    }
  }
  SessionStore.resetNextClosedId();

  // Open a new window.
  let win = await openWindow("about:robots");

  // Open and close a tab.
  await openAndCloseTab(win, "about:mozilla");
  is(
    SessionStore.lastClosedObjectType,
    "tab",
    "The last closed object is a tab"
  );

  // Record the first closedId created.
  is(1, SessionStore.getClosedTabCount(), "We have 1 closed tab");
  let initialClosedId = SessionStore.getClosedTabDataForWindow(win)[0].closedId;

  // Open and close another window.
  let win2 = await openWindow("about:mozilla");
  await closeWindow(win2); // closedId == initialClosedId + 1
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );

  // Open and close another tab in the first window.
  await openAndCloseTab(win, "about:robots"); // closedId == initialClosedId + 2
  is(
    SessionStore.lastClosedObjectType,
    "tab",
    "The last closed object is a tab"
  );

  // Undo closing the second tab.
  let tab = SessionStore.undoCloseById(initialClosedId + 2);
  await promiseBrowserLoaded(tab.linkedBrowser);
  is(
    tab.linkedBrowser.currentURI.spec,
    "about:robots",
    "The expected tab was re-opened"
  );

  let notTab = SessionStore.undoCloseById(initialClosedId + 2);
  is(notTab, undefined, "Re-opened tab cannot be unClosed again by closedId");

  // Now the last closed object should be a window again.
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );

  // Undo closing the first tab.
  let tab2 = SessionStore.undoCloseById(initialClosedId);
  await promiseBrowserLoaded(tab2.linkedBrowser);
  is(
    tab2.linkedBrowser.currentURI.spec,
    "about:mozilla",
    "The expected tab was re-opened"
  );

  // Close the two tabs we re-opened.
  await promiseRemoveTabAndSessionState(tab); // closedId == initialClosedId + 3
  is(
    SessionStore.lastClosedObjectType,
    "tab",
    "The last closed object is a tab"
  );
  await promiseRemoveTabAndSessionState(tab2); // closedId == initialClosedId + 4
  is(
    SessionStore.lastClosedObjectType,
    "tab",
    "The last closed object is a tab"
  );

  // Open another new window.
  let win3 = await openWindow("about:mozilla");

  // Close both windows.
  await closeWindow(win); // closedId == initialClosedId + 5
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );
  await closeWindow(win3); // closedId == initialClosedId + 6
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );

  // Undo closing the second window.
  win = SessionStore.undoCloseById(initialClosedId + 6);
  await BrowserTestUtils.waitForEvent(win, "load");

  // Make sure we wait until this window is restored.
  await BrowserTestUtils.waitForEvent(
    win.gBrowser.tabContainer,
    "SSTabRestored"
  );

  is(
    win.gBrowser.selectedBrowser.currentURI.spec,
    "about:mozilla",
    "The expected window was re-opened"
  );

  let notWin = SessionStore.undoCloseById(initialClosedId + 6);
  is(
    notWin,
    undefined,
    "Re-opened window cannot be unClosed again by closedId"
  );

  // Close the window again.
  await closeWindow(win);
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );

  // Undo closing the first window.
  win = SessionStore.undoCloseById(initialClosedId + 5);

  await BrowserTestUtils.waitForEvent(win, "load");

  // Make sure we wait until this window is restored.
  await BrowserTestUtils.waitForEvent(
    win.gBrowser.tabContainer,
    "SSTabRestored"
  );

  is(
    win.gBrowser.selectedBrowser.currentURI.spec,
    "about:robots",
    "The expected window was re-opened"
  );

  // Close the window again.
  await closeWindow(win);
  is(
    SessionStore.lastClosedObjectType,
    "window",
    "The last closed object is a window"
  );
});