Bug 1451307 - Part 3 - Add a test for showing the TP UI for iframes and unloaded documents. r=francois

MozReview-Commit-ID: FrcwyIbBTFL --HG-- extra : rebase_source : c21ec0fe49dd1cf3f845c249281fa5bdd6d9c622
2018-04-19 22:37:49 +02:00 · 2018-04-19 22:37:49 +02:00 · 9a539486f3
--- a/browser/base/content/test/trackingUI/browser.ini
+++ b/browser/base/content/test/trackingUI/browser.ini
@ -15,4 +15,10 @@ support-files =
  file_trackingUI_6.html
  file_trackingUI_6.js
  file_trackingUI_6.js^headers^
+[browser_trackingUI_iframe.js]
+support-files =
+  trackingPage_iframe.html
 [browser_trackingUI_telemetry.js]
+[browser_trackingUI_unloaded_documents.js]
+support-files =
+  trackingPage_onunload.html
--- a/browser/base/content/test/trackingUI/browser_trackingUI_iframe.js
+++ b/browser/base/content/test/trackingUI/browser_trackingUI_iframe.js
@ -0,0 +1,29 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+const URL = getRootDirectory(gTestPath).replace("chrome://mochitests/content", "http://example.org") + "trackingPage_iframe.html";
+
+/**
+ * Tests that sites that only contain trackers in iframes still
+ * get the tracking protection UI.
+ */
+
+add_task(async function test_iframe() {
+  await UrlClassifierTestUtils.addTestTrackers();
+
+  await SpecialPowers.pushPrefEnv({ set: [[PREF, true]] });
+
+  await BrowserTestUtils.withNewTab(URL, async function(browser) {
+    var TrackingProtection = browser.ownerGlobal.TrackingProtection;
+    ok(TrackingProtection, "got TP object");
+    ok(TrackingProtection.enabled, "TP is enabled");
+
+    is(TrackingProtection.content.getAttribute("state"), "blocked-tracking-content",
+        'content: state="blocked-tracking-content"');
+    is(TrackingProtection.icon.getAttribute("state"), "blocked-tracking-content",
+        'icon: state="blocked-tracking-content"');
+    is(TrackingProtection.icon.getAttribute("tooltiptext"),
+       gNavigatorBundle.getString("trackingProtection.icon.activeTooltip"), "correct tooltip");
+  });
+
+  UrlClassifierTestUtils.cleanupTestTrackers();
+});
--- a/browser/base/content/test/trackingUI/browser_trackingUI_unloaded_documents.js
+++ b/browser/base/content/test/trackingUI/browser_trackingUI_unloaded_documents.js
@ -0,0 +1,33 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+const URL = getRootDirectory(gTestPath).replace("chrome://mochitests/content", "http://example.org") + "trackingPage_onunload.html";
+
+/**
+ * Tests that visiting a new page while a request to a tracker is still
+ * in flight (such as when it was triggered on unload) does not cause
+ * the tracking UI to appear.
+ */
+
+add_task(async function test_onunload() {
+  await UrlClassifierTestUtils.addTestTrackers();
+
+  await SpecialPowers.pushPrefEnv({ set: [[PREF, true]] });
+
+  let tab = await BrowserTestUtils.openNewForegroundTab(gBrowser, URL);
+  let browser = tab.linkedBrowser;
+
+  let loaded = BrowserTestUtils.browserLoaded(browser, false, "http://example.com/");
+  await BrowserTestUtils.loadURI(browser, "http://example.com/");
+  await loaded;
+  var TrackingProtection = browser.ownerGlobal.TrackingProtection;
+  ok(TrackingProtection, "got TP object");
+  ok(TrackingProtection.enabled, "TP is enabled");
+
+  ok(!TrackingProtection.container.hidden, "The container is visible");
+  ok(!TrackingProtection.content.hasAttribute("state"), "content: no state");
+  ok(!TrackingProtection.icon.hasAttribute("state"), "icon: no state");
+  ok(!TrackingProtection.icon.hasAttribute("tooltiptext"), "icon: no tooltip");
+
+  BrowserTestUtils.removeTab(tab);
+  UrlClassifierTestUtils.cleanupTestTrackers();
+});
--- a/browser/base/content/test/trackingUI/trackingPage_iframe.html
+++ b/browser/base/content/test/trackingUI/trackingPage_iframe.html
@ -0,0 +1,11 @@
+<!DOCTYPE HTML>
+<!-- Any copyright is dedicated to the Public Domain.
+   - http://creativecommons.org/publicdomain/zero/1.0/ -->
+<html dir="ltr" xml:lang="en-US" lang="en-US">
+  <head>
+    <meta charset="utf8">
+  </head>
+  <body>
+    <iframe src="http://example.com/browser/browser/base/content/test/trackingUI/trackingPage.html"></iframe>
+  </body>
+</html>
--- a/browser/base/content/test/trackingUI/trackingPage_onunload.html
+++ b/browser/base/content/test/trackingUI/trackingPage_onunload.html
@ -0,0 +1,18 @@
+<!DOCTYPE HTML>
+<!-- Any copyright is dedicated to the Public Domain.
+   - http://creativecommons.org/publicdomain/zero/1.0/ -->
+<html dir="ltr" xml:lang="en-US" lang="en-US">
+  <head>
+    <meta charset="utf8">
+    <script>
+      window.onunload = function() {
+        var img = new Image();
+        img.src = "http://tracking.example.com/";
+        document.body.appendChild(img);
+      };
+    </script>
+  </head>
+  <body>
+    Testing tracking requests onunload.
+  </body>
+</html>