test_storage_manager_estimate.js

Enable keyboard shortcuts

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();