From fbaf0bb26397aa498eb9156f06d5a6fe34dd7dd8 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 03:14:29 +0200
Subject: Merging upstream version 125.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 ...st_bouncetracking_clearExpiredUserActivation.js | 85 ++++++++++++++++++++++
 .../test/xpcshell/xpcshell.toml                    |  2 +
 2 files changed, 87 insertions(+)
 create mode 100644 toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/test_bouncetracking_clearExpiredUserActivation.js

(limited to 'toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell')

diff --git a/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/test_bouncetracking_clearExpiredUserActivation.js b/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/test_bouncetracking_clearExpiredUserActivation.js
new file mode 100644
index 0000000000..28a1350b3e
--- /dev/null
+++ b/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/test_bouncetracking_clearExpiredUserActivation.js
@@ -0,0 +1,85 @@
+/* Any copyright is dedicated to the Public Domain.
+http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+/**
+ * Test that expired user activations are cleared by the the helper method
+ * testClearExpiredUserActivations.
+ */
+add_task(async function test() {
+  // Need a profile to data clearing calls.
+  do_get_profile();
+
+  let btp = Cc["@mozilla.org/bounce-tracking-protection;1"].getService(
+    Ci.nsIBounceTrackingProtection
+  );
+
+  // Reset global bounce tracking state.
+  btp.clearAll();
+
+  // Assert initial test state.
+  Assert.deepEqual(
+    btp.testGetBounceTrackerCandidateHosts({}),
+    [],
+    "No tracker candidates initially."
+  );
+  Assert.deepEqual(
+    btp.testGetUserActivationHosts({}),
+    [],
+    "No user activation hosts initially."
+  );
+
+  // Get the bounce tracking activation lifetime. The pref is in seconds, we
+  // need to convert it to microseconds, as the user activation timestamps are
+  // in microseconds (PRTime).
+  let bounceTrackingActivationLifetimeUSec =
+    1000 *
+    1000 *
+    Services.prefs.getIntPref(
+      "privacy.bounceTrackingProtection.bounceTrackingActivationLifetimeSec"
+    );
+
+  // Add some test data for user activation.
+  btp.testAddUserActivation({}, "not-expired1.com", Date.now() * 1000);
+  btp.testAddUserActivation(
+    {},
+    "not-expired2.com",
+    Date.now() * 1000 - bounceTrackingActivationLifetimeUSec / 2
+  );
+  btp.testAddUserActivation(
+    { privateBrowsingId: 1 },
+    "pbm-not-expired.com",
+    Date.now() * 1000
+  );
+  btp.testAddUserActivation(
+    {},
+    "expired1.com",
+    Date.now() * 1000 - bounceTrackingActivationLifetimeUSec * 2
+  );
+  btp.testAddUserActivation(
+    {},
+    "expired2.com",
+    Date.now() * 1000 - (bounceTrackingActivationLifetimeUSec + 1000 * 1000)
+  );
+  btp.testAddUserActivation({ privateBrowsingId: 1 }, "pbm-expired.com", 1);
+
+  // Clear expired user activations.
+  btp.testClearExpiredUserActivations();
+
+  // Assert that expired user activations have been cleared.
+  Assert.deepEqual(
+    btp.testGetUserActivationHosts({}).sort(),
+    ["not-expired1.com", "not-expired2.com"],
+    "Expired user activation flags have been cleared for normal browsing."
+  );
+
+  Assert.deepEqual(
+    btp.testGetUserActivationHosts({ privateBrowsingId: 1 }).sort(),
+    ["pbm-not-expired.com"],
+    "Expired user activation flags have been cleared for private browsing."
+  );
+
+  // Reset global bounce tracking state.
+  btp.clearAll();
+});
diff --git a/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/xpcshell.toml b/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/xpcshell.toml
index 16e270b85c..c3aeee502f 100644
--- a/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/xpcshell.toml
+++ b/toolkit/components/antitracking/bouncetrackingprotection/test/xpcshell/xpcshell.toml
@@ -5,4 +5,6 @@ prefs = [
   "privacy.bounceTrackingProtection.bounceTrackingPurgeTimerPeriodSec=0",
 ]
 
+["test_bouncetracking_clearExpiredUserActivation.js"]
+
 ["test_bouncetracking_purge.js"]
-- 
cgit v1.2.3