1 files changed, 166 insertions, 0 deletions
diff --git a/browser/components/enterprisepolicies/tests/browser/browser_policy_allowfileselectiondialogs.js b/browser/components/enterprisepolicies/tests/browser/browser_policy_allowfileselectiondialogs.js
new file mode 100644
index 0000000000..94f6bba631
--- /dev/null
+++ b/browser/components/enterprisepolicies/tests/browser/browser_policy_allowfileselectiondialogs.js
@@ -0,0 +1,166 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+add_setup(async function setup() {
+  await setupPolicyEngineWithJson({
+    policies: {
+      AllowFileSelectionDialogs: false,
+    },
+  });
+});
+
+add_task(async function test_file_commands_disabled() {
+  // Since testing will apply the policy after the browser has already started,
+  // we will need to open a new window to actually see changes from the policy
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  let savePageCommand = newWin.document.getElementById("Browser:SavePage");
+  let openFileCommand = newWin.document.getElementById("Browser:OpenFile");
+
+  Assert.equal(
+    savePageCommand.getAttribute("disabled"),
+    "true",
+    "Browser:SavePage command is disabled"
+  );
+  Assert.equal(
+    openFileCommand.getAttribute("disabled"),
+    "true",
+    "Browser:OpenFile command is disabled"
+  );
+  await BrowserTestUtils.closeWindow(newWin);
+});
+
+add_task(async function test_file_buttons_disabled() {
+  // Since testing will apply the policy after the browser has already started,
+  // we will need to open a new window to actually see changes from the policy
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  newWin.CustomizableUI.addWidgetToArea("save-page-button", "nav-bar");
+  newWin.CustomizableUI.addWidgetToArea("open-file-button", "nav-bar");
+
+  let savePageButton = newWin.document.getElementById("save-page-button");
+  let openFileButton = newWin.document.getElementById("open-file-button");
+
+  Assert.equal(
+    savePageButton.getAttribute("disabled"),
+    "true",
+    "save-page-button is disabled"
+  );
+  Assert.equal(
+    openFileButton.getAttribute("disabled"),
+    "true",
+    "open-file-button is disabled"
+  );
+
+  newWin.CustomizableUI.reset();
+  await BrowserTestUtils.closeWindow(newWin);
+});
+
+add_task(async function test_context_menu_items_disabled() {
+  await BrowserTestUtils.withNewTab("https://example.org/", async browser => {
+    let contextMenu = document.getElementById("contentAreaContextMenu");
+    let promiseContextMenuOpen = BrowserTestUtils.waitForEvent(
+      contextMenu,
+      "popupshown"
+    );
+
+    await BrowserTestUtils.synthesizeMouse(
+      "a",
+      0,
+      0,
+      {
+        type: "contextmenu",
+        button: 2,
+        centered: true,
+      },
+      browser
+    );
+
+    await promiseContextMenuOpen;
+
+    for (let item of [
+      "context-saveimage",
+      "context-savepage",
+      "context-savelink",
+      "context-savevideo",
+      "context-saveaudio",
+      "context-video-saveimage",
+      "context-saveaudio",
+    ]) {
+      Assert.equal(
+        document.getElementById(item).disabled,
+        true,
+        `${item} item is disabled`
+      );
+    }
+
+    contextMenu.hidePopup();
+  });
+});
+
+add_task(async function test_notification() {
+  // Since testing will apply the policy after the browser has already started,
+  // we will need to open a new window to actually see changes from the policy
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  await BrowserTestUtils.withNewTab(
+    { gBrowser: newWin.gBrowser, url: "https://example.org/" },
+    async browser => {
+      await SpecialPowers.spawn(browser, [], async function () {
+        let elem = content.document.createElement("input");
+        elem.setAttribute("type", "file");
+
+        content.document.notifyUserGestureActivation();
+        elem.showPicker();
+      });
+
+      let notificationBox = browser.getTabBrowser().getNotificationBox(browser);
+
+      let notification = await TestUtils.waitForCondition(() =>
+        notificationBox.getNotificationWithValue("filepicker-blocked")
+      );
+
+      Assert.ok(
+        notification,
+        "filepicker-blocked notification appears when showPicker is called"
+      );
+    }
+  );
+  await BrowserTestUtils.closeWindow(newWin);
+});
+
+add_task(async function test_cancel_event() {
+  await BrowserTestUtils.withNewTab("https://example.org/", async browser => {
+    let eventType = await SpecialPowers.spawn(browser, [], async function () {
+      let elem = content.document.createElement("input");
+      elem.setAttribute("type", "file");
+      let cancelPromise = new Promise(resolve =>
+        elem.addEventListener("cancel", resolve, { once: true })
+      );
+      content.document.notifyUserGestureActivation();
+      elem.showPicker();
+      let event = await cancelPromise;
+      return event.type;
+    });
+
+    Assert.equal(
+      eventType,
+      "cancel",
+      "cancel event should be dispatched when showPicker is called"
+    );
+  });
+});
+
+add_task(async function test_nsIFilePicker_open() {
+  let picker = Cc["@mozilla.org/filepicker;1"].createInstance(Ci.nsIFilePicker);
+
+  picker.init(window, "", Ci.nsIFilePicker.modeSave);
+
+  let result = await new Promise(resolve => picker.open(res => resolve(res)));
+
+  Assert.equal(
+    result,
+    Ci.nsIFilePicker.returnCancel,
+    "nsIFilePicker.open callback should immediately answer returnCancel"
+  );
+});