Bug 1573605 - More thoroughly test category item sections for various states. r=johannh

Differential Revision: https://phabricator.services.mozilla.com/D42237

--HG--
extra : moz-landing-system : lando
This commit is contained in:
Nihanth Subramanya 2019-08-20 12:57:57 +00:00
parent 14aa4dc1ba
commit 3f8d6e85d7
2 changed files with 100 additions and 90 deletions

View File

@ -4,6 +4,7 @@ const TP_PB_PREF = "privacy.trackingprotection.pbmode.enabled";
const TPC_PREF = "network.cookie.cookieBehavior";
const CM_PREF = "privacy.trackingprotection.cryptomining.enabled";
const FP_PREF = "privacy.trackingprotection.fingerprinting.enabled";
const ST_PREF = "privacy.socialtracking.block_cookies.enabled";
ChromeUtils.import(
"resource://testing-common/CustomizableUITestUtils.jsm",
@ -17,29 +18,12 @@ registerCleanupFunction(function() {
Services.prefs.clearUserPref(CAT_PREF);
Services.prefs.clearUserPref(CM_PREF);
Services.prefs.clearUserPref(FP_PREF);
Services.prefs.clearUserPref(ST_PREF);
});
add_task(async function testSubcategoryLabels() {
add_task(async function testCookieCategoryLabels() {
await BrowserTestUtils.withNewTab("http://www.example.com", async function() {
let categoryItem = document.getElementById(
"protections-popup-category-tracking-protection"
);
Services.prefs.setBoolPref(TP_PREF, true);
await TestUtils.waitForCondition(
() => categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(categoryItem.classList.contains("blocked"));
Services.prefs.setBoolPref(TP_PREF, false);
await TestUtils.waitForCondition(
() => !categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(!categoryItem.classList.contains("blocked"));
categoryItem = document.getElementById(
"protections-popup-category-cookies"
);
let categoryLabelDisabled = document.getElementById(
@ -167,41 +151,79 @@ add_task(async function testSubcategoryLabels() {
!categoryLabelDisabled.hidden &&
categoryLabelEnabled.hidden
);
categoryItem = document.getElementById(
"protections-popup-category-fingerprinters"
);
Services.prefs.setBoolPref(FP_PREF, true);
await TestUtils.waitForCondition(
() => categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(categoryItem.classList.contains("blocked"));
Services.prefs.setBoolPref(FP_PREF, false);
await TestUtils.waitForCondition(
() => !categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(!categoryItem.classList.contains("blocked"));
categoryItem = document.getElementById(
"protections-popup-category-cryptominers"
);
Services.prefs.setBoolPref(CM_PREF, true);
await TestUtils.waitForCondition(
() => categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(categoryItem.classList.contains("blocked"));
Services.prefs.setBoolPref(CM_PREF, false);
await TestUtils.waitForCondition(
() => !categoryItem.classList.contains("blocked"),
"The category item has updated correctly"
);
ok(!categoryItem.classList.contains("blocked"));
});
});
let categoryItems = [
"protections-popup-category-tracking-protection",
"protections-popup-category-socialblock",
"protections-popup-category-cookies",
"protections-popup-category-cryptominers",
"protections-popup-category-fingerprinters",
].map(id => document.getElementById(id));
let categoryEnabledPrefs = [TP_PREF, ST_PREF, TPC_PREF, CM_PREF, FP_PREF];
let detectedStateFlags = [
Ci.nsIWebProgressListener.STATE_BLOCKED_TRACKING_CONTENT,
Ci.nsIWebProgressListener.STATE_BLOCKED_SOCIALTRACKING_CONTENT,
Ci.nsIWebProgressListener.STATE_COOKIES_LOADED,
Ci.nsIWebProgressListener.STATE_BLOCKED_CRYPTOMINING_CONTENT,
Ci.nsIWebProgressListener.STATE_BLOCKED_FINGERPRINTING_CONTENT,
];
async function waitForClass(item, className, shouldBePresent = true) {
await TestUtils.waitForCondition(() => {
return item.classList.contains(className) == shouldBePresent;
}, `Target class ${className} should be ${shouldBePresent ? "present" : "not present"} on item ${item.id}`);
ok(
item.classList.contains(className) == shouldBePresent,
`item.classList.contains(${className}) is ${shouldBePresent} for ${item.id}`
);
}
add_task(async function testCategorySections() {
for (let pref of categoryEnabledPrefs) {
if (pref == TPC_PREF) {
Services.prefs.setIntPref(TPC_PREF, Ci.nsICookieService.BEHAVIOR_ACCEPT);
} else {
Services.prefs.setBoolPref(pref, false);
}
}
await BrowserTestUtils.withNewTab("http://www.example.com", async function() {
for (let item of categoryItems) {
await waitForClass(item, "notFound");
await waitForClass(item, "blocked", false);
}
// For every item, we enable the category and spoof a content blocking event,
// and check that .notFound goes away and .blocked is set. Then we disable the
// category and checks that .blocked goes away, and .notFound is still unset.
let contentBlockingState = 0;
for (let i = 0; i < categoryItems.length; i++) {
let itemToTest = categoryItems[i];
let enabledPref = categoryEnabledPrefs[i];
contentBlockingState |= detectedStateFlags[i];
if (enabledPref == TPC_PREF) {
Services.prefs.setIntPref(
TPC_PREF,
Ci.nsICookieService.BEHAVIOR_REJECT
);
} else {
Services.prefs.setBoolPref(enabledPref, true);
}
gProtectionsHandler.onContentBlockingEvent(contentBlockingState);
await waitForClass(itemToTest, "notFound", false);
await waitForClass(itemToTest, "blocked", true);
if (enabledPref == TPC_PREF) {
Services.prefs.setIntPref(TPC_PREF, Ci.nsICookieService.BEHAVIOR_ALLOW);
} else {
Services.prefs.setBoolPref(enabledPref, false);
}
await waitForClass(itemToTest, "notFound", false);
await waitForClass(itemToTest, "blocked", false);
}
});
});

View File

@ -53,21 +53,9 @@ registerCleanupFunction(function() {
Services.prefs.clearUserPref(DTSCBN_PREF);
});
// This is a special version of "hidden" that doesn't check for item
// visibility and just asserts the display and opacity attributes.
// That way we can test elements even when their panel is hidden...
function hidden(sel) {
let win = tabbrowser.ownerGlobal;
let el = win.document.querySelector(sel);
let display = win.getComputedStyle(el).getPropertyValue("display", null);
let opacity = win.getComputedStyle(el).getPropertyValue("opacity", null);
return display === "none" || opacity === "0";
}
function clickButton(sel) {
let win = tabbrowser.ownerGlobal;
let el = win.document.querySelector(sel);
el.doCommand();
function notFound(id) {
let doc = tabbrowser.ownerGlobal.document;
return doc.getElementById(id).classList.contains("notFound");
}
function testBenignPage() {
@ -99,12 +87,12 @@ function testBenignPage() {
"icon box is visible"
);
ok(
hidden("#protections-popup-category-cookies"),
"Not showing cookie restrictions category"
notFound("protections-popup-category-cookies"),
"Cookie restrictions category is not found"
);
ok(
hidden("#protections-popup-category-tracking-protection"),
"Not showing trackers category"
notFound("protections-popup-category-tracking-protection"),
"Trackers category is not found"
);
}
@ -139,12 +127,12 @@ function testBenignPageWithException() {
);
ok(
hidden("#protections-popup-category-cookies"),
"Not showing cookie restrictions category"
notFound("protections-popup-category-cookies"),
"Cookie restrictions category is not found"
);
ok(
hidden("#protections-popup-category-tracking-protection"),
"Not showing trackers category"
notFound("protections-popup-category-tracking-protection"),
"Trackers category is not found"
);
}
@ -191,18 +179,18 @@ function testTrackingPage(window) {
);
ok(
!hidden("#protections-popup-category-tracking-protection"),
"Showing trackers category"
!notFound("protections-popup-category-tracking-protection"),
"Trackers category is detected"
);
if (gTrackingPageURL == COOKIE_PAGE) {
ok(
!hidden("#protections-popup-category-cookies"),
"Showing cookie restrictions category"
!notFound("protections-popup-category-cookies"),
"Cookie restrictions category is detected"
);
} else {
ok(
hidden("#protections-popup-category-cookies"),
"Not showing cookie restrictions category"
notFound("protections-popup-category-cookies"),
"Cookie restrictions category is not found"
);
}
}
@ -238,18 +226,18 @@ function testTrackingPageUnblocked(blockedByTP, window) {
);
ok(
!hidden("#protections-popup-category-tracking-protection"),
"Showing trackers category"
!notFound("protections-popup-category-tracking-protection"),
"Trackers category is detected"
);
if (gTrackingPageURL == COOKIE_PAGE) {
ok(
!hidden("#protections-popup-category-cookies"),
"Showing cookie restrictions category"
!notFound("protections-popup-category-cookies"),
"Cookie restrictions category is detected"
);
} else {
ok(
hidden("#protections-popup-category-cookies"),
"Not showing cookie restrictions category"
notFound("protections-popup-category-cookies"),
"Cookie restrictions category is not found"
);
}
}