1 files changed, 91 insertions, 0 deletions
diff --git a/devtools/client/debugger/test/mochitest/browser_dbg-slow-script.js b/devtools/client/debugger/test/mochitest/browser_dbg-slow-script.js
new file mode 100644
index 0000000000..246c13a4e7
--- /dev/null
+++ b/devtools/client/debugger/test/mochitest/browser_dbg-slow-script.js
@@ -0,0 +1,91 @@
+/* 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
+ * file, You can obtain one at <http://mozilla.org/MPL/2.0/>. */
+
+"use strict";
+
+// Tests the slow script warning
+
+add_task(async function openDebuggerFirst() {
+  // In mochitest, the timeout is disable, so set it to a short, but non zero duration
+  await pushPref("dom.max_script_run_time", 1);
+  // Prevents having to click on the page to have the dialog to appear
+  await pushPref("dom.max_script_run_time.require_critical_input", false);
+
+  const dbg = await initDebugger("doc-slow-script.html");
+
+  const alert = BrowserTestUtils.waitForGlobalNotificationBar(
+    window,
+    "process-hang"
+  );
+
+  info("Execute an infinite loop");
+  invokeInTab("infiniteLoop");
+
+  info("Wait for the slow script warning");
+  const notification = await alert;
+
+  info("Click on the debug script button");
+  const buttons = notification.buttonContainer.getElementsByTagName("button");
+  // The first button is "stop", the second is "debug script"
+  buttons[1].click();
+
+  info("Waiting for the debugger to be paused");
+  await waitForPaused(dbg);
+  const source = findSource(dbg, "doc-slow-script.html");
+  assertPausedAtSourceAndLine(dbg, source.id, 14);
+
+  info("Close toolbox and tab");
+  await dbg.toolbox.closeToolbox();
+  await removeTab(gBrowser.selectedTab);
+});
+
+add_task(async function openDebuggerFromDialog() {
+  const tab = await addTab(EXAMPLE_URL + "doc-slow-script.html");
+
+  const alert = BrowserTestUtils.waitForGlobalNotificationBar(
+    window,
+    "process-hang"
+  );
+
+  // /!\ Hack this attribute in order to force showing the "debug script" button
+  //     on all channels. Otherwise it is only displayed in dev edition.
+  tab.linkedBrowser.browsingContext.watchedByDevTools = true;
+
+  info("Execute an infinite loop");
+  // Note that spawn will return a promise that may be rejected because of the infinite loop
+  // And mochitest may consider this as an error. So ignore any rejection.
+  SpecialPowers.spawn(gBrowser.selectedBrowser, [], function() {
+    content.wrappedJSObject.infiniteLoop();
+  }).catch(e => {});
+
+  info("Wait for the slow script warning");
+  const notification = await alert;
+
+  info("Click on the debug script button");
+  const buttons = notification.buttonContainer.getElementsByTagName("button");
+  // The first button is "stop", the second is "debug script"
+  buttons[1].click();
+
+  info("Wait for the toolbox to appear and have the debugger initialized");
+  await waitFor(async () => {
+    const tb = await gDevTools.getToolboxForTab(gBrowser.selectedTab);
+    if (tb) {
+      await tb.getPanelWhenReady("jsdebugger");
+      return true;
+    }
+    return false;
+  });
+  const toolbox = await gDevTools.getToolboxForTab(gBrowser.selectedTab);
+  ok(toolbox, "Got a toolbox");
+  const dbg = createDebuggerContext(toolbox);
+
+  info("Waiting for the debugger to be paused");
+  await waitForPaused(dbg);
+  const source = findSource(dbg, "doc-slow-script.html");
+  assertPausedAtSourceAndLine(dbg, source.id, 14);
+
+  info("Close toolbox and tab");
+  await dbg.toolbox.closeToolbox();
+  await removeTab(gBrowser.selectedTab);
+});