1 files changed, 155 insertions, 0 deletions
diff --git a/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_Embed.js b/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_Embed.js
new file mode 100644
index 0000000000..86f584763c
--- /dev/null
+++ b/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_Embed.js
@@ -0,0 +1,155 @@
+Services.scriptloader.loadSubScript(
+  "chrome://mochitests/content/browser/toolkit/components/antitracking/test/browser/storage_access_head.js",
+  this
+);
+
+async function requestStorageAccessUnderSiteAndExpectSuccess() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  var p = document.requestStorageAccessUnderSite("http://example.org");
+  try {
+    await p;
+    ok(true, "Must resolve.");
+  } catch {
+    ok(false, "Must not reject.");
+  }
+}
+
+async function requestStorageAccessUnderSiteAndExpectFailure() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  var p = document.requestStorageAccessUnderSite("http://example.org");
+  try {
+    await p;
+    ok(false, "Must not resolve.");
+  } catch {
+    ok(true, "Must reject.");
+  }
+}
+
+async function completeStorageAccessRequestFromSiteAndExpectSuccess() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  var p = document.completeStorageAccessRequestFromSite("http://example.org");
+  try {
+    await p;
+    ok(true, "Must resolve.");
+  } catch {
+    ok(false, "Must not reject.");
+  }
+}
+
+async function completeStorageAccessRequestFromSiteAndExpectFailure() {
+  SpecialPowers.wrap(document).notifyUserGestureActivation();
+  var p = document.completeStorageAccessRequestFromSite("http://example.org");
+  try {
+    await p;
+    ok(false, "Must not resolve.");
+  } catch {
+    ok(true, "Must reject.");
+  }
+}
+
+async function setIntermediatePreference() {
+  await SpecialPowers.pushPermissions([
+    {
+      type: "AllowStorageAccessRequest^http://example.org",
+      allow: 1,
+      context: "http://example.com/",
+    },
+  ]);
+}
+
+async function configurePrefs() {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["dom.storage_access.enabled", true],
+      ["dom.storage_access.forward_declared.enabled", true],
+      [
+        "network.cookie.cookieBehavior",
+        BEHAVIOR_REJECT_TRACKER_AND_PARTITION_FOREIGN,
+      ],
+      ["dom.storage_access.auto_grants", false],
+      ["dom.storage_access.max_concurrent_auto_grants", 1],
+    ],
+  });
+}
+
+add_task(async function rSAUS_sameOriginIframe() {
+  await configurePrefs();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    () => {},
+    TEST_DOMAIN_7 + TEST_PATH + "3rdParty.html",
+    requestStorageAccessUnderSiteAndExpectSuccess
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function rSAUS_sameSiteIframe() {
+  await configurePrefs();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    () => {},
+    TEST_DOMAIN_8 + TEST_PATH + "3rdParty.html",
+    requestStorageAccessUnderSiteAndExpectSuccess
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function rSAUS_crossSiteIframe() {
+  await configurePrefs();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    () => {},
+    TEST_DOMAIN + TEST_PATH + "3rdParty.html",
+    requestStorageAccessUnderSiteAndExpectFailure
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function cSAR_sameOriginIframe() {
+  await configurePrefs();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    setIntermediatePreference,
+    TEST_DOMAIN_7 + TEST_PATH + "3rdParty.html",
+    completeStorageAccessRequestFromSiteAndExpectSuccess
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function cSAR_sameSiteIframe() {
+  await configurePrefs();
+  await setIntermediatePreference();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    () => {},
+    TEST_DOMAIN_8 + TEST_PATH + "3rdParty.html",
+    completeStorageAccessRequestFromSiteAndExpectSuccess
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async function cSAR_crossSiteIframe() {
+  await configurePrefs();
+  await openPageAndRunCode(
+    TEST_TOP_PAGE_7,
+    setIntermediatePreference,
+    TEST_DOMAIN + TEST_PATH + "3rdParty.html",
+    completeStorageAccessRequestFromSiteAndExpectFailure
+  );
+  await cleanUpData();
+  await SpecialPowers.flushPrefEnv();
+});
+
+add_task(async () => {
+  Services.perms.removeAll();
+  await new Promise(resolve => {
+    Services.clearData.deleteData(Ci.nsIClearDataService.CLEAR_ALL, value =>
+      resolve()
+    );
+  });
+});