1 files changed, 33 insertions, 0 deletions
diff --git a/browser/components/urlbar/tests/browser/browser_waitForLoadStartOrTimeout.js b/browser/components/urlbar/tests/browser/browser_waitForLoadStartOrTimeout.js
new file mode 100644
index 0000000000..0fc6f0739f
--- /dev/null
+++ b/browser/components/urlbar/tests/browser/browser_waitForLoadStartOrTimeout.js
@@ -0,0 +1,33 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+/**
+ * Tests the waitForLoadStartOrTimeout test helper function in head.js.
+ */
+
+"use strict";
+
+add_task(async function load() {
+  await BrowserTestUtils.withNewTab("about:blank", async () => {
+    let url = "https://example.com/";
+    await UrlbarTestUtils.promiseAutocompleteResultPopup({
+      window,
+      value: url,
+    });
+
+    let loadPromise = waitForLoadStartOrTimeout();
+    EventUtils.synthesizeKey("KEY_Enter");
+    let uri = await loadPromise;
+    info("Page should have loaded before timeout");
+
+    Assert.equal(uri.spec, url, "example.com should have loaded");
+  });
+});
+
+add_task(async function timeout() {
+  await Assert.rejects(
+    waitForLoadStartOrTimeout(),
+    /timed out/,
+    "Should have timed out"
+  );
+});