Bug 1600109 - Implement setup function in head.js to set up prefs and telemetry, and call it in all tests. r=dragana,JuniorHsu

Differential Revision: https://phabricator.services.mozilla.com/D58195

--HG--
extra : moz-landing-system : lando

browser/extensions/doh-rollout/test/browser/browser_cleanFlow.js

@@ -1,5 +1,7 @@
 "use strict";
 
+add_task(setup);
+
 add_task(async function testCleanFlow() {
   // Set up a passing environment and enable DoH.
   setPassingHeuristics();

browser/extensions/doh-rollout/test/browser/browser_dirtyEnable.js

@@ -1,5 +1,7 @@
 "use strict";
 
+add_task(setup);
+
 add_task(async function testDirtyEnable() {
   // Set up a failing environment, pre-set DoH to enabled, and verify that
   // when the add-on is enabled, it doesn't do anything - DoH remains turned on.

browser/extensions/doh-rollout/test/browser/browser_doorhangerUserReject.js

@@ -1,5 +1,7 @@
 "use strict";
 
+add_task(setup);
+
 add_task(async function testDoorhangerUserReject() {
   // Set up a passing environment and enable DoH.
   setPassingHeuristics();

browser/extensions/doh-rollout/test/browser/browser_policyOverride.js

@@ -1,5 +1,7 @@
 "use strict";
 
+add_task(setup);
+
 const { EnterprisePolicyTesting } = ChromeUtils.import(
   "resource://testing-common/EnterprisePolicyTesting.jsm",
   null

browser/extensions/doh-rollout/test/browser/browser_userInterference.js

@@ -1,5 +1,7 @@
 "use strict";
 
+add_task(setup);
+
 add_task(async function testUserInterference() {
   // Set up a passing environment and enable DoH.
   setPassingHeuristics();

browser/extensions/doh-rollout/test/browser/head.js

@@ -12,10 +12,6 @@ ChromeUtils.defineModuleGetter(
   "resource://gre/modules/Preferences.jsm"
 );
 
-SpecialPowers.pushPrefEnv({
-  set: [["security.notification_enable_delay", 0]],
-});
-
 const ADDON_ID = "doh-rollout@mozilla.org";
 
 const prefs = {
@@ -55,6 +51,20 @@ const fakeFailingHeuristics = JSON.stringify({
   policy: "disable_doh",
 });
 
+async function setup() {
+  SpecialPowers.pushPrefEnv({
+    set: [["security.notification_enable_delay", 0]],
+  });
+  let oldCanRecord = Services.telemetry.canRecordExtended;
+  Services.telemetry.canRecordExtended = true;
+  Services.telemetry.clearEvents();
+
+  registerCleanupFunction(() => {
+    Services.telemetry.canRecordExtended = oldCanRecord;
+    Services.telemetry.clearEvents();
+  });
+}
+
 function setPassingHeuristics() {
   Preferences.set(prefs.MOCK_HEURISTICS_PREF, fakePassingHeuristics);
 }