1 files changed, 99 insertions, 0 deletions
diff --git a/toolkit/components/antitracking/test/browser/browser_storageAccess_userActivation.js b/toolkit/components/antitracking/test/browser/browser_storageAccess_userActivation.js
new file mode 100644
index 0000000000..1fdc021e28
--- /dev/null
+++ b/toolkit/components/antitracking/test/browser/browser_storageAccess_userActivation.js
@@ -0,0 +1,99 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+Services.scriptloader.loadSubScript(
+  "chrome://mochitests/content/browser/browser/modules/test/browser/head.js",
+  this
+);
+Services.scriptloader.loadSubScript(
+  "chrome://mochitests/content/browser/toolkit/components/antitracking/test/browser/storage_access_head.js",
+  this
+);
+
+// Callback function to check if the user activation remained active
+async function requestStorageAccessAndExpectUserActivationActive() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  let p = document.requestStorageAccess();
+  try {
+    await p;
+    ok(true, "gain storage access.");
+  } catch {
+    ok(false, "denied storage access.");
+  }
+
+  is(
+    SpecialPowers.wrap(document).hasBeenUserGestureActivated,
+    true,
+    `check has-been-user-activated on the top-level-document`
+  );
+  is(
+    SpecialPowers.wrap(document).hasValidTransientUserGestureActivation,
+    true,
+    `check has-valid-transient-user-activation on the top-level-document`
+  );
+
+  SpecialPowers.wrap(document).clearUserGestureActivation();
+}
+
+// Callback function to check if the user activation was consumed
+async function requestStorageAccessAndExpectUserActivationConsumed() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  const lastUserGesture = SpecialPowers.wrap(document).lastUserGestureTimeStamp;
+
+  let p = document.requestStorageAccess();
+  try {
+    await p;
+    ok(true, "gain storage access.");
+  } catch {
+    ok(false, "denied storage access.");
+  }
+
+  is(
+    SpecialPowers.wrap(document).hasBeenUserGestureActivated,
+    true,
+    `check has-been-user-activated on the top-level-document`
+  );
+  is(
+    SpecialPowers.wrap(document).lastUserGestureTimeStamp,
+    lastUserGesture,
+    `check has-valid-transient-user-activation on the top-level-document`
+  );
+
+  SpecialPowers.wrap(document).clearUserGestureActivation();
+}
+
+add_task(async function testActiveEvent() {
+  await setPreferences();
+
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_HTTPS,
+    getExpectPopupAndClick("accept"),
+    TEST_3RD_PARTY_PAGE,
+    requestStorageAccessAndExpectUserActivationActive
+  );
+
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function testConsumedEvent() {
+  await setPreferences();
+  SpecialPowers.Services.prefs.setIntPref(
+    "dom.user_activation.transient.timeout",
+    1000
+  );
+
+  const timeout =
+    SpecialPowers.getIntPref("dom.user_activation.transient.timeout") + 1000;
+  // Timeout to consume transient activation
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_HTTPS,
+    getExpectPopupAndClickAfterDelay("accept", timeout),
+    TEST_3RD_PARTY_PAGE,
+    requestStorageAccessAndExpectUserActivationConsumed
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});