1 files changed, 50 insertions, 0 deletions
diff --git a/browser/components/newtab/test/browser/browser_newtab_trigger.js b/browser/components/newtab/test/browser/browser_newtab_trigger.js
new file mode 100644
index 0000000000..dbc1b71e21
--- /dev/null
+++ b/browser/components/newtab/test/browser/browser_newtab_trigger.js
@@ -0,0 +1,50 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { ASRouter } = ChromeUtils.import(
+  "resource://activity-stream/lib/ASRouter.jsm"
+);
+
+let sendTriggerMessageSpy;
+let triggerMatch;
+
+add_setup(function () {
+  let sandbox = sinon.createSandbox();
+  sendTriggerMessageSpy = sandbox.spy(ASRouter, "sendTriggerMessage");
+  triggerMatch = sandbox.match({ id: "defaultBrowserCheck" });
+
+  registerCleanupFunction(() => {
+    sandbox.restore();
+  });
+});
+
+async function testPageTrigger(url, waitForLoad, expectedTrigger) {
+  sendTriggerMessageSpy.resetHistory();
+  let tab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    url,
+    waitForLoad
+  );
+
+  await BrowserTestUtils.waitForCondition(
+    () => sendTriggerMessageSpy.calledWith(expectedTrigger),
+    `After ${url} finishes loading`
+  );
+  Assert.ok(
+    sendTriggerMessageSpy.calledWith(expectedTrigger),
+    `Found the expected ${expectedTrigger.id} trigger`
+  );
+
+  BrowserTestUtils.removeTab(tab);
+  sendTriggerMessageSpy.resetHistory();
+}
+
+add_task(function test_newtab_trigger() {
+  return testPageTrigger("about:newtab", false, triggerMatch);
+});
+
+add_task(function test_abouthome_trigger() {
+  return testPageTrigger("about:home", true, triggerMatch);
+});