corysanin/tubestation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bug 1965387 - Add features in NimbusTestUtils.setupTest r=nimbus-reviewers,relud

Browse Source 
Differential Revision: https://phabricator.services.mozilla.com/D248560
This commit is contained in:
Beth Rennie
2025-05-09 17:59:45 +00:00
committed by brennie@mozilla.com
parent 3e492ab9fe
commit d4d273fa1d
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
toolkit/components/nimbus/test/NimbusTestUtils.sys.mjs
View File

@@ -726,9 +726,15 @@ export const NimbusTestUtils = {
experiments, experiments,
secureExperiments, secureExperiments,
clearTelemetry = false, clearTelemetry = false,
features,
} = {}) { } = {}) {
const sandbox = lazy.sinon.createSandbox(); const sandbox = lazy.sinon.createSandbox();
let cleanupFeatures = null;
if (Array.isArray(features)) {
cleanupFeatures = NimbusTestUtils.addTestFeatures(...features);
}
const store = NimbusTestUtils.stubs.store(storePath); const store = NimbusTestUtils.stubs.store(storePath);
const manager = NimbusTestUtils.stubs.manager(store); const manager = NimbusTestUtils.stubs.manager(store);
const loader = NimbusTestUtils.stubs.rsLoader(manager); const loader = NimbusTestUtils.stubs.rsLoader(manager);
@@ -751,6 +757,10 @@ export const NimbusTestUtils = {
ExperimentAPI._resetForTests(); ExperimentAPI._resetForTests();
sandbox.restore(); sandbox.restore();
if (cleanupFeatures) {
cleanupFeatures();
}
if (clearTelemetry) { if (clearTelemetry) {
Services.fog.testResetFOG(); Services.fog.testResetFOG();
Services.telemetry.clearEvents(); Services.telemetry.clearEvents();