1 files changed, 114 insertions, 0 deletions
diff --git a/dom/indexedDB/test/unit/test_storageOption_pref.js b/dom/indexedDB/test/unit/test_storageOption_pref.js
new file mode 100644
index 0000000000..d9bd1dcf77
--- /dev/null
+++ b/dom/indexedDB/test/unit/test_storageOption_pref.js
@@ -0,0 +1,114 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+/*
+ * Tests that the dom.indexedDB.storageOption.enabled pref is able to
+ * control whether the storage option is accepted by indexedDB.open().
+ */
+
+var testGenerator = testSteps();
+
+function* testSteps() {
+  registerCleanupFunction(() => {
+    Services.prefs.clearUserPref("dom.indexedDB.storageOption.enabled");
+  });
+
+  const name = "Splendid Test";
+  const version = 1;
+  const origin = "https://example.com";
+
+  // Avoid trying to show permission prompts on a window.
+  let principal = Services.scriptSecurityManager.createContentPrincipalFromOrigin(
+    origin
+  );
+  Services.perms.addFromPrincipal(
+    principal,
+    "indexedDB",
+    Ci.nsIPermissionManager.ALLOW_ACTION
+  );
+
+  const objectStoreName = "Foo";
+  const data = { key: 1, value: "bar" };
+
+  // Turn the storage option off, content databases should be "default".
+  Services.prefs.setBoolPref("dom.indexedDB.storageOption.enabled", false);
+
+  // Open a database with content privileges.
+  let request = indexedDB.openForPrincipal(principal, name, {
+    version,
+    storage: "persistent",
+  });
+
+  request.onerror = errorHandler;
+  request.onupgradeneeded = grabEventAndContinueHandler;
+  request.onsuccess = grabEventAndContinueHandler;
+  let event = yield undefined;
+
+  is(event.type, "upgradeneeded", "Got correct event type");
+
+  let db = event.target.result;
+  db.onerror = errorHandler;
+
+  let objectStore = db.createObjectStore(objectStoreName, {});
+
+  event = yield undefined;
+
+  is(event.type, "success", "Got correct event type");
+
+  is(db.name, name, "Correct name");
+  is(db.version, version, "Correct version");
+  is(db.storage, "default", "Correct persistence type");
+
+  objectStore = db
+    .transaction([objectStoreName], "readwrite")
+    .objectStore(objectStoreName);
+
+  request = objectStore.get(data.key);
+  request.onsuccess = grabEventAndContinueHandler;
+  event = yield undefined;
+
+  is(event.target.result, undefined, "Got no data");
+
+  request = objectStore.add(data.value, data.key);
+  request.onsuccess = grabEventAndContinueHandler;
+  event = yield undefined;
+
+  is(event.target.result, data.key, "Got correct key");
+
+  // Turn the storage option on, content databases should be able to get
+  // "persistent" now. Because persistent storage is separate from default
+  // storage, we will not find the database we just created and will
+  // receive an upgradeneeded event.
+  Services.prefs.setBoolPref("dom.indexedDB.storageOption.enabled", true);
+
+  request = indexedDB.openForPrincipal(principal, name, {
+    version,
+    storage: "persistent",
+  });
+
+  request.onerror = errorHandler;
+  request.onupgradeneeded = grabEventAndContinueHandler;
+  request.onsuccess = grabEventAndContinueHandler;
+  event = yield undefined;
+
+  is(event.type, "upgradeneeded", "Got correct event type");
+
+  db = event.target.result;
+  db.onerror = errorHandler;
+
+  event = yield undefined;
+
+  is(event.type, "success", "Got correct event type");
+
+  is(db.name, name, "Correct name");
+  is(db.version, version, "Correct version");
+  is(db.storage, "persistent", "Correct persistence type");
+
+  // A new database was created (because it changed persistence type).
+  is(db.objectStoreNames.length, 0, "Got no data");
+
+  finishTest();
+  yield undefined;
+}