1 files changed, 43 insertions, 30 deletions

diff --git a/browser/components/urlbar/tests/browser/browser_contextualsearch.js b/browser/components/urlbar/tests/browser/browser_contextualsearch.js
index 60e489a542..449d1864c2 100644
--- a/browser/components/urlbar/tests/browser/browser_contextualsearch.js
+++ b/browser/components/urlbar/tests/browser/browser_contextualsearch.js
@@ -3,22 +3,54 @@
 
 "use strict";
 
-const { UrlbarProviderContextualSearch } = ChromeUtils.importESModule(
-  "resource:///modules/UrlbarProviderContextualSearch.sys.mjs"
+const { ActionsProviderContextualSearch } = ChromeUtils.importESModule(
+  "resource:///modules/ActionsProviderContextualSearch.sys.mjs"
+);
+
+const { AddonTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/AddonTestUtils.sys.mjs"
 );
 
 add_setup(async function setup() {
   await SpecialPowers.pushPrefEnv({
-    set: [["browser.urlbar.contextualSearch.enabled", true]],
+    set: [
+      ["browser.urlbar.contextualSearch.enabled", true],
+      ["browser.urlbar.secondaryActions.featureGate", true],
+    ],
   });
-});
 
-add_task(async function test_selectContextualSearchResult_already_installed() {
-  await SearchTestUtils.installSearchExtension({
+  let ext = await SearchTestUtils.installSearchExtension({
     name: "Contextual",
     search_url: "https://example.com/browser",
   });
+  await AddonTestUtils.waitForSearchProviderStartup(ext);
+});
+
+add_task(async function test_no_engine() {
+  const ENGINE_TEST_URL = "https://example.org/";
+  let onLoaded = BrowserTestUtils.browserLoaded(
+    gBrowser.selectedBrowser,
+    false,
+    ENGINE_TEST_URL
+  );
+  BrowserTestUtils.startLoadingURIString(
+    gBrowser.selectedBrowser,
+    ENGINE_TEST_URL
+  );
+  await onLoaded;
+
+  await UrlbarTestUtils.promiseAutocompleteResultPopup({
+    window,
+    value: "test",
+  });
 
+  Assert.ok(
+    UrlbarTestUtils.getResultCount(window) > 0,
+    "At least one result is shown"
+  );
+});
+
+add_task(async function test_selectContextualSearchResult_already_installed() {
   const ENGINE_TEST_URL = "https://example.com/";
   let onLoaded = BrowserTestUtils.browserLoaded(
     gBrowser.selectedBrowser,
@@ -44,25 +76,15 @@ add_task(async function test_selectContextualSearchResult_already_installed() {
     window,
     value: query,
   });
-  const resultIndex = UrlbarTestUtils.getResultCount(window) - 1;
-  const result = await UrlbarTestUtils.getDetailsOfResultAt(
-    window,
-    resultIndex
-  );
-
-  is(
-    result.dynamicType,
-    "contextualSearch",
-    "Second last result is a contextual search result"
-  );
 
   info("Focus and select the contextual search result");
-  UrlbarTestUtils.setSelectedRowIndex(window, resultIndex);
   let onLoad = BrowserTestUtils.browserLoaded(
     gBrowser.selectedBrowser,
     false,
     expectedUrl
   );
+
+  EventUtils.synthesizeKey("KEY_Tab");
   EventUtils.synthesizeKey("KEY_Enter");
   await onLoad;
 
@@ -95,25 +117,14 @@ add_task(async function test_selectContextualSearchResult_not_installed() {
     window,
     value: query,
   });
-  const resultIndex = UrlbarTestUtils.getResultCount(window) - 1;
-  const result = await UrlbarTestUtils.getDetailsOfResultAt(
-    window,
-    resultIndex
-  );
-
-  Assert.equal(
-    result.dynamicType,
-    "contextualSearch",
-    "Second last result is a contextual search result"
-  );
 
   info("Focus and select the contextual search result");
-  UrlbarTestUtils.setSelectedRowIndex(window, resultIndex);
   let onLoad = BrowserTestUtils.browserLoaded(
     gBrowser.selectedBrowser,
     false,
     EXPECTED_URL
   );
+  EventUtils.synthesizeKey("KEY_Tab");
   EventUtils.synthesizeKey("KEY_Enter");
   await onLoad;
 
@@ -122,4 +133,6 @@ add_task(async function test_selectContextualSearchResult_not_installed() {
     EXPECTED_URL,
     "Selecting the contextual search result opens the search URL"
   );
+
+  ActionsProviderContextualSearch.resetForTesting();
 });