diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /toolkit/modules/tests/xpcshell/test_AllowedAppSources.js | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'toolkit/modules/tests/xpcshell/test_AllowedAppSources.js')
-rw-r--r-- | toolkit/modules/tests/xpcshell/test_AllowedAppSources.js | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/toolkit/modules/tests/xpcshell/test_AllowedAppSources.js b/toolkit/modules/tests/xpcshell/test_AllowedAppSources.js new file mode 100644 index 0000000000..dcd56e2718 --- /dev/null +++ b/toolkit/modules/tests/xpcshell/test_AllowedAppSources.js @@ -0,0 +1,112 @@ +/* Any copyright is dedicated to the Public Domain. + * http://creativecommons.org/publicdomain/zero/1.0/ + */ +"use strict"; + +const { OsEnvironment } = ChromeUtils.importESModule( + "resource://gre/modules/OsEnvironment.sys.mjs" +); +const { TelemetryTestUtils } = ChromeUtils.importESModule( + "resource://testing-common/TelemetryTestUtils.sys.mjs" +); + +// Valid values that telemetry might report. Except for "Error" and +// "NoSuchFeature", these are also the values that we read from the registry. +var APP_SOURCE_ANYWHERE = "Anywhere"; +var APP_SOURCE_RECOMMENDATIONS = "Recommendations"; +var APP_SOURCE_PREFER_STORE = "PreferStore"; +var APP_SOURCE_STORE_ONLY = "StoreOnly"; +var APP_SOURCE_NO_SUCH_FEATURE = "NoSuchFeature"; +var APP_SOURCE_COLLECTION_ERROR = "Error"; + +// This variable will store the value that we will fake reading from the +// registry (i.e. the value that would normally be read from +// SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\AicEnabled"). A value +// of undefined represents the registry value not existing. +var gRegistryData; +// If this is set to true, we will throw an exception rather than providing a +// value when Policy.getAllowedAppSources is called. +var gErrorReadingRegistryValue = false; +// This holds the value that Policy.windowsVersionHasAppSourcesFeature will +// return, to allow us to simulate detection of operating system versions +// without the App Sources feature. +var gHaveAppSourcesFeature = true; +// If this is set to true, we will throw an exception rather than providing a +// value when Policy.windowsVersionHasAppSourcesFeature is called. +var gErrorDetectingAppSourcesFeature = false; + +function setup() { + // Allow the test to run successfully for all products whether they have the + // probe or not. + Services.prefs.setBoolPref( + "toolkit.telemetry.testing.overrideProductsCheck", + true + ); + + // Mock up the function that gets the allowed app sources so we don't actually + // have to change OS settings to test. + OsEnvironment.Policy.getAllowedAppSources = () => { + if (gErrorReadingRegistryValue) { + throw new Error("Arbitrary Testing Error"); + } + // To mimic the normal functionality here, we want to return undefined if + // the registry value doesn't exist. But that is already how we are + // representing that state in gRegistryData. So no conversion is necessary. + return gRegistryData; + }; + OsEnvironment.Policy.windowsVersionHasAppSourcesFeature = () => { + if (gErrorDetectingAppSourcesFeature) { + throw new Error("Arbitrary Testing Error"); + } + return gHaveAppSourcesFeature; + }; +} + +function runTest( + { + registryData = "", + registryValueExists = true, + registryReadError = false, + osHasAppSourcesFeature = true, + errorDetectingAppSourcesFeature = false, + }, + expectedScalarValue +) { + if (registryValueExists) { + gRegistryData = registryData; + } else { + gRegistryData = undefined; + } + gErrorReadingRegistryValue = registryReadError; + gHaveAppSourcesFeature = osHasAppSourcesFeature; + gErrorDetectingAppSourcesFeature = errorDetectingAppSourcesFeature; + + OsEnvironment.reportAllowedAppSources(); + + TelemetryTestUtils.assertScalar( + TelemetryTestUtils.getProcessScalars("parent"), + "os.environment.allowed_app_sources", + expectedScalarValue, + "The allowed app sources reported should match the expected sources" + ); +} + +add_task(async function testAppSources() { + setup(); + + runTest({ registryData: APP_SOURCE_ANYWHERE }, APP_SOURCE_ANYWHERE); + runTest( + { registryData: APP_SOURCE_RECOMMENDATIONS }, + APP_SOURCE_RECOMMENDATIONS + ); + runTest({ registryData: APP_SOURCE_PREFER_STORE }, APP_SOURCE_PREFER_STORE); + runTest({ registryData: APP_SOURCE_STORE_ONLY }, APP_SOURCE_STORE_ONLY); + runTest({ registryValueExists: false }, APP_SOURCE_ANYWHERE); + runTest({ registryReadError: true }, APP_SOURCE_COLLECTION_ERROR); + runTest({ registryData: "UnexpectedValue" }, APP_SOURCE_COLLECTION_ERROR); + runTest({ osHasAppSourcesFeature: false }, APP_SOURCE_NO_SUCH_FEATURE); + runTest( + { errorDetectingAppSourcesFeature: true }, + APP_SOURCE_COLLECTION_ERROR + ); +}); |