1 files changed, 145 insertions, 0 deletions
diff --git a/browser/components/firefoxview/tests/browser/browser_setup_primary_password.js b/browser/components/firefoxview/tests/browser/browser_setup_primary_password.js
new file mode 100644
index 0000000000..7f8722d808
--- /dev/null
+++ b/browser/components/firefoxview/tests/browser/browser_setup_primary_password.js
@@ -0,0 +1,145 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+const { LoginTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/LoginTestUtils.sys.mjs"
+);
+
+async function tearDown(sandbox) {
+  sandbox?.restore();
+  Services.prefs.clearUserPref("services.sync.lastTabFetch");
+}
+
+function setupMocks() {
+  const sandbox = (gSandbox = setupRecentDeviceListMocks());
+  return sandbox;
+}
+
+add_setup(async function () {
+  registerCleanupFunction(async () => {
+    // reset internal state so it doesn't affect the next tests
+    TabsSetupFlowManager.resetInternalState();
+    LoginTestUtils.primaryPassword.disable();
+    await tearDown(gSandbox);
+  });
+  await SpecialPowers.pushPrefEnv({
+    set: [["services.sync.username", "username@example.com"]],
+  });
+});
+
+add_task(async function test_primary_password_locked() {
+  LoginTestUtils.primaryPassword.enable();
+  const sandbox = setupMocks();
+
+  await withFirefoxView({}, async browser => {
+    sandbox.stub(TabsSetupFlowManager, "syncTabs").resolves(null);
+    const syncedTabsMock = sandbox.stub(SyncedTabs, "getRecentTabs");
+    syncedTabsMock.resolves(getMockTabData(syncedTabsData1));
+
+    const { document } = browser.contentWindow;
+    Services.obs.notifyObservers(null, UIState.ON_UPDATE);
+
+    info("waiting for the error setup step to be visible");
+    await waitForVisibleSetupStep(browser, {
+      expectedVisible: "#tabpickup-steps-view0",
+    });
+
+    const errorStateHeader = document.querySelector(
+      "#tabpickup-steps-view0-header"
+    );
+    await BrowserTestUtils.waitForMutationCondition(
+      errorStateHeader,
+      { childList: true },
+      () => errorStateHeader.textContent.includes("Enter your Primary Password")
+    );
+
+    ok(
+      errorStateHeader.getAttribute("data-l10n-id").includes("password-locked"),
+      "Correct error message is shown"
+    );
+
+    const errorLink = document.querySelector("#error-state-link");
+    ok(
+      errorLink && BrowserTestUtils.is_visible(errorLink),
+      "Error link is visible"
+    );
+    ok(
+      errorLink.getAttribute("data-l10n-id").includes("password-locked-link"),
+      "Correct link text is shown"
+    );
+
+    const primaryButton = document.querySelector("#error-state-button");
+    ok(
+      primaryButton && BrowserTestUtils.is_visible(primaryButton),
+      "Error primary button is visible"
+    );
+
+    const clearErrorStub = sandbox.stub(
+      TabsSetupFlowManager,
+      "tryToClearError"
+    );
+    info("Setup state:" + TabsSetupFlowManager.currentSetupState.name);
+
+    info("clicking the error panel button");
+    primaryButton.click();
+    ok(
+      clearErrorStub.called,
+      "tryToClearError was called when the try-again button was clicked"
+    );
+    TabsSetupFlowManager.tryToClearError.restore();
+
+    info("Clearing the primary password");
+    LoginTestUtils.primaryPassword.disable();
+    ok(
+      !TabsSetupFlowManager.isPrimaryPasswordLocked,
+      "primary password is unlocked"
+    );
+
+    info("notifying of the primary-password unlock");
+    const clearErrorSpy = sandbox.spy(TabsSetupFlowManager, "tryToClearError");
+    // we stubbed out sync, so pretend it ran.
+    info("notifying of sync:finish");
+    Services.obs.notifyObservers(null, "weave:service:sync:finish");
+
+    const setupContainer = document.querySelector(".sync-setup-container");
+    // wait until the setup container gets hidden before checking if the tabs container is visible
+    // as it may not exist until then
+    let setupHiddenPromise = BrowserTestUtils.waitForMutationCondition(
+      setupContainer,
+      {
+        attributeFilter: ["hidden"],
+      },
+      () => {
+        return BrowserTestUtils.is_hidden(setupContainer);
+      }
+    );
+
+    Services.obs.notifyObservers(null, "passwordmgr-crypto-login");
+    await setupHiddenPromise;
+    ok(
+      clearErrorSpy.called,
+      "tryToClearError was called when the primary-password unlock notification was received"
+    );
+    // We expect the waiting state until we get a sync update/finished
+    info("Setup state:" + TabsSetupFlowManager.currentSetupState.name);
+
+    ok(TabsSetupFlowManager.waitingForTabs, "Now waiting for tabs");
+    ok(
+      document
+        .querySelector("#tabpickup-tabs-container")
+        .classList.contains("loading"),
+      "Synced tabs container has loading class"
+    );
+
+    info("notifying of sync:finish");
+    Services.obs.notifyObservers(null, "weave:service:sync:finish");
+    await TestUtils.waitForTick();
+    ok(
+      !document
+        .querySelector("#tabpickup-tabs-container")
+        .classList.contains("loading"),
+      "Synced tabs isn't loading any more"
+    );
+  });
+  await tearDown(sandbox);
+});