Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
- This test gets skipped with pattern: os == 'win' && msix
- Manifest: security/manager/ssl/tests/unit/xpcshell.toml
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
"use strict";
do_get_profile(); // must be done before instantiating nsIDataStorageManager
let dataStorageManager = Cc[
"@mozilla.org/security/datastoragemanager;1"
].getService(Ci.nsIDataStorageManager);
let dataStorage = dataStorageManager.get(
Ci.nsIDataStorageManager.ClientAuthRememberList
);
add_task(function test_data_storage() {
// There shouldn't be anything in the data storage to begin with.
Assert.equal(dataStorage.getAll().length, 0);
// Test putting a simple key/value pair.
dataStorage.put("test", "value", Ci.nsIDataStorage.Persistent);
Assert.equal(dataStorage.get("test", Ci.nsIDataStorage.Persistent), "value");
// Test that getting a value with the same key but of a different type throws.
Assert.throws(
() => dataStorage.get("test", Ci.nsIDataStorage.Private),
/NS_ERROR_NOT_AVAILABLE/,
"getting a value of a type that hasn't been set yet should throw"
);
// Put with Private data shouldn't affect Persistent data
dataStorage.put("test", "private", Ci.nsIDataStorage.Private);
Assert.equal(dataStorage.get("test", Ci.nsIDataStorage.Private), "private");
Assert.equal(dataStorage.get("test", Ci.nsIDataStorage.Persistent), "value");
// Put of a previously-present key overwrites it (if of the same type)
dataStorage.put("test", "new", Ci.nsIDataStorage.Persistent);
Assert.equal(dataStorage.get("test", Ci.nsIDataStorage.Persistent), "new");
// Removal should work
dataStorage.remove("test", Ci.nsIDataStorage.Persistent);
Assert.throws(
() => dataStorage.get("test", Ci.nsIDataStorage.Persistent),
/NS_ERROR_NOT_AVAILABLE/,
"getting a removed value should throw"
);
// But removing one type shouldn't affect the other
Assert.equal(dataStorage.get("test", Ci.nsIDataStorage.Private), "private");
// Test removing the other type as well
dataStorage.remove("test", Ci.nsIDataStorage.Private);
Assert.throws(
() => dataStorage.get("test", Ci.nsIDataStorage.Private),
/NS_ERROR_NOT_AVAILABLE/,
"getting a removed value should throw"
);
// Saturate the storage tables (there is a maximum of 2048 entries for each
// type of data).
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(4, "0");
dataStorage.put(
`key${padded}`,
`value${padded}`,
Ci.nsIDataStorage.Persistent
);
dataStorage.put(
`key${padded}`,
`value${padded}`,
Ci.nsIDataStorage.Private
);
}
// Ensure the data can be read back.
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(4, "0");
let val = dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Persistent);
Assert.equal(val, `value${padded}`);
val = dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Private);
Assert.equal(val, `value${padded}`);
}
// Remove each entry.
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(4, "0");
dataStorage.remove(`key${padded}`, Ci.nsIDataStorage.Persistent);
dataStorage.remove(`key${padded}`, Ci.nsIDataStorage.Private);
}
// Ensure the entries are not present.
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(4, "0");
Assert.throws(
() => dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Persistent),
/NS_ERROR_NOT_AVAILABLE/,
"getting a removed value should throw"
);
Assert.throws(
() => dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Private),
/NS_ERROR_NOT_AVAILABLE/,
"getting a removed value should throw"
);
}
// Getting all entries should return an empty array.
Assert.equal(dataStorage.getAll().length, 0);
// Add new entries.
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(5, "*");
dataStorage.put(
`key${padded}`,
`value${padded}`,
Ci.nsIDataStorage.Persistent
);
dataStorage.put(
`key${padded}`,
`value${padded}`,
Ci.nsIDataStorage.Private
);
}
// Ensure each new entry was added.
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(5, "*");
let val = dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Persistent);
Assert.equal(val, `value${padded}`);
val = dataStorage.get(`key${padded}`, Ci.nsIDataStorage.Private);
Assert.equal(val, `value${padded}`);
}
// Getting all of the entries all at once should work.
let entries = dataStorage.getAll();
Assert.equal(entries.length, 4096);
let persistentEntries = entries.filter(
entry => entry.type == Ci.nsIDataStorage.Persistent
);
Assert.equal(persistentEntries.length, 2048);
let privateEntries = entries.filter(
entry => entry.type == Ci.nsIDataStorage.Private
);
Assert.equal(privateEntries.length, 2048);
let compareEntries = (a, b) => {
if (a.key < b.key) {
return -1;
}
if (a.key == b.key) {
return 0;
}
return 1;
};
persistentEntries.sort(compareEntries);
privateEntries.sort(compareEntries);
for (let i = 0; i < 2048; i++) {
let padded = i.toString().padStart(5, "*");
Assert.equal(persistentEntries[i].key, `key${padded}`);
Assert.equal(persistentEntries[i].value, `value${padded}`);
Assert.equal(privateEntries[i].key, `key${padded}`);
Assert.equal(privateEntries[i].value, `value${padded}`);
}
});