1 files changed, 140 insertions, 0 deletions
diff --git a/dom/base/test/useractivation/test_useractivation_transient_consuming.html b/dom/base/test/useractivation/test_useractivation_transient_consuming.html
new file mode 100644
index 0000000000..3d40d9a7a2
--- /dev/null
+++ b/dom/base/test/useractivation/test_useractivation_transient_consuming.html
@@ -0,0 +1,140 @@
+<!DOCTYPE HTML>
+<html>
+<head>
+  <title>User activation test: consume transient flag</title>
+  <script src="/tests/SimpleTest/SimpleTest.js"></script>
+  <script src="/tests/SimpleTest/EventUtils.js"></script>
+  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css" />
+</head>
+<body>
+<iframe></iframe>
+<iframe></iframe>
+<script>
+
+SimpleTest.requestFlakyTimeout("Timeouts are needed to test transient user_activation");
+
+let timeout = SpecialPowers.getIntPref("dom.user_activation.transient.timeout") + 1000;
+
+function waitForEvent(aTarget, aEvent, aCallback) {
+  return new Promise((aResolve) => {
+    aTarget.addEventListener(aEvent, function listener(event) {
+      aCallback(event);
+      aResolve();
+    }, { once: true });
+  });
+}
+
+add_task(async function checkInitialStatus() {
+  ok(!SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on top-level document");
+  ok(!SpecialPowers.wrap(frames[0].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on first iframe");
+  ok(!SpecialPowers.wrap(frames[1].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on second iframe");
+});
+
+add_task(async function consumeTransientUserActivation() {
+  // Trigger user activation on the first iframe.
+  SpecialPowers.wrap(frames[0].document).notifyUserGestureActivation();
+
+  // Try to consume transient user activation.
+  ok(!SpecialPowers.wrap(frames[1].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on second iframe");
+  ok(SpecialPowers.wrap(frames[0].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on first iframe");
+  // Consuming a transient-user-activation should affect all tree.
+  ok(!SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on top-level document");
+
+  // Check has-valid-transient-user-activation
+  ok(!SpecialPowers.wrap(document).hasValidTransientUserGestureActivation,
+     "check has-valid-transient-user-activation on the top-level document");
+  ok(!SpecialPowers.wrap(frames[0].document).hasValidTransientUserGestureActivation,
+     "check has-valid-transient-user-activation on the first iframe");
+  ok(!SpecialPowers.wrap(frames[1].document).hasValidTransientUserGestureActivation,
+     "check has-valid-transient-user-activation on the second iframe");
+
+  // Should not affect has-been-user-activated
+  ok(SpecialPowers.wrap(document).hasBeenUserGestureActivated,
+     "check has-been-user-activated on the top-level document");
+  ok(SpecialPowers.wrap(frames[0].document).hasBeenUserGestureActivated,
+     "check has-been-user-activated on the first iframe");
+  ok(!SpecialPowers.wrap(frames[1].document).hasBeenUserGestureActivated,
+     "check has-been-user-activated on the second iframe");
+});
+
+add_task(async function consumeTransientUserActivationTimeout() {
+  // Trigger user activation on the first iframe.
+  SpecialPowers.wrap(frames[0].document).notifyUserGestureActivation();
+
+  // Should not able to consume successfully after timeout.
+  await new Promise((aResolve) => {
+    setTimeout(() => {
+      ok(!SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+         "consume transient-user-activation on top-level document");
+      ok(!SpecialPowers.wrap(frames[0].document).consumeTransientUserGestureActivation(),
+         "consume transient-user-activation on first iframe");
+      ok(!SpecialPowers.wrap(frames[1].document).consumeTransientUserGestureActivation(),
+         "consume transient-user-activation on second iframe");
+      aResolve();
+    }, timeout);
+  });
+
+  // Trigger user activation again.
+  SpecialPowers.wrap(frames[0].document).notifyUserGestureActivation();
+
+  // Try to consume transient user activation.
+  ok(!SpecialPowers.wrap(frames[1].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on second iframe");
+  ok(SpecialPowers.wrap(frames[0].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on first iframe");
+  // Consuming a transient-user-activation should affect all tree.
+  ok(!SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on top-level document");
+});
+
+add_task(async function iframeNavigation() {
+  // Trigger user activation on the first iframe.
+  SpecialPowers.wrap(frames[0].document).notifyUserGestureActivation();
+
+  // Navigate away from current page.
+  frames[0].frameElement.src = "file_empty.html";
+  await waitForEvent(frames[0].frameElement, "load", () => {});
+
+  // Try to consume transient user activation.
+  ok(!SpecialPowers.wrap(frames[1].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on second iframe");
+  ok(!SpecialPowers.wrap(frames[0].document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on first iframe");
+  ok(SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+     "consume transient-user-activation on top-level document");
+});
+
+add_task(async function triggerUserActivationOnCrossOriginFrame() {
+  // Reset the activation flag.
+  SpecialPowers.wrap(document).clearUserGestureActivation();
+
+  // load cross-origin test page on iframe.
+  frames[0].frameElement.src = "https://example.com/tests/dom/base/test/useractivation/file_iframe_consume_user_activation.html";
+  await waitForEvent(window, "message", (event) => {
+    if (event.data === "done") {
+      ok(!SpecialPowers.wrap(document).consumeTransientUserGestureActivation(),
+         "consume transient-user-activation on top-level document");
+      ok(!SpecialPowers.wrap(document).hasValidTransientUserGestureActivation,
+         "check has-valid-transient-user-activation on the top-level document");
+      ok(!SpecialPowers.wrap(frames[1].document).hasValidTransientUserGestureActivation,
+         "check has-valid-transient-user-activation on the second iframe");
+    } else {
+      ok(false, "receive unexpected message: " + event.data);
+    }
+  });
+});
+
+add_task(async function endTests() {
+  // Reset the activation flag in order not to interfere following test in the
+  // verify mode which would run the test using same document couple times.
+  SpecialPowers.wrap(document).clearUserGestureActivation();
+});
+
+</script>
+</body>