1 files changed, 242 insertions, 0 deletions
diff --git a/toolkit/components/telemetry/tests/unit/test_ChildScalars.js b/toolkit/components/telemetry/tests/unit/test_ChildScalars.js
new file mode 100644
index 0000000000..c6c1660aec
--- /dev/null
+++ b/toolkit/components/telemetry/tests/unit/test_ChildScalars.js
@@ -0,0 +1,242 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/
+*/
+
+ChromeUtils.import("resource://gre/modules/Services.jsm", this);
+ChromeUtils.import("resource://gre/modules/TelemetryController.jsm", this);
+ChromeUtils.import("resource://gre/modules/TelemetrySession.jsm", this);
+ChromeUtils.import("resource://gre/modules/PromiseUtils.jsm", this);
+ChromeUtils.import("resource://testing-common/ContentTaskUtils.jsm", this);
+
+const MESSAGE_CHILD_TEST_DONE = "ChildTest:Done";
+
+const PLATFORM_VERSION = "1.9.2";
+const APP_VERSION = "1";
+const APP_ID = "xpcshell@tests.mozilla.org";
+const APP_NAME = "XPCShell";
+
+const UINT_SCALAR = "telemetry.test.unsigned_int_kind";
+const KEYED_UINT_SCALAR = "telemetry.test.keyed_unsigned_int";
+const KEYED_BOOL_SCALAR = "telemetry.test.keyed_boolean_kind";
+const CONTENT_ONLY_UINT_SCALAR = "telemetry.test.content_only_uint";
+const ALL_PROCESSES_UINT_SCALAR = "telemetry.test.all_processes_uint";
+const ALL_CHILD_PROCESSES_STRING_SCALAR =
+  "telemetry.test.all_child_processes_string";
+
+function run_child_test() {
+  // Attempt to set some scalar values from the "content" process.
+  // The next scalars are not allowed to be recorded in the content process.
+  Telemetry.scalarSet(UINT_SCALAR, 1);
+  Telemetry.keyedScalarSet(KEYED_UINT_SCALAR, "should-not-be-recorded", 1);
+
+  // The next scalars shou be recorded in only the content process.
+  Telemetry.scalarSet(CONTENT_ONLY_UINT_SCALAR, 37);
+  Telemetry.scalarSet(ALL_CHILD_PROCESSES_STRING_SCALAR, "all-child-processes");
+
+  // The next scalar will be recorded in the parent and content processes.
+  Telemetry.keyedScalarSet(KEYED_BOOL_SCALAR, "content-key", true);
+  Telemetry.keyedScalarSet(KEYED_BOOL_SCALAR, "content-key2", false);
+  Telemetry.scalarSet(ALL_PROCESSES_UINT_SCALAR, 37);
+}
+
+function setParentScalars() {
+  // The following scalars are not allowed to be recorded in the parent process.
+  Telemetry.scalarSet(CONTENT_ONLY_UINT_SCALAR, 15);
+  Telemetry.scalarSet(ALL_CHILD_PROCESSES_STRING_SCALAR, "all-child-processes");
+
+  // The next ones will be recorded only in the parent.
+  Telemetry.scalarSet(UINT_SCALAR, 15);
+
+  // This last batch will be available both in the parent and child processes.
+  Telemetry.keyedScalarSet(KEYED_BOOL_SCALAR, "parent-key", false);
+  Telemetry.scalarSet(ALL_PROCESSES_UINT_SCALAR, 37);
+}
+
+function checkParentScalars(processData) {
+  const scalars = processData.scalars;
+  const keyedScalars = processData.keyedScalars;
+
+  // Check the plain scalars, make sure we're only recording what we expect.
+  Assert.ok(
+    !(CONTENT_ONLY_UINT_SCALAR in scalars),
+    "Scalars must not be recorded in other processes unless allowed."
+  );
+  Assert.ok(
+    !(ALL_CHILD_PROCESSES_STRING_SCALAR in scalars),
+    "Scalars must not be recorded in other processes unless allowed."
+  );
+  Assert.ok(
+    UINT_SCALAR in scalars,
+    `${UINT_SCALAR} must be recorded in the parent process.`
+  );
+  Assert.equal(
+    scalars[UINT_SCALAR],
+    15,
+    `${UINT_SCALAR} must have the correct value (parent process).`
+  );
+  Assert.ok(
+    ALL_PROCESSES_UINT_SCALAR in scalars,
+    `${ALL_PROCESSES_UINT_SCALAR} must be recorded in the parent process.`
+  );
+  Assert.equal(
+    scalars[ALL_PROCESSES_UINT_SCALAR],
+    37,
+    `${ALL_PROCESSES_UINT_SCALAR} must have the correct value (parent process).`
+  );
+
+  // Now check the keyed scalars.
+  Assert.ok(
+    KEYED_BOOL_SCALAR in keyedScalars,
+    `${KEYED_BOOL_SCALAR} must be recorded in the parent process.`
+  );
+  Assert.ok(
+    "parent-key" in keyedScalars[KEYED_BOOL_SCALAR],
+    `${KEYED_BOOL_SCALAR} must be recorded in the parent process.`
+  );
+  Assert.equal(
+    Object.keys(keyedScalars[KEYED_BOOL_SCALAR]).length,
+    1,
+    `${KEYED_BOOL_SCALAR} must only contain the expected key in parent process.`
+  );
+  Assert.equal(
+    keyedScalars[KEYED_BOOL_SCALAR]["parent-key"],
+    false,
+    `${KEYED_BOOL_SCALAR} must have the correct value (parent process).`
+  );
+}
+
+function checkContentScalars(processData) {
+  const scalars = processData.scalars;
+  const keyedScalars = processData.keyedScalars;
+
+  // Check the plain scalars for the content process.
+  Assert.ok(
+    !(UINT_SCALAR in scalars),
+    "Scalars must not be recorded in other processes unless allowed."
+  );
+  Assert.ok(
+    !(KEYED_UINT_SCALAR in keyedScalars),
+    "Keyed scalars must not be recorded in other processes unless allowed."
+  );
+  Assert.ok(
+    CONTENT_ONLY_UINT_SCALAR in scalars,
+    `${CONTENT_ONLY_UINT_SCALAR} must be recorded in the content process.`
+  );
+  Assert.equal(
+    scalars[CONTENT_ONLY_UINT_SCALAR],
+    37,
+    `${CONTENT_ONLY_UINT_SCALAR} must have the correct value (content process).`
+  );
+  Assert.ok(
+    ALL_CHILD_PROCESSES_STRING_SCALAR in scalars,
+    `${ALL_CHILD_PROCESSES_STRING_SCALAR} must be recorded in the content process.`
+  );
+  Assert.equal(
+    scalars[ALL_CHILD_PROCESSES_STRING_SCALAR],
+    "all-child-processes",
+    `${ALL_CHILD_PROCESSES_STRING_SCALAR} must have the correct value (content process).`
+  );
+  Assert.ok(
+    ALL_PROCESSES_UINT_SCALAR in scalars,
+    `${ALL_PROCESSES_UINT_SCALAR} must be recorded in the content process.`
+  );
+  Assert.equal(
+    scalars[ALL_PROCESSES_UINT_SCALAR],
+    37,
+    `${ALL_PROCESSES_UINT_SCALAR} must have the correct value (content process).`
+  );
+
+  // Check the keyed scalars.
+  Assert.ok(
+    KEYED_BOOL_SCALAR in keyedScalars,
+    `${KEYED_BOOL_SCALAR} must be recorded in the content process.`
+  );
+  Assert.ok(
+    "content-key" in keyedScalars[KEYED_BOOL_SCALAR],
+    `${KEYED_BOOL_SCALAR} must be recorded in the content process.`
+  );
+  Assert.ok(
+    "content-key2" in keyedScalars[KEYED_BOOL_SCALAR],
+    `${KEYED_BOOL_SCALAR} must be recorded in the content process.`
+  );
+  Assert.equal(
+    keyedScalars[KEYED_BOOL_SCALAR]["content-key"],
+    true,
+    `${KEYED_BOOL_SCALAR} must have the correct value (content process).`
+  );
+  Assert.equal(
+    keyedScalars[KEYED_BOOL_SCALAR]["content-key2"],
+    false,
+    `${KEYED_BOOL_SCALAR} must have the correct value (content process).`
+  );
+  Assert.equal(
+    Object.keys(keyedScalars[KEYED_BOOL_SCALAR]).length,
+    2,
+    `${KEYED_BOOL_SCALAR} must contain the expected keys in content process.`
+  );
+}
+
+/**
+ * This function waits until content scalars are reported into the
+ * scalar snapshot.
+ */
+async function waitForContentScalars() {
+  await ContentTaskUtils.waitForCondition(() => {
+    const scalars = Telemetry.getSnapshotForScalars("main", false);
+    return Object.keys(scalars).includes("content");
+  });
+}
+
+add_task(async function() {
+  if (!runningInParent) {
+    TelemetryController.testSetupContent();
+    run_child_test();
+    do_send_remote_message(MESSAGE_CHILD_TEST_DONE);
+    return;
+  }
+
+  // Setup.
+  do_get_profile(true);
+  loadAddonManager(APP_ID, APP_NAME, APP_VERSION, PLATFORM_VERSION);
+  finishAddonManagerStartup();
+  fakeIntlReady();
+  await TelemetryController.testSetup();
+  if (runningInParent) {
+    setParentScalars();
+    // Make sure we don't generate unexpected pings due to pref changes.
+    await setEmptyPrefWatchlist();
+  }
+
+  // Run test in child, don't wait for it to finish: just wait for the
+  // MESSAGE_CHILD_TEST_DONE.
+  run_test_in_child("test_ChildScalars.js");
+  await do_await_remote_message(MESSAGE_CHILD_TEST_DONE);
+
+  // Once scalars are set by the content process, they don't immediately get
+  // sent to the parent process. Wait for the Telemetry IPC Timer to trigger
+  // and batch send the data back to the parent process.
+  await waitForContentScalars();
+
+  // Get an "environment-changed" ping rather than a "test-ping", as
+  // scalar measurements are only supported in subsession pings.
+  const payload = TelemetrySession.getPayload("environment-change");
+
+  // Validate the scalar data.
+  Assert.ok("processes" in payload, "Should have processes section");
+  Assert.ok(
+    "content" in payload.processes,
+    "Should have child process section"
+  );
+  Assert.ok(
+    "scalars" in payload.processes.content,
+    "Child process section should have scalars."
+  );
+  Assert.ok(
+    "keyedScalars" in payload.processes.content,
+    "Child process section should have keyed scalars."
+  );
+  checkParentScalars(payload.processes.parent);
+  checkContentScalars(payload.processes.content);
+
+  do_test_finished();
+});