1 files changed, 88 insertions, 0 deletions
diff --git a/browser/components/extensions/test/browser/browser_ext_popup_focus.js b/browser/components/extensions/test/browser/browser_ext_popup_focus.js
new file mode 100644
index 0000000000..4cf46f2be5
--- /dev/null
+++ b/browser/components/extensions/test/browser/browser_ext_popup_focus.js
@@ -0,0 +1,88 @@
+/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set sts=2 sw=2 et tw=80: */
+"use strict";
+
+const DUMMY_PAGE =
+  "http://example.com/browser/browser/components/extensions/test/browser/file_dummy.html";
+
+add_task(async function testPageActionFocus() {
+  let extension = ExtensionTestUtils.loadExtension({
+    manifest: {
+      page_action: {
+        default_popup: "popup.html",
+        show_matches: ["<all_urls>"],
+      },
+    },
+    files: {
+      "popup.html": `<!DOCTYPE html><html><head><meta charset="utf-8">
+        <script src="popup.js"></script>
+        </head><body>
+        </body></html>
+      `,
+      "popup.js": function () {
+        window.addEventListener(
+          "focus",
+          event => {
+            browser.test.log("extension popup received focus event");
+            browser.test.assertEq(
+              true,
+              document.hasFocus(),
+              "document should be focused"
+            );
+            browser.test.notifyPass("focused");
+          },
+          { once: true }
+        );
+        browser.test.log(`extension popup loaded`);
+      },
+    },
+  });
+
+  await extension.startup();
+
+  await BrowserTestUtils.withNewTab(DUMMY_PAGE, async () => {
+    await clickPageAction(extension);
+    await extension.awaitFinish("focused");
+    await closePageAction(extension);
+  });
+
+  await extension.unload();
+});
+
+add_task(async function testBrowserActionFocus() {
+  let extension = ExtensionTestUtils.loadExtension({
+    manifest: {
+      browser_action: { default_popup: "popup.html" },
+    },
+    files: {
+      "popup.html": `<!DOCTYPE html><html><head><meta charset="utf-8">
+        <script src="popup.js"></script>
+        </head><body>
+        </body></html>
+      `,
+      "popup.js": function () {
+        window.addEventListener(
+          "focus",
+          event => {
+            browser.test.log("extension popup received focus event");
+            browser.test.assertEq(
+              true,
+              document.hasFocus(),
+              "document should be focused"
+            );
+            browser.test.notifyPass("focused");
+          },
+          { once: true }
+        );
+        browser.test.log(`extension popup loaded`);
+      },
+    },
+  });
+  await extension.startup();
+
+  await clickBrowserAction(extension);
+  await extension.awaitFinish("focused");
+  await closeBrowserAction(extension);
+
+  await extension.unload();
+});