gecko-dev/browser/base/content/test/sync/browser_fxa_badge.js

/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

const { AppMenuNotifications } = ChromeUtils.import(
  "resource://gre/modules/AppMenuNotifications.jsm"
);

add_task(async function test_unconfigured_no_badge() {
  const oldUIState = UIState.get;

  UIState.get = () => ({
    status: UIState.STATUS_NOT_CONFIGURED,
  });
  Services.obs.notifyObservers(null, UIState.ON_UPDATE);
  checkFxABadge(false);

  UIState.get = oldUIState;
});

add_task(async function test_signedin_no_badge() {
  const oldUIState = UIState.get;

  UIState.get = () => ({
    status: UIState.STATUS_SIGNED_IN,
    lastSync: new Date(),
    email: "foo@bar.com",
  });
  Services.obs.notifyObservers(null, UIState.ON_UPDATE);
  checkFxABadge(false);

  UIState.get = oldUIState;
});

add_task(async function test_unverified_badge_shown() {
  const oldUIState = UIState.get;

  UIState.get = () => ({
    status: UIState.STATUS_NOT_VERIFIED,
    email: "foo@bar.com",
  });
  Services.obs.notifyObservers(null, UIState.ON_UPDATE);
  checkFxABadge(true);

  UIState.get = oldUIState;
});

add_task(async function test_loginFailed_badge_shown() {
  const oldUIState = UIState.get;

  UIState.get = () => ({
    status: UIState.STATUS_LOGIN_FAILED,
    email: "foo@bar.com",
  });
  Services.obs.notifyObservers(null, UIState.ON_UPDATE);
  checkFxABadge(true);

  UIState.get = oldUIState;
});

function checkFxABadge(shouldBeShown) {
  let isShown = false;
  for (let notification of AppMenuNotifications.notifications) {
    if (notification.id == "fxa-needs-authentication") {
      isShown = true;
      break;
    }
  }
  is(isShown, shouldBeShown, "Fxa badge shown matches expected value.");
}