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
129
130
131
132
133
134
135
136
137
138
139
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
add_task(async function elevation_dialog() {
await SpecialPowers.pushPrefEnv({
set: [[PREF_APP_UPDATE_DISABLEDFORTESTING, false]],
});
// Create a mock of nsIAppStartup's quit method so clicking the restart button
// won't restart the application.
let { startup } = Services;
let appStartup = {
QueryInterface: ChromeUtils.generateQI(["nsIAppStartup"]),
quit(mode) {
if (elevationDialog) {
elevationDialog.close();
elevationDialog = null;
}
},
};
Services.startup = appStartup;
registerCleanupFunction(() => {
Services.startup = startup;
});
registerCleanupFunction(async () => {
let win = Services.wm.getMostRecentWindow("Update:Elevation");
if (win) {
win.close();
await TestUtils.waitForCondition(
() => !Services.wm.getMostRecentWindow("Update:Elevation"),
"The Update Elevation dialog should have closed"
);
}
});
// Test clicking the "Restart Later" button
let elevationDialog = await waitForElevationDialog();
await TestUtils.waitForTick();
elevationDialog.document.getElementById("elevateExtra2").click();
await TestUtils.waitForCondition(
() => !Services.wm.getMostRecentWindow("Update:Elevation"),
"The Update Elevation dialog should have closed"
);
ok(!!gUpdateManager.readyUpdate, "There should be a ready update");
is(
gUpdateManager.readyUpdate.state,
STATE_PENDING_ELEVATE,
"The ready update state should equal " + STATE_PENDING_ELEVATE
);
is(
readStatusFile(),
STATE_PENDING_ELEVATE,
"The status file state should equal " + STATE_PENDING_ELEVATE
);
// Test clicking the "No Thanks" button
elevationDialog = await waitForElevationDialog();
await TestUtils.waitForTick();
elevationDialog.document.getElementById("elevateExtra1").click();
await TestUtils.waitForCondition(
() => !Services.wm.getMostRecentWindow("Update:Elevation"),
"The Update Elevation dialog should have closed"
);
ok(!gUpdateManager.readyUpdate, "There should not be a ready update");
is(
readStatusFile(),
STATE_NONE,
"The status file state should equal " + STATE_NONE
);
// Test clicking the "Restart <brandShortName>" button
elevationDialog = await waitForElevationDialog();
await TestUtils.waitForTick();
elevationDialog.document.getElementById("elevateAccept").click();
await TestUtils.waitForCondition(
() => !Services.wm.getMostRecentWindow("Update:Elevation"),
"The Update Elevation dialog should have closed"
);
ok(!!gUpdateManager.readyUpdate, "There should be a ready update");
is(
gUpdateManager.readyUpdate.state,
STATE_PENDING_ELEVATE,
"The active update state should equal " + STATE_PENDING_ELEVATE
);
is(
readStatusFile(),
STATE_PENDING,
"The status file state should equal " + STATE_PENDING
);
});
/**
* Waits for the Update Elevation Dialog to load.
*
* @return A promise that returns the domWindow for the Update Elevation Dialog
* and resolves when the Update Elevation Dialog loads.
*/
function waitForElevationDialog() {
return new Promise(resolve => {
var listener = {
onOpenWindow: aXULWindow => {
debugDump("Update Elevation dialog shown...");
Services.wm.removeListener(listener);
async function elevationDialogOnLoad() {
domwindow.removeEventListener("load", elevationDialogOnLoad, true);
let chromeURI =
"chrome://mozapps/content/update/updateElevation.xhtml";
is(
domwindow.document.location.href,
chromeURI,
"Update Elevation appeared"
);
resolve(domwindow);
}
var domwindow = aXULWindow.docShell.domWindow;
domwindow.addEventListener("load", elevationDialogOnLoad, true);
},
onCloseWindow: aXULWindow => {},
};
Services.wm.addListener(listener);
// Add the active-update.xml and update.status files used for these tests,
// reload the update manager, and then simulate startup so the Update
// Elevation Dialog is opened.
let patchProps = { state: STATE_PENDING_ELEVATE };
let patches = getLocalPatchString(patchProps);
let updateProps = { checkInterval: "1" };
let updates = getLocalUpdateString(updateProps, patches);
writeUpdatesToXMLFile(getLocalUpdatesXMLString(updates), true);
writeStatusFile(STATE_PENDING_ELEVATE);
reloadUpdateManagerData();
testPostUpdateProcessing();
});
}
|
