Magic-Mirror/gecko-dev
1
0
Fork 0
mirror of https://github.com/mozilla/gecko-dev.git synced 2024-12-04 02:57:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bug 1928062 - Update BTP tests to take into account MODE_DISABLED. r=anti-tracking-reviewers,bvandersloot, a=dmeehan

Browse Source 
Differential Revision: https://phabricator.services.mozilla.com/D227819
This commit is contained in:
Paul Zuehlcke 2024-11-04 15:57:02 +00:00
parent 6e0bc2d3d3
commit c8937aac92
10 changed files with 173 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_dry_run.js
View File

@ -6,6 +6,10 @@
const TEST_ORIGIN = "https://itisatracker.org";
const TEST_BASE_DOMAIN = "itisatracker.org";
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
async function runPurgeTest(expectPurge) {
ok(!SiteDataTestUtils.hasCookies(TEST_ORIGIN), "No cookies initially.");

8
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_mode_prefs.js
View File

@ -54,6 +54,10 @@ add_task(async function test_mode_pref() {
await runTestModePref(MODE_ENABLED, true);
await runTestModePref(MODE_ENABLED_STANDBY, false);
await runTestModePref(MODE_ENABLED_DRY_RUN, true);
// Run MODE_DISABLED again now that the BTP service has been initialized from
// the previous enabled tests. This tests the case where the feature is
// disabled but XPCOM is keeping the BTP singleton alive.
await runTestModePref(MODE_DISABLED, false);
});
/**
@ -61,6 +65,10 @@ add_task(async function test_mode_pref() {
* cleared.
*/
add_task(async function test_mode_switch_clears_bounce_candidates() {
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
// Start with MODE_ENABLED
let modeOriginal = Services.prefs.getIntPref(BTP_MODE_PREF);
registerCleanupFunction(() => {

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_oa_isolation.js
View File

@ -17,6 +17,10 @@ let ORIGIN_TRACKER_CONTAINER =
OA_CONTAINER
).origin;
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_purge.js
View File

@ -5,6 +5,10 @@
const BOUNCE_TRACKING_GRACE_PERIOD_SEC = 30;
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_remoteExceptions.js
View File

@ -7,6 +7,10 @@ const { RemoteSettings } = ChromeUtils.importESModule(
"resource://services-settings/remote-settings.sys.mjs"
);
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
// Name of the RemoteSettings collection containing exceptions.
const COLLECTION_NAME = "bounce-tracking-protection-exceptions";

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_simple.js
View File

@ -3,6 +3,10 @@
"use strict";
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_telemetry_purge_count.js
View File

@ -3,6 +3,10 @@
"use strict";
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
function assertCounterNull() {
is(
Glean.bounceTrackingProtection.purgeCount.success.testGetValue(),

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_userActivation.js
View File

@ -7,6 +7,10 @@
* Tests that user activation is correctly recorded for BTP.
*/
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
function assertNoUserActivationHosts() {
is(
bounceTrackingProtection.testGetUserActivationHosts({}).length,

4
toolkit/components/antitracking/bouncetrackingprotection/test/browser/browser_bouncetracking_webAuthUserActivation.js
View File

@ -3,6 +3,10 @@
"use strict";
let bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
// Import helpers from webauthn tests.
Services.scriptloader.loadSubScript(
"chrome://mochitests/content/browser/dom/webauthn/tests/browser/head.js",

207
toolkit/components/antitracking/bouncetrackingprotection/test/browser/head.js
View File

@ -7,13 +7,6 @@ const { SiteDataTestUtils } = ChromeUtils.importESModule(
"resource://testing-common/SiteDataTestUtils.sys.mjs"
);
XPCOMUtils.defineLazyServiceGetter(
this,
"bounceTrackingProtection",
"@mozilla.org/bounce-tracking-protection;1",
"nsIBounceTrackingProtection"
);
const SITE_A = "example.com";
const ORIGIN_A = `https://${SITE_A}`;
@ -308,19 +301,52 @@ async function runTestBounce(options = {}) {
} = options;
info(`runTestBounce ${JSON.stringify(options)}`);
Assert.equal(
bounceTrackingProtection.testGetBounceTrackerCandidateHosts(
originAttributes
).length,
0,
"No bounce tracker hosts initially."
);
Assert.equal(
bounceTrackingProtection.testGetUserActivationHosts(originAttributes)
.length,
0,
"No user activation hosts initially."
);
let btpIsDisabled =
Services.prefs.getIntPref("privacy.bounceTrackingProtection.mode") ==
Ci.nsIBounceTrackingProtection.MODE_DISABLED;
let bounceTrackingProtection;
try {
bounceTrackingProtection = Cc[
"@mozilla.org/bounce-tracking-protection;1"
].getService(Ci.nsIBounceTrackingProtection);
} catch (error) {
// Only in MODE_DISABLED this may throw because
// `BounceTrackingProtection::GetSingleton` will return `nullptr`.
if (!btpIsDisabled) {
throw error;
}
}
if (btpIsDisabled) {
Assert.ok(!expectCandidate, "Expect no classification in disabled mode.");
Assert.ok(
!expectRecordBounces,
"Expect no record bounces in disabled mode."
);
Assert.ok(!expectPurge, "Expect no purge in disabled mode.");
} else {
Assert.ok(
bounceTrackingProtection,
"BTP singleton must be available in any of the 'enabled' modes."
);
}
if (bounceTrackingProtection) {
Assert.equal(
bounceTrackingProtection.testGetBounceTrackerCandidateHosts(
originAttributes
).length,
0,
"No bounce tracker hosts initially."
);
Assert.equal(
bounceTrackingProtection.testGetUserActivationHosts(originAttributes)
.length,
0,
"No user activation hosts initially."
);
}
let win = window;
let { privateBrowsingId, userContextId } = originAttributes;
@ -418,53 +444,84 @@ async function runTestBounce(options = {}) {
await new Promise(resolve => setTimeout(resolve, 0));
}
Assert.deepEqual(
bounceTrackingProtection
.testGetBounceTrackerCandidateHosts(originAttributes)
.map(entry => entry.siteHost),
expectCandidate ? [SITE_TRACKER] : [],
`Should ${
expectCandidate ? "" : "not "
}have identified ${SITE_TRACKER} as a bounce tracker.`
);
if (btpIsDisabled) {
// In MODE_DISABLED `bounceTrackingProtection` may still be defined if it
// was previously accessed in an enabled state. In that case make sure
// nothing is recorded.
if (bounceTrackingProtection) {
Assert.deepEqual(
bounceTrackingProtection
.testGetBounceTrackerCandidateHosts(originAttributes)
.map(entry => entry.siteHost),
[],
"Should not have identified any bounce trackers"
);
Assert.deepEqual(
bounceTrackingProtection
.testGetUserActivationHosts(originAttributes)
.map(entry => entry.siteHost),
[],
"Should not have recorded any user activation"
);
} else {
info("BTP singleton is unavailable because mode is MODE_DISABLED.");
}
} else {
// Any of the "enabled" modes.
Assert.deepEqual(
bounceTrackingProtection
.testGetBounceTrackerCandidateHosts(originAttributes)
.map(entry => entry.siteHost),
expectCandidate ? [SITE_TRACKER] : [],
`Should ${
expectCandidate ? "" : "not "
}have identified ${SITE_TRACKER} as a bounce tracker.`
);
let expectedUserActivationHosts = [SITE_A];
if (!closeTabAfterBounce) {
// If we didn't close the tab early we should have user activation for the
// destination site.
expectedUserActivationHosts.push(SITE_B);
let expectedUserActivationHosts = [SITE_A];
if (!closeTabAfterBounce) {
// If we didn't close the tab early we should have user activation for the
// destination site.
expectedUserActivationHosts.push(SITE_B);
}
Assert.deepEqual(
bounceTrackingProtection
.testGetUserActivationHosts(originAttributes)
.map(entry => entry.siteHost)
.sort(),
expectedUserActivationHosts.sort(),
"Should only have user activation for sites where we clicked links."
);
}
Assert.deepEqual(
bounceTrackingProtection
.testGetUserActivationHosts(originAttributes)
.map(entry => entry.siteHost)
.sort(),
expectedUserActivationHosts.sort(),
"Should only have user activation for sites where we clicked links."
);
// If the caller specified a function to run after the bounce, run it now.
await postBounceCallback();
// Run tracker purging. If the feature is disabled this throws.
let mode = Services.prefs.getIntPref("privacy.bounceTrackingProtection.mode");
let expectPurgingToThrow =
mode != Ci.nsIBounceTrackingProtection.MODE_ENABLED &&
mode != Ci.nsIBounceTrackingProtection.MODE_ENABLED_DRY_RUN;
if (bounceTrackingProtection) {
// Run tracker purging. If the feature is disabled this throws.
let mode = Services.prefs.getIntPref(
"privacy.bounceTrackingProtection.mode"
);
let expectPurgingToThrow =
mode != Ci.nsIBounceTrackingProtection.MODE_ENABLED &&
mode != Ci.nsIBounceTrackingProtection.MODE_ENABLED_DRY_RUN;
if (expectPurgingToThrow) {
await Assert.rejects(
bounceTrackingProtection.testRunPurgeBounceTrackers(),
/NS_ERROR_NOT_AVAILABLE/,
"testRunPurgeBounceTrackers should reject when BTP is disabled."
);
if (expectPurgingToThrow) {
await Assert.rejects(
bounceTrackingProtection.testRunPurgeBounceTrackers(),
/NS_ERROR_NOT_AVAILABLE/,
"testRunPurgeBounceTrackers should reject when BTP is disabled."
);
} else {
Assert.deepEqual(
await bounceTrackingProtection.testRunPurgeBounceTrackers(),
expectPurge ? [SITE_TRACKER] : [],
`Should ${expectPurge ? "" : "not "}purge state for ${SITE_TRACKER}.`
);
}
} else {
Assert.deepEqual(
await bounceTrackingProtection.testRunPurgeBounceTrackers(),
expectPurge ? [SITE_TRACKER] : [],
`Should ${expectPurge ? "" : "not "}purge state for ${SITE_TRACKER}.`
);
info("BTP is disabled. Skipping purge call.");
}
// Clean up
@ -475,22 +532,24 @@ async function runTestBounce(options = {}) {
if (usePrivateWindow) {
await BrowserTestUtils.closeWindow(win);
info(
"Closing the last PBM window should trigger a purge of all PBM state."
);
Assert.ok(
!bounceTrackingProtection.testGetBounceTrackerCandidateHosts(
originAttributes
).length,
"No bounce tracker hosts after closing private window."
);
Assert.ok(
!bounceTrackingProtection.testGetUserActivationHosts(originAttributes)
.length,
"No user activation hosts after closing private window."
);
if (bounceTrackingProtection) {
info(
"Closing the last PBM window should trigger a purge of all PBM state."
);
Assert.ok(
!bounceTrackingProtection.testGetBounceTrackerCandidateHosts(
originAttributes
).length,
"No bounce tracker hosts after closing private window."
);
Assert.ok(
!bounceTrackingProtection.testGetUserActivationHosts(originAttributes)
.length,
"No user activation hosts after closing private window."
);
}
}
bounceTrackingProtection.clearAll();
bounceTrackingProtection?.clearAll();
if (!skipSiteDataCleanup) {
await SiteDataTestUtils.clear();
}