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

"use strict";

/**
 * Ensure that history entries that should not be persisted are restored in the
 * same state.
 */
add_task(async function check_history_not_persisted() {
  // Create an about:blank tab
  let tab = BrowserTestUtils.addTab(gBrowser, "about:blank");
  let browser = tab.linkedBrowser;
  await promiseBrowserLoaded(browser);

  // Retrieve the tab state.
  await TabStateFlusher.flush(browser);
  let state = JSON.parse(ss.getTabState(tab));
  ok(!state.entries[0].persist, "Should have collected the persistence state");
  BrowserTestUtils.removeTab(tab);
  browser = null;

  // Open a new tab to restore into.
  tab = BrowserTestUtils.addTab(gBrowser, "about:blank");
  browser = tab.linkedBrowser;
  await promiseTabState(tab, state);

  let sessionHistory = browser.browsingContext.sessionHistory;

  is(sessionHistory.count, 1, "Should be a single history entry");
  is(
    sessionHistory.getEntryAtIndex(0).URI.spec,
    "about:blank",
    "Should be the right URL"
  );

  // Load a new URL into the tab, it should replace the about:blank history entry
  BrowserTestUtils.loadURIString(browser, "about:robots");
  await promiseBrowserLoaded(browser, false, "about:robots");

  sessionHistory = browser.browsingContext.sessionHistory;

  is(sessionHistory.count, 1, "Should be a single history entry");
  is(
    sessionHistory.getEntryAtIndex(0).URI.spec,
    "about:robots",
    "Should be the right URL"
  );

  // Cleanup.
  BrowserTestUtils.removeTab(tab);
});

/**
 * Check that entries default to being persisted when the attribute doesn't
 * exist
 */
add_task(async function check_history_default_persisted() {
  // Create an about:blank tab
  let tab = BrowserTestUtils.addTab(gBrowser, "about:blank");
  let browser = tab.linkedBrowser;
  await promiseBrowserLoaded(browser);

  // Retrieve the tab state.
  await TabStateFlusher.flush(browser);
  let state = JSON.parse(ss.getTabState(tab));
  delete state.entries[0].persist;
  BrowserTestUtils.removeTab(tab);
  browser = null;

  // Open a new tab to restore into.
  tab = BrowserTestUtils.addTab(gBrowser, "about:blank");
  browser = tab.linkedBrowser;
  await promiseTabState(tab, state);

  let sessionHistory = browser.browsingContext.sessionHistory;

  is(sessionHistory.count, 1, "Should be a single history entry");
  is(
    sessionHistory.getEntryAtIndex(0).URI.spec,
    "about:blank",
    "Should be the right URL"
  );

  // Load a new URL into the tab, it should replace the about:blank history entry
  BrowserTestUtils.loadURIString(browser, "about:robots");
  await promiseBrowserLoaded(browser, false, "about:robots");

  sessionHistory = browser.browsingContext.sessionHistory;

  is(sessionHistory.count, 2, "Should be two history entries");
  is(
    sessionHistory.getEntryAtIndex(0).URI.spec,
    "about:blank",
    "Should be the right URL"
  );
  is(
    sessionHistory.getEntryAtIndex(1).URI.spec,
    "about:robots",
    "Should be the right URL"
  );

  // Cleanup.
  BrowserTestUtils.removeTab(tab);
});
