1 files changed, 106 insertions, 0 deletions
diff --git a/browser/components/newtab/test/browser/browser_aboutwelcome_import.js b/browser/components/newtab/test/browser/browser_aboutwelcome_import.js
new file mode 100644
index 0000000000..76716ec47f
--- /dev/null
+++ b/browser/components/newtab/test/browser/browser_aboutwelcome_import.js
@@ -0,0 +1,106 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+const IMPORT_SCREEN = {
+  id: "AW_IMPORT",
+  content: {
+    primary_button: {
+      label: "import",
+      action: {
+        navigate: true,
+        type: "SHOW_MIGRATION_WIZARD",
+      },
+    },
+  },
+};
+
+const FORCE_LEGACY =
+  Services.prefs.getCharPref(
+    "browser.migrate.content-modal.about-welcome-behavior",
+    "default"
+  ) === "legacy";
+
+add_task(async function test_wait_import_modal() {
+  await setAboutWelcomeMultiStage(
+    JSON.stringify([IMPORT_SCREEN, { id: "AW_NEXT", content: {} }])
+  );
+  const { cleanup, browser } = await openMRAboutWelcome();
+
+  // execution
+  await test_screen_content(
+    browser,
+    "renders IMPORT screen",
+    //Expected selectors
+    ["main.AW_IMPORT", "button[value='primary_button']"],
+
+    //Unexpected selectors:
+    ["main.AW_NEXT"]
+  );
+
+  const wizardPromise = BrowserTestUtils.waitForMigrationWizard(
+    window,
+    FORCE_LEGACY
+  );
+  const prefsTab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    "about:preferences"
+  );
+  await onButtonClick(browser, "button.primary");
+  const wizard = await wizardPromise;
+
+  await test_screen_content(
+    browser,
+    "still shows IMPORT screen",
+    //Expected selectors
+    ["main.AW_IMPORT", "button[value='primary_button']"],
+
+    //Unexpected selectors:
+    ["main.AW_NEXT"]
+  );
+
+  await BrowserTestUtils.closeMigrationWizard(wizard, FORCE_LEGACY);
+
+  await test_screen_content(
+    browser,
+    "moved to NEXT screen",
+    //Expected selectors
+    ["main.AW_NEXT"],
+
+    //Unexpected selectors:
+    []
+  );
+
+  // cleanup
+  await SpecialPowers.popPrefEnv(); // for setAboutWelcomeMultiStage
+  BrowserTestUtils.removeTab(prefsTab);
+  await cleanup();
+});
+
+add_task(async function test_wait_import_spotlight() {
+  const spotlightPromise = TestUtils.topicObserved("subdialog-loaded");
+  ChromeUtils.import(
+    "resource://activity-stream/lib/Spotlight.jsm"
+  ).Spotlight.showSpotlightDialog(gBrowser.selectedBrowser, {
+    content: { modal: "tab", screens: [IMPORT_SCREEN] },
+  });
+  const [win] = await spotlightPromise;
+
+  const wizardPromise = BrowserTestUtils.waitForMigrationWizard(
+    window,
+    FORCE_LEGACY
+  );
+  const prefsTab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    "about:preferences"
+  );
+  win.document
+    .querySelector(".onboardingContainer button[value='primary_button']")
+    .click();
+  const wizard = await wizardPromise;
+
+  await BrowserTestUtils.closeMigrationWizard(wizard, FORCE_LEGACY);
+
+  // cleanup
+  BrowserTestUtils.removeTab(prefsTab);
+});