1 files changed, 62 insertions, 0 deletions
diff --git a/dom/indexedDB/test/browser_forgetThisSite.js b/dom/indexedDB/test/browser_forgetThisSite.js
new file mode 100644
index 0000000000..fd35069d84
--- /dev/null
+++ b/dom/indexedDB/test/browser_forgetThisSite.js
@@ -0,0 +1,62 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+let { ForgetAboutSite } = ChromeUtils.importESModule(
+  "resource://gre/modules/ForgetAboutSite.sys.mjs"
+);
+
+const domains = ["mochi.test:8888", "www.example.com"];
+
+const addPath = "/browser/dom/indexedDB/test/browser_forgetThisSiteAdd.html";
+const getPath = "/browser/dom/indexedDB/test/browser_forgetThisSiteGet.html";
+
+const testPageURL1 = "http://" + domains[0] + addPath;
+const testPageURL2 = "http://" + domains[1] + addPath;
+const testPageURL3 = "http://" + domains[0] + getPath;
+const testPageURL4 = "http://" + domains[1] + getPath;
+
+add_task(async function test1() {
+  requestLongerTimeout(2);
+  // Avoids the prompt
+  setPermission(testPageURL1, "indexedDB");
+  setPermission(testPageURL2, "indexedDB");
+
+  // Set database version for domain 1
+  gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser);
+  BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, testPageURL1);
+  await waitForMessage(11, gBrowser);
+  gBrowser.removeCurrentTab();
+});
+
+add_task(async function test2() {
+  // Set database version for domain 2
+  gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser);
+  BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, testPageURL2);
+  await waitForMessage(11, gBrowser);
+  gBrowser.removeCurrentTab();
+});
+
+add_task(async function test3() {
+  // Remove database from domain 2
+  ForgetAboutSite.removeDataFromDomain(domains[1]).then(() => {
+    setPermission(testPageURL4, "indexedDB");
+  });
+});
+
+add_task(async function test4() {
+  // Get database version for domain 1
+  gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser);
+  BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, testPageURL3);
+  await waitForMessage(11, gBrowser);
+  gBrowser.removeCurrentTab();
+});
+
+add_task(async function test5() {
+  // Get database version for domain 2
+  gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser);
+  BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, testPageURL4);
+  await waitForMessage(1, gBrowser);
+  gBrowser.removeCurrentTab();
+});