66 lines
2.2 KiB
JavaScript
66 lines
2.2 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
import { OSKeyStore } from "resource://gre/modules/OSKeyStore.sys.mjs";
|
|
import { AppConstants } from "resource://gre/modules/AppConstants.sys.mjs";
|
|
|
|
const lazy = {};
|
|
ChromeUtils.defineESModuleGetters(lazy, {
|
|
UpdateUtils: "resource://gre/modules/UpdateUtils.sys.mjs",
|
|
});
|
|
import { TestUtils } from "resource://testing-common/TestUtils.sys.mjs";
|
|
|
|
// Debug builds will be treated as "official" builds for the purposes of the automated testing for behavior of OSKeyStore.ensureLoggedIn
|
|
// We want to ensure that we catch test failures that would only otherwise show up in official builds
|
|
const isCanaryBuildForOSKeyStore = AppConstants.DEBUG;
|
|
|
|
export var OSKeyStoreTestUtils = {
|
|
TEST_ONLY_REAUTH: "toolkit.osKeyStore.unofficialBuildOnlyLogin",
|
|
|
|
setup() {
|
|
this.ORIGINAL_STORE_LABEL = OSKeyStore.STORE_LABEL;
|
|
OSKeyStore.STORE_LABEL = "test-" + Math.random().toString(36).substr(2);
|
|
},
|
|
|
|
async cleanup() {
|
|
await OSKeyStore.cleanup();
|
|
OSKeyStore.STORE_LABEL = this.ORIGINAL_STORE_LABEL;
|
|
},
|
|
|
|
/**
|
|
* Checks whether or not the test can be run by bypassing
|
|
* the OS login dialog. We do not want the user to be able to
|
|
* do so with in official builds.
|
|
* @returns {boolean} True if the test can be performed.
|
|
*/
|
|
canTestOSKeyStoreLogin() {
|
|
return (
|
|
lazy.UpdateUtils.getUpdateChannel(false) == "default" &&
|
|
!isCanaryBuildForOSKeyStore
|
|
);
|
|
},
|
|
|
|
// Wait for the observer message that simulates login success or failure.
|
|
async waitForOSKeyStoreLogin(login = false) {
|
|
const str = login ? "pass" : "cancel";
|
|
|
|
let prevValue = Services.prefs.getStringPref(this.TEST_ONLY_REAUTH, "");
|
|
Services.prefs.setStringPref(this.TEST_ONLY_REAUTH, str);
|
|
|
|
await TestUtils.topicObserved(
|
|
"oskeystore-testonly-reauth",
|
|
(subject, data) => data == str
|
|
);
|
|
|
|
Services.prefs.setStringPref(this.TEST_ONLY_REAUTH, prevValue);
|
|
},
|
|
|
|
base64ToUint8Array(base64) {
|
|
let binaryString = atob(base64);
|
|
let bytes = new Uint8Array(binaryString.length);
|
|
for (let i = 0; i < binaryString.length; i++) {
|
|
bytes[i] = binaryString.charCodeAt(i);
|
|
}
|
|
return bytes;
|
|
},
|
|
};
|