1 files changed, 246 insertions, 0 deletions
diff --git a/toolkit/components/normandy/test/browser/browser_actions_AddonRollbackAction.js b/toolkit/components/normandy/test/browser/browser_actions_AddonRollbackAction.js
new file mode 100644
index 0000000000..b6db1d1a2c
--- /dev/null
+++ b/toolkit/components/normandy/test/browser/browser_actions_AddonRollbackAction.js
@@ -0,0 +1,246 @@
+"use strict";
+
+const { AddonRollbackAction } = ChromeUtils.importESModule(
+  "resource://normandy/actions/AddonRollbackAction.sys.mjs"
+);
+const { AddonRolloutAction } = ChromeUtils.importESModule(
+  "resource://normandy/actions/AddonRolloutAction.sys.mjs"
+);
+const { BaseAction } = ChromeUtils.importESModule(
+  "resource://normandy/actions/BaseAction.sys.mjs"
+);
+const { AddonRollouts } = ChromeUtils.importESModule(
+  "resource://normandy/lib/AddonRollouts.sys.mjs"
+);
+const { NormandyTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/NormandyTestUtils.sys.mjs"
+);
+
+// Test that a simple recipe unenrolls as expected
+decorate_task(
+  AddonRollouts.withTestMock(),
+  ensureAddonCleanup(),
+  withMockNormandyApi(),
+  withStub(TelemetryEnvironment, "setExperimentInactive"),
+  withSendEventSpy(),
+  async function simple_recipe_unenrollment({
+    mockNormandyApi,
+    setExperimentInactiveStub,
+    sendEventSpy,
+  }) {
+    const rolloutRecipe = {
+      id: 1,
+      arguments: {
+        slug: "test-rollout",
+        extensionApiId: 1,
+      },
+    };
+    mockNormandyApi.extensionDetails = {
+      [rolloutRecipe.arguments.extensionApiId]: extensionDetailsFactory({
+        id: rolloutRecipe.arguments.extensionApiId,
+      }),
+    };
+
+    const webExtStartupPromise =
+      AddonTestUtils.promiseWebExtensionStartup(FIXTURE_ADDON_ID);
+
+    const rolloutAction = new AddonRolloutAction();
+    await rolloutAction.processRecipe(
+      rolloutRecipe,
+      BaseAction.suitability.FILTER_MATCH
+    );
+    is(rolloutAction.lastError, null, "lastError should be null");
+
+    await webExtStartupPromise;
+
+    const rollbackRecipe = {
+      id: 2,
+      arguments: {
+        rolloutSlug: "test-rollout",
+      },
+    };
+
+    const rollbackAction = new AddonRollbackAction();
+    ok(
+      await AddonRollouts.has(rolloutRecipe.arguments.slug),
+      "Rollout should have been added"
+    );
+    await rollbackAction.processRecipe(
+      rollbackRecipe,
+      BaseAction.suitability.FILTER_MATCH
+    );
+
+    const addon = await AddonManager.getAddonByID(FIXTURE_ADDON_ID);
+    ok(!addon, "add-on is uninstalled");
+
+    const rollouts = await AddonRollouts.getAll();
+    Assert.deepEqual(
+      rollouts,
+      [
+        {
+          recipeId: rolloutRecipe.id,
+          slug: "test-rollout",
+          state: AddonRollouts.STATE_ROLLED_BACK,
+          extensionApiId: 1,
+          addonId: FIXTURE_ADDON_ID,
+          addonVersion: "1.0",
+          xpiUrl: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].url,
+          xpiHash: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].hash,
+          xpiHashAlgorithm: "sha256",
+          enrollmentId: rollouts[0].enrollmentId,
+        },
+      ],
+      "Rollback should be stored in db"
+    );
+    ok(
+      NormandyTestUtils.isUuid(rollouts[0].enrollmentId),
+      "enrollmentId should be a UUID"
+    );
+
+    sendEventSpy.assertEvents([
+      ["enroll", "addon_rollout", rollbackRecipe.arguments.rolloutSlug],
+      ["unenroll", "addon_rollback", rollbackRecipe.arguments.rolloutSlug],
+    ]);
+
+    Assert.deepEqual(
+      setExperimentInactiveStub.args,
+      [["test-rollout"]],
+      "the telemetry experiment should deactivated"
+    );
+  }
+);
+
+// Add-on already uninstalled
+decorate_task(
+  AddonRollouts.withTestMock(),
+  ensureAddonCleanup(),
+  withMockNormandyApi(),
+  withSendEventSpy(),
+  async function addon_already_uninstalled({ mockNormandyApi, sendEventSpy }) {
+    const rolloutRecipe = {
+      id: 1,
+      arguments: {
+        slug: "test-rollout",
+        extensionApiId: 1,
+      },
+    };
+    mockNormandyApi.extensionDetails = {
+      [rolloutRecipe.arguments.extensionApiId]: extensionDetailsFactory({
+        id: rolloutRecipe.arguments.extensionApiId,
+      }),
+    };
+
+    const webExtStartupPromise =
+      AddonTestUtils.promiseWebExtensionStartup(FIXTURE_ADDON_ID);
+
+    const rolloutAction = new AddonRolloutAction();
+    await rolloutAction.processRecipe(
+      rolloutRecipe,
+      BaseAction.suitability.FILTER_MATCH
+    );
+    is(rolloutAction.lastError, null, "lastError should be null");
+
+    await webExtStartupPromise;
+
+    const rollbackRecipe = {
+      id: 2,
+      arguments: {
+        rolloutSlug: "test-rollout",
+      },
+    };
+
+    let addon = await AddonManager.getAddonByID(FIXTURE_ADDON_ID);
+    await addon.uninstall();
+
+    const rollbackAction = new AddonRollbackAction();
+    await rollbackAction.processRecipe(
+      rollbackRecipe,
+      BaseAction.suitability.FILTER_MATCH
+    );
+
+    addon = await AddonManager.getAddonByID(FIXTURE_ADDON_ID);
+    ok(!addon, "add-on is uninstalled");
+
+    const rollouts = await AddonRollouts.getAll();
+    Assert.deepEqual(
+      rollouts,
+      [
+        {
+          recipeId: rolloutRecipe.id,
+          slug: "test-rollout",
+          state: AddonRollouts.STATE_ROLLED_BACK,
+          extensionApiId: 1,
+          addonId: FIXTURE_ADDON_ID,
+          addonVersion: "1.0",
+          xpiUrl: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].url,
+          xpiHash: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].hash,
+          xpiHashAlgorithm: "sha256",
+          enrollmentId: rollouts[0].enrollmentId,
+        },
+      ],
+      "Rollback should be stored in db"
+    );
+    ok(
+      NormandyTestUtils.isUuid(rollouts[0].enrollmentId),
+      "enrollment ID should be a UUID"
+    );
+
+    sendEventSpy.assertEvents([
+      ["enroll", "addon_rollout", rollbackRecipe.arguments.rolloutSlug],
+      ["unenroll", "addon_rollback", rollbackRecipe.arguments.rolloutSlug],
+    ]);
+  }
+);
+
+// Already rolled back, do nothing
+decorate_task(
+  AddonRollouts.withTestMock(),
+  ensureAddonCleanup(),
+  withMockNormandyApi(),
+  withSendEventSpy(),
+  async function already_rolled_back({ sendEventSpy }) {
+    const rollout = {
+      recipeId: 1,
+      slug: "test-rollout",
+      state: AddonRollouts.STATE_ROLLED_BACK,
+      extensionApiId: 1,
+      addonId: FIXTURE_ADDON_ID,
+      addonVersion: "1.0",
+      xpiUrl: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].url,
+      xpiHash: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].hash,
+      xpiHashAlgorithm: "sha256",
+    };
+    AddonRollouts.add(rollout);
+
+    const action = new AddonRollbackAction();
+    await action.processRecipe(
+      {
+        id: 2,
+        arguments: {
+          rolloutSlug: "test-rollout",
+        },
+      },
+      BaseAction.suitability.FILTER_MATCH
+    );
+
+    Assert.deepEqual(
+      await AddonRollouts.getAll(),
+      [
+        {
+          recipeId: 1,
+          slug: "test-rollout",
+          state: AddonRollouts.STATE_ROLLED_BACK,
+          extensionApiId: 1,
+          addonId: FIXTURE_ADDON_ID,
+          addonVersion: "1.0",
+          xpiUrl: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].url,
+          xpiHash: FIXTURE_ADDON_DETAILS["normandydriver-a-1.0"].hash,
+          xpiHashAlgorithm: "sha256",
+        },
+      ],
+      "Rollback should be stored in db"
+    );
+
+    sendEventSpy.assertEvents([]);
+  }
+);