1 files changed, 94 insertions, 0 deletions
diff --git a/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_StorageAccessPermission.js b/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_StorageAccessPermission.js
new file mode 100644
index 0000000000..0b4f3e7273
--- /dev/null
+++ b/toolkit/components/antitracking/test/browser/browser_storageAccess_TopLevel_StorageAccessPermission.js
@@ -0,0 +1,94 @@
+add_task(
+  async function testStorageAccessPermissionRequestStorageAccessUnderSite() {
+    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],
+      ],
+    });
+    let tab = await BrowserTestUtils.openNewForegroundTab({
+      gBrowser,
+      url: TEST_4TH_PARTY_PAGE,
+    });
+    let browser = tab.linkedBrowser;
+    await SpecialPowers.spawn(browser, [TEST_DOMAIN], async tp => {
+      await SpecialPowers.pushPermissions([
+        {
+          type: "3rdPartyStorage^http://not-tracking.example.com",
+          allow: 1,
+          context: tp,
+        },
+      ]);
+      SpecialPowers.wrap(content.document).notifyUserGestureActivation();
+      var p = content.document.requestStorageAccessUnderSite(tp);
+      try {
+        await p;
+        ok(true, "Must resolve.");
+      } catch {
+        ok(false, "Must not reject.");
+      }
+    });
+
+    await BrowserTestUtils.removeTab(tab);
+  }
+);
+
+add_task(
+  async function testStorageAccessPermissionCompleteStorageAccessRequestFromSite() {
+    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],
+      ],
+    });
+    let tab = await BrowserTestUtils.openNewForegroundTab({
+      gBrowser,
+      url: TEST_TOP_PAGE,
+    });
+    let browser = tab.linkedBrowser;
+    await SpecialPowers.spawn(browser, [TEST_4TH_PARTY_DOMAIN], async tp => {
+      await SpecialPowers.pushPermissions([
+        {
+          type: "AllowStorageAccessRequest^http://example.com",
+          allow: 1,
+          context: content.document,
+        },
+        {
+          type: "3rdPartyStorage^http://not-tracking.example.com",
+          allow: 1,
+          context: content.document,
+        },
+      ]);
+      SpecialPowers.wrap(content.document).notifyUserGestureActivation();
+      var p = content.document.completeStorageAccessRequestFromSite(tp);
+      try {
+        await p;
+        ok(true, "Must resolve.");
+      } catch {
+        ok(false, "Must not reject.");
+      }
+    });
+    await BrowserTestUtils.removeTab(tab);
+  }
+);
+
+add_task(async () => {
+  Services.perms.removeAll();
+  await new Promise(resolve => {
+    Services.clearData.deleteData(Ci.nsIClearDataService.CLEAR_ALL, value =>
+      resolve()
+    );
+  });
+});