summaryrefslogtreecommitdiffstats
path: root/browser/components/sessionstore/test/browser_394759_basic.js
blob: 442f04cd0ce2304133ebe5f5fcd7fe858175d087 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/* 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.importESModule(
  "resource:///modules/sessionstore/SessionStore.sys.mjs"
);

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

    setPropertyOfFormField(browser, "#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();

        // 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);
}