Bug 1968644 - Part 9: Add more tests for loads triggered by tracking scripts. r=valentin,anti-tracking-reviewers,emz

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

Author: artines1

toolkit/components/antitracking/test/browser/browser.toml

@@ -341,6 +341,7 @@ support-files = [
   "style.css^headers^",
   "test.font.woff",
   "test.font.woff^headers^",
+  "triggerLoads.sjs",
 ]
 
 ["browser_urlDecorationStripping.js"]

toolkit/components/antitracking/test/browser/browser_triggeringClassificationFlags.js

@@ -364,3 +364,48 @@ add_task(async function test_first_party_flags_in_about_blank_iframe() {
 
   BrowserTestUtils.removeTab(tab);
 });
+
+// Test that loads from an embedded tracking script have the correct flags.
+add_task(async function test_embedded_tracking_script() {
+  // Open the test page and inject the tracking iframe
+  let tab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    TEST_TOP_PAGE
+  );
+
+  for (const testCase of TEST_CASES) {
+    // Skip the font face test because we don't have a definite way to determine
+    // who triggers the load in this case.
+    if (testCase.name === "Font") {
+      continue;
+    }
+
+    let obsPromise = observeAndCheck(
+      testCase.url,
+      0,
+      Ci.nsIClassifiedChannel.CLASSIFIED_TRACKING
+    );
+
+    // Inject a tracking script into the page.
+    await SpecialPowers.spawn(
+      tab.linkedBrowser,
+      [
+        TEST_3RD_PARTY_DOMAIN_HTTP +
+          TEST_PATH +
+          "triggerLoads.sjs?type=" +
+          testCase.name +
+          "&url=" +
+          testCase.url,
+      ],
+      async src => {
+        let script = content.document.createElement("script");
+        script.src = src;
+        content.document.body.appendChild(script);
+      }
+    );
+
+    await obsPromise;
+  }
+
+  BrowserTestUtils.removeTab(tab);
+});

toolkit/components/antitracking/test/browser/triggerLoads.sjs

@@ -0,0 +1,86 @@
+function handleRequest(aRequest, aResponse) {
+  aResponse.setStatusLine(aRequest.httpVersion, 200);
+  aResponse.setHeader("Content-Type", "text/javascript", false);
+  aResponse.setHeader("Access-Control-Allow-Origin", "*");
+
+  let params = new URLSearchParams(aRequest.queryString);
+  let type = params.get("type");
+  let url = params.get("url");
+
+  switch (type) {
+    case "XHR":
+      aResponse.write(`
+        let xhr = new XMLHttpRequest();
+        xhr.open("GET", "${url}");
+        xhr.send();
+      `);
+      break;
+
+    case "Fetch":
+      aResponse.write(`
+        fetch("${url}");
+      `);
+      break;
+
+    case "Image":
+      aResponse.write(`
+        let img = document.createElement("img");
+        img.src = "${url}";
+        document.body.appendChild(img);
+      `);
+      break;
+
+    case "CSS":
+      aResponse.write(`
+        let link = document.createElement("link");
+        link.rel = "stylesheet";
+        link.href = "${url}";
+        document.head.appendChild(link);
+      `);
+      break;
+
+    case "Video":
+      aResponse.write(`
+        let video = document.createElement("video");
+        video.src = "${url}";
+        document.body.appendChild(video);
+      `);
+      break;
+
+    case "Audio":
+      aResponse.write(`
+        let audio = document.createElement("audio");
+        audio.src = "${url}";
+        document.body.appendChild(audio);
+      `);
+      break;
+
+    case "Iframe":
+      aResponse.write(`
+        let iframe = document.createElement("iframe");
+        iframe.src = "${url}";
+        document.body.appendChild(iframe);
+      `);
+      break;
+
+    case "Script":
+      aResponse.write(`
+        let script = document.createElement("script");
+        script.src = "${url}";
+        document.body.appendChild(script);
+      `);
+      break;
+
+    case "WebSocket":
+      aResponse.write(`
+        let ws = new WebSocket(
+          "ws://mochi.test:8888/browser/toolkit/components/antitracking/test/browser/file_ws_handshake_delay",
+          ["test"]
+        );
+      `);
+      break;
+
+    default:
+      throw new Error("Unsupported load type: " + aRequest.queryString);
+  }
+}