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
|
"use strict";
/**
* This test is for the undoCloseById function.
*/
Cu.import("resource:///modules/sessionstore/SessionStore.jsm");
function openAndCloseTab(window, url) {
let tab = window.gBrowser.addTab(url);
yield promiseBrowserLoaded(tab.linkedBrowser, true, url);
yield TabStateFlusher.flush(tab.linkedBrowser);
yield promiseRemoveTab(tab);
}
function* openWindow(url) {
let win = yield promiseNewWindowLoaded();
let flags = Ci.nsIWebNavigation.LOAD_FLAGS_REPLACE_HISTORY;
win.gBrowser.selectedBrowser.loadURIWithFlags(url, flags);
yield promiseBrowserLoaded(win.gBrowser.selectedBrowser, true, url);
return win;
}
function closeWindow(win) {
yield BrowserTestUtils.closeWindow(win);
// Wait 20 ms to allow SessionStorage a chance to register the closed window.
yield new Promise(resolve => setTimeout(resolve, 20));
}
add_task(function* test_undoCloseById() {
// Clear the lists of closed windows and tabs.
forgetClosedWindows();
while (SessionStore.getClosedTabCount(window)) {
SessionStore.forgetClosedTab(window, 0);
}
// Open a new window.
let win = yield openWindow("about:robots");
// Open and close a tab.
yield openAndCloseTab(win, "about:mozilla");
is(SessionStore.lastClosedObjectType, "tab", "The last closed object is a tab");
// Record the first closedId created.
let initialClosedId = SessionStore.getClosedTabData(win, false)[0].closedId;
// Open and close another window.
let win2 = yield openWindow("about:mozilla");
yield 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.
yield 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);
yield 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);
yield promiseBrowserLoaded(tab2.linkedBrowser);
is(tab2.linkedBrowser.currentURI.spec, "about:mozilla", "The expected tab was re-opened");
// Close the two tabs we re-opened.
yield promiseRemoveTab(tab); // closedId == initialClosedId + 3
is(SessionStore.lastClosedObjectType, "tab", "The last closed object is a tab");
yield promiseRemoveTab(tab2); // closedId == initialClosedId + 4
is(SessionStore.lastClosedObjectType, "tab", "The last closed object is a tab");
// Open another new window.
let win3 = yield openWindow("about:mozilla");
// Close both windows.
yield closeWindow(win); // closedId == initialClosedId + 5
is(SessionStore.lastClosedObjectType, "window", "The last closed object is a window");
yield 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);
yield BrowserTestUtils.waitForEvent(win, "load");
// Make sure we wait until this window is restored.
yield 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.
yield closeWindow(win);
is(SessionStore.lastClosedObjectType, "window", "The last closed object is a window");
// Undo closing the first window.
win = SessionStore.undoCloseById(initialClosedId + 5);
yield BrowserTestUtils.waitForEvent(win, "load");
// Make sure we wait until this window is restored.
yield BrowserTestUtils.waitForEvent(win.gBrowser.tabContainer,
"SSTabRestored");
is(win.gBrowser.selectedBrowser.currentURI.spec, "about:robots", "The expected window was re-opened");
// Close the window again.
yield closeWindow(win);
is(SessionStore.lastClosedObjectType, "window", "The last closed object is a window");
});
|
