1 files changed, 151 insertions, 0 deletions
diff --git a/browser/components/urlbar/tests/engagementTelemetry/browser/browser_glean_telemetry_engagement_type.js b/browser/components/urlbar/tests/engagementTelemetry/browser/browser_glean_telemetry_engagement_type.js
new file mode 100644
index 0000000000..b2b2233b52
--- /dev/null
+++ b/browser/components/urlbar/tests/engagementTelemetry/browser/browser_glean_telemetry_engagement_type.js
@@ -0,0 +1,151 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// Test for the following data of engagement telemetry.
+// - engagement_type
+
+add_setup(async function () {
+  await setup();
+});
+
+add_task(async function engagement_type_click() {
+  await doTest(async browser => {
+    await openPopup("x");
+    await doClick();
+
+    assertEngagementTelemetry([{ engagement_type: "click" }]);
+  });
+});
+
+add_task(async function engagement_type_enter() {
+  await doTest(async browser => {
+    await openPopup("x");
+    await doEnter();
+
+    assertEngagementTelemetry([{ engagement_type: "enter" }]);
+  });
+});
+
+add_task(async function engagement_type_go_button() {
+  await doTest(async browser => {
+    await openPopup("x");
+    EventUtils.synthesizeMouseAtCenter(gURLBar.goButton, {});
+
+    assertEngagementTelemetry([{ engagement_type: "go_button" }]);
+  });
+});
+
+add_task(async function engagement_type_drop_go() {
+  await doTest(async browser => {
+    await doDropAndGo("example.com");
+
+    assertEngagementTelemetry([{ engagement_type: "drop_go" }]);
+  });
+});
+
+add_task(async function engagement_type_paste_go() {
+  await doTest(async browser => {
+    await doPasteAndGo("www.example.com");
+
+    assertEngagementTelemetry([{ engagement_type: "paste_go" }]);
+  });
+});
+
+add_task(async function engagement_type_dismiss() {
+  const cleanupQuickSuggest = await ensureQuickSuggestInit({
+    // eslint-disable-next-line mozilla/valid-lazy
+    config: lazy.QuickSuggestTestUtils.BEST_MATCH_CONFIG,
+  });
+
+  for (const isBestMatchTest of [true, false]) {
+    const prefs = isBestMatchTest
+      ? [
+          ["browser.urlbar.bestMatch.enabled", true],
+          ["browser.urlbar.bestMatch.blockingEnabled", true],
+          ["browser.urlbar.quicksuggest.blockingEnabled", false],
+        ]
+      : [
+          ["browser.urlbar.bestMatch.enabled", false],
+          ["browser.urlbar.bestMatch.blockingEnabled", false],
+          ["browser.urlbar.quicksuggest.blockingEnabled", true],
+        ];
+    await SpecialPowers.pushPrefEnv({ set: prefs });
+
+    await doTest(async browser => {
+      await openPopup("sponsored");
+
+      const originalResultCount = UrlbarTestUtils.getResultCount(window);
+      await selectRowByURL("https://example.com/sponsored");
+      if (UrlbarPrefs.get("resultMenu")) {
+        UrlbarTestUtils.openResultMenuAndPressAccesskey(window, "D");
+      } else {
+        doClickSubButton(".urlbarView-button-block");
+      }
+      await BrowserTestUtils.waitForCondition(
+        () => originalResultCount != UrlbarTestUtils.getResultCount(window)
+      );
+
+      assertEngagementTelemetry([{ engagement_type: "dismiss" }]);
+
+      // The view should stay open after dismissing the result. Now pick the
+      // heuristic result. Another "click" engagement event should be recorded.
+      Assert.ok(
+        gURLBar.view.isOpen,
+        "View should remain open after dismissing result"
+      );
+      await doClick();
+      assertEngagementTelemetry([
+        { engagement_type: "dismiss" },
+        { engagement_type: "click", interaction: "typed" },
+      ]);
+    });
+
+    await doTest(async browser => {
+      await openPopup("sponsored");
+
+      const originalResultCount = UrlbarTestUtils.getResultCount(window);
+      await selectRowByURL("https://example.com/sponsored");
+      EventUtils.synthesizeKey("KEY_Delete", { shiftKey: true });
+      await BrowserTestUtils.waitForCondition(
+        () => originalResultCount != UrlbarTestUtils.getResultCount(window)
+      );
+
+      assertEngagementTelemetry([{ engagement_type: "dismiss" }]);
+    });
+
+    await SpecialPowers.popPrefEnv();
+  }
+
+  cleanupQuickSuggest();
+});
+
+add_task(async function engagement_type_help() {
+  const cleanupQuickSuggest = await ensureQuickSuggestInit();
+
+  for (const isBestMatchTest of [true, false]) {
+    await SpecialPowers.pushPrefEnv({
+      set: [["browser.urlbar.bestMatch.enabled", isBestMatchTest]],
+    });
+
+    await doTest(async browser => {
+      await openPopup("sponsored");
+      await selectRowByURL("https://example.com/sponsored");
+      const onTabOpened = BrowserTestUtils.waitForNewTab(gBrowser);
+      if (UrlbarPrefs.get("resultMenu")) {
+        UrlbarTestUtils.openResultMenuAndPressAccesskey(window, "L");
+      } else {
+        doClickSubButton(".urlbarView-button-help");
+      }
+      const tab = await onTabOpened;
+      BrowserTestUtils.removeTab(tab);
+
+      assertEngagementTelemetry([{ engagement_type: "help" }]);
+    });
+
+    await SpecialPowers.popPrefEnv();
+  }
+
+  cleanupQuickSuggest();
+});