1 files changed, 70 insertions, 0 deletions
diff --git a/browser/extensions/formautofill/test/browser/address/browser_address_autofill_nimbus.js b/browser/extensions/formautofill/test/browser/address/browser_address_autofill_nimbus.js
new file mode 100644
index 0000000000..0ea100cf8f
--- /dev/null
+++ b/browser/extensions/formautofill/test/browser/address/browser_address_autofill_nimbus.js
@@ -0,0 +1,70 @@
+"use strict";
+
+const { ExperimentAPI } = ChromeUtils.importESModule(
+  "resource://nimbus/ExperimentAPI.sys.mjs"
+);
+
+const { ExperimentFakes } = ChromeUtils.importESModule(
+  "resource://testing-common/NimbusTestUtils.sys.mjs"
+);
+
+const { FormAutofill } = ChromeUtils.importESModule(
+  "resource://autofill/FormAutofill.sys.mjs"
+);
+
+add_setup(async function () {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["extensions.formautofill.addresses.experiments.enabled", false],
+      ["extensions.formautofill.addresses.supportedCountries", "FR"],
+    ],
+  });
+});
+
+add_task(async function test_address_autofill_feature_enabled() {
+  await ExperimentAPI.ready();
+
+  const cleanupExperiment = await ExperimentFakes.enrollWithFeatureConfig({
+    featureId: "address-autofill-feature",
+    value: { status: true },
+  });
+
+  is(
+    NimbusFeatures["address-autofill-feature"].getVariable("status"),
+    true,
+    "Nimbus feature should be enabled"
+  );
+
+  is(
+    FormAutofill.isAutofillAddressesAvailable,
+    true,
+    "Address autofill should be available when feature is enabled in nimbus."
+  );
+
+  await cleanupExperiment();
+});
+
+add_task(async function test_address_autofill_feature_disabled() {
+  await ExperimentAPI.ready();
+
+  const cleanupExperiment = await ExperimentFakes.enrollWithFeatureConfig({
+    featureId: "address-autofill-feature",
+    value: { status: false },
+  });
+
+  NimbusFeatures["address-autofill-feature"].recordExposureEvent();
+
+  is(
+    NimbusFeatures["address-autofill-feature"].getVariable("status"),
+    false,
+    "Nimbus feature shouldn't be enabled"
+  );
+
+  is(
+    FormAutofill.isAutofillAddressesAvailable,
+    false,
+    "Address autofill shouldn't be available when feature is off in nimbus."
+  );
+
+  await cleanupExperiment();
+});