Revision control
Copy as Markdown
Other Tools
/* 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
"use strict";
var { close_compose_window, open_compose_new_mail } =
ChromeUtils.importESModule(
);
/**
* Tests that restoring the default toolbar set does not end customization.
*/
add_task(async function test_restoreDefaultToolbarSet() {
const composeWindow = await open_compose_new_mail();
const toolbox = composeWindow.document.getElementById("compose-toolbox");
const toolbar = composeWindow.document.getElementById("composeToolbar2");
Assert.ok(
!toolbar.currentSet.split(",").includes("button-print"),
"The print button should not be in the default toolbar set"
);
toolbar.insertItem("button-print");
Assert.ok(
toolbar.currentSet.split(",").includes("button-print"),
"The print button should be added to the toolbar"
);
const useSheet = Services.prefs.getBoolPref("toolbar.customization.usesheet");
let customizeWindowPromise;
if (useSheet) {
const customizeFrame = composeWindow.document.getElementById(
"customizeToolbarSheetIFrame"
);
customizeWindowPromise = TestUtils.waitForCondition(
() =>
customizeFrame.contentDocument?.documentURI ==
customizeFrame.contentDocument.readyState == "complete",
"The toolbar customization sheet should finish loading"
).then(() => customizeFrame.contentWindow);
} else {
customizeWindowPromise = BrowserTestUtils.domWindowOpenedAndLoaded(
null,
win =>
win.document.documentURI ==
);
}
composeWindow.CustomizeMailToolbar(
"compose-toolbox",
"CustomizeComposeToolbar"
);
const customizeWindow = await customizeWindowPromise;
registerCleanupFunction(async () => {
if (useSheet && toolbox.customizing) {
customizeWindow.onClose();
} else if (!useSheet && !customizeWindow.closed) {
await BrowserTestUtils.closeWindow(customizeWindow);
}
if (!composeWindow.closed) {
await close_compose_window(composeWindow, false);
}
});
Assert.ok(toolbox.customizing, "The toolbar should be in customization mode");
customizeWindow.document.getElementById("restoreDefault").click();
await TestUtils.waitForTick();
Assert.ok(
!customizeWindow.closed,
"The customization window should remain open"
);
Assert.ok(
toolbox.customizing,
"The toolbar should remain in customization mode"
);
Assert.ok(
!toolbar.querySelector("#button-print"),
"The print button should be removed by restoring the default set"
);
const closePromise = customizeWindow.frameElement
? null
: BrowserTestUtils.domWindowClosed(customizeWindow);
customizeWindow.document.getElementById("donebutton").click();
if (closePromise) {
await closePromise;
}
await TestUtils.waitForCondition(
() => !toolbox.customizing,
"Customization mode should end after clicking Done"
);
Assert.ok(!toolbox.customizing, "Customization mode should end when done");
await close_compose_window(composeWindow, false);
});