1 files changed, 91 insertions, 0 deletions
diff --git a/browser/components/urlbar/tests/browser/browser_edit_invalid_url.js b/browser/components/urlbar/tests/browser/browser_edit_invalid_url.js
new file mode 100644
index 0000000000..5a710c1285
--- /dev/null
+++ b/browser/components/urlbar/tests/browser/browser_edit_invalid_url.js
@@ -0,0 +1,91 @@
+/* Any copyright is dedicated to the Public Domain.
+ * https://creativecommons.org/publicdomain/zero/1.0/ */
+
+// Checks that we trim invalid urls when they are selected, so that if the user
+// modifies the selected url, or just closes the results pane, we do a visit
+// rather than searching for the trimmed string.
+
+const url = BrowserUIUtils.trimURLProtocol + "invalid.somehost/mytest";
+
+add_setup(async function () {
+  await SpecialPowers.pushPrefEnv({
+    set: [["browser.urlbar.trimURLs", true]],
+  });
+  await PlacesTestUtils.addVisits(url);
+  registerCleanupFunction(PlacesUtils.history.clear);
+});
+
+add_task(async function test_escape() {
+  await UrlbarTestUtils.promiseAutocompleteResultPopup({
+    window,
+    value: "invalid",
+  });
+  // Look for our result.
+  let resultCount = UrlbarTestUtils.getResultCount(window);
+  Assert.greater(resultCount, 1, "There should be at least two results");
+  for (let i = 0; i < resultCount; i++) {
+    let result = await UrlbarTestUtils.getDetailsOfResultAt(window, i);
+    info(`Result at ${i} has url ${result.url}`);
+    if (result.url.startsWith(url)) {
+      break;
+    }
+    EventUtils.synthesizeKey("KEY_ArrowDown");
+  }
+  Assert.equal(
+    gURLBar.value,
+    url,
+    "The string displayed in the textbox should be the untrimmed url"
+  );
+  // Close the results pane by ESC.
+  await UrlbarTestUtils.promisePopupClose(window, () => {
+    EventUtils.synthesizeKey("KEY_Escape");
+  });
+  // Confirm the result and check the loaded page.
+  let promise = waitforLoadURL();
+  EventUtils.synthesizeKey("KEY_Enter");
+  let loadedUrl = await promise;
+  Assert.equal(loadedUrl, url, "Should try to load a url");
+});
+
+add_task(async function test_edit_url() {
+  await UrlbarTestUtils.promiseAutocompleteResultPopup({
+    window,
+    value: "invalid",
+  });
+  // Look for our result.
+  let resultCount = UrlbarTestUtils.getResultCount(window);
+  Assert.greater(resultCount, 1, "There should be at least two results");
+  for (let i = 1; i < resultCount; i++) {
+    EventUtils.synthesizeKey("KEY_ArrowDown");
+    let result = await UrlbarTestUtils.getDetailsOfResultAt(window, i);
+    info(`Result at ${i} has url ${result.url}`);
+    if (result.url.startsWith(url)) {
+      break;
+    }
+  }
+  Assert.equal(
+    gURLBar.value,
+    url,
+    "The string displayed in the textbox should be the untrimmed url"
+  );
+  // Modify the url.
+  EventUtils.synthesizeKey("2");
+  let result = await UrlbarTestUtils.getDetailsOfResultAt(window, 0);
+  Assert.equal(result.type, UrlbarUtils.RESULT_TYPE.URL, "Should visit a url");
+  Assert.equal(result.url, url + "2", "Should visit the modified url");
+
+  // Confirm the result and check the loaded page.
+  let promise = waitforLoadURL();
+  EventUtils.synthesizeKey("KEY_Enter");
+  let loadedUrl = await promise;
+  Assert.equal(loadedUrl, url + "2", "Should try to load the modified url");
+});
+
+async function waitforLoadURL() {
+  let sandbox = sinon.createSandbox();
+  let loadedUrl = await new Promise(resolve =>
+    sandbox.stub(gURLBar, "_loadURL").callsFake(resolve)
+  );
+  sandbox.restore();
+  return loadedUrl;
+}