1 files changed, 89 insertions, 0 deletions
diff --git a/dom/indexedDB/test/unit/test_storage_manager_estimate.js b/dom/indexedDB/test/unit/test_storage_manager_estimate.js
new file mode 100644
index 0000000000..286e70de96
--- /dev/null
+++ b/dom/indexedDB/test/unit/test_storage_manager_estimate.js
@@ -0,0 +1,89 @@
+var testGenerator = testSteps();
+
+function* testSteps() {
+  const name = this.window
+    ? window.location.pathname
+    : "test_storage_manager_estimate.js";
+  const objectStoreName = "storagesManager";
+  const arraySize = 1e6;
+
+  ok("estimate" in navigator.storage, "Has estimate function");
+  is(typeof navigator.storage.estimate, "function", "estimate is function");
+  ok(
+    navigator.storage.estimate() instanceof Promise,
+    "estimate() method exists and returns a Promise"
+  );
+
+  navigator.storage.estimate().then(estimation => {
+    testGenerator.next(estimation.usage);
+  });
+
+  let before = yield undefined;
+
+  let request = indexedDB.open(name, 1);
+  request.onerror = errorHandler;
+  request.onupgradeneeded = grabEventAndContinueHandler;
+  request.onsuccess = continueToNextStep;
+  let event = yield undefined;
+
+  let db = event.target.result;
+  db.onerror = errorHandler;
+
+  let objectStore = db.createObjectStore(objectStoreName, {});
+  yield undefined;
+
+  navigator.storage.estimate().then(estimation => {
+    testGenerator.next(estimation.usage);
+  });
+  let usageAfterCreate = yield undefined;
+  ok(
+    usageAfterCreate > before,
+    "estimated usage must increase after createObjectStore"
+  );
+
+  let txn = db.transaction(objectStoreName, "readwrite");
+  objectStore = txn.objectStore(objectStoreName);
+  objectStore.put(new Uint8Array(arraySize), "k");
+  txn.oncomplete = continueToNextStep;
+  txn.onabort = errorHandler;
+  txn.onerror = errorHandler;
+  event = yield undefined;
+
+  navigator.storage.estimate().then(estimation => {
+    testGenerator.next(estimation.usage);
+  });
+  let usageAfterPut = yield undefined;
+  ok(
+    usageAfterPut > usageAfterCreate,
+    "estimated usage must increase after putting large object"
+  );
+  db.close();
+
+  finishTest();
+}
+
+/* exported setup */
+async function setup(isXOrigin) {
+  // Bug 1746646: Make mochitests work with TCP enabled (cookieBehavior = 5)
+  // Acquire storage access permission here so that the iframe has
+  // first-party access to the storage estimate. Without this, it is
+  // isolated and this test will always fail
+  if (isXOrigin) {
+    await SpecialPowers.pushPrefEnv({
+      set: [
+        [
+          "privacy.partition.always_partition_third_party_non_cookie_storage",
+          false,
+        ],
+      ],
+    });
+    SpecialPowers.wrap(document).notifyUserGestureActivation();
+    await SpecialPowers.addPermission(
+      "storageAccessAPI",
+      true,
+      window.location.href
+    );
+    await SpecialPowers.wrap(document).requestStorageAccess();
+  }
+  runTest();
+}