90 lines
2.6 KiB
JavaScript
90 lines
2.6 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/
|
|
*/
|
|
|
|
"use strict";
|
|
|
|
const { AboutWelcomeTelemetry } = ChromeUtils.importESModule(
|
|
"resource:///modules/aboutwelcome/AboutWelcomeTelemetry.sys.mjs"
|
|
);
|
|
const { AttributionCode } = ChromeUtils.importESModule(
|
|
"resource:///modules/AttributionCode.sys.mjs"
|
|
);
|
|
const { sinon } = ChromeUtils.importESModule(
|
|
"resource://testing-common/Sinon.sys.mjs"
|
|
);
|
|
const TELEMETRY_PREF = "browser.newtabpage.activity-stream.telemetry";
|
|
|
|
add_setup(function setup() {
|
|
do_get_profile();
|
|
Services.fog.initializeFOG();
|
|
});
|
|
|
|
add_task(function test_enabled() {
|
|
registerCleanupFunction(() => {
|
|
Services.prefs.clearUserPref(TELEMETRY_PREF);
|
|
});
|
|
Services.prefs.setBoolPref(TELEMETRY_PREF, true);
|
|
|
|
const AWTelemetry = new AboutWelcomeTelemetry();
|
|
|
|
equal(AWTelemetry.telemetryEnabled, true, "Telemetry should be on");
|
|
|
|
Services.prefs.setBoolPref(TELEMETRY_PREF, false);
|
|
|
|
equal(AWTelemetry.telemetryEnabled, false, "Telemetry should be off");
|
|
});
|
|
|
|
add_task(async function test_pingPayload() {
|
|
registerCleanupFunction(() => {
|
|
Services.prefs.clearUserPref(TELEMETRY_PREF);
|
|
});
|
|
Services.prefs.setBoolPref(TELEMETRY_PREF, true);
|
|
const AWTelemetry = new AboutWelcomeTelemetry();
|
|
sinon.stub(AWTelemetry, "_createPing").resolves({ event: "MOCHITEST" });
|
|
|
|
let pingSubmitted = false;
|
|
GleanPings.messagingSystem.testBeforeNextSubmit(() => {
|
|
pingSubmitted = true;
|
|
Assert.equal(Glean.messagingSystem.event.testGetValue(), "MOCHITEST");
|
|
});
|
|
await AWTelemetry.sendTelemetry();
|
|
|
|
ok(pingSubmitted, "Glean ping was submitted");
|
|
});
|
|
|
|
add_task(function test_mayAttachAttribution() {
|
|
const sandbox = sinon.createSandbox();
|
|
const AWTelemetry = new AboutWelcomeTelemetry();
|
|
|
|
sandbox.stub(AttributionCode, "getCachedAttributionData").returns(null);
|
|
|
|
let ping = AWTelemetry._maybeAttachAttribution({});
|
|
|
|
equal(ping.attribution, undefined, "Should not set attribution if it's null");
|
|
|
|
sandbox.restore();
|
|
sandbox.stub(AttributionCode, "getCachedAttributionData").returns({});
|
|
ping = AWTelemetry._maybeAttachAttribution({});
|
|
|
|
equal(
|
|
ping.attribution,
|
|
undefined,
|
|
"Should not set attribution if it's empty"
|
|
);
|
|
|
|
const attr = {
|
|
source: "google.com",
|
|
medium: "referral",
|
|
campaign: "Firefox-Brand-US-Chrome",
|
|
content: "(not set)",
|
|
experiment: "(not set)",
|
|
variation: "(not set)",
|
|
ua: "chrome",
|
|
};
|
|
sandbox.restore();
|
|
sandbox.stub(AttributionCode, "getCachedAttributionData").returns(attr);
|
|
ping = AWTelemetry._maybeAttachAttribution({});
|
|
|
|
equal(ping.attribution, attr, "Should set attribution if it presents");
|
|
});
|