1 files changed, 69 insertions, 0 deletions
diff --git a/toolkit/components/telemetry/tests/unit/test_SyncPingIntegration.js b/toolkit/components/telemetry/tests/unit/test_SyncPingIntegration.js
new file mode 100644
index 0000000000..cd1a1c682b
--- /dev/null
+++ b/toolkit/components/telemetry/tests/unit/test_SyncPingIntegration.js
@@ -0,0 +1,69 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/
+*/
+
+// Enable the collection (during test) for all products so even products
+// that don't collect the data will be able to run the test without failure.
+Services.prefs.setBoolPref(
+  "toolkit.telemetry.testing.overrideProductsCheck",
+  true
+);
+
+add_task(async function test_setup() {
+  // Addon manager needs a profile directory
+  do_get_profile();
+});
+
+add_task(async function test_register_twice_fails() {
+  TelemetryController.registerSyncPingShutdown(() => {});
+  Assert.throws(
+    () => TelemetryController.registerSyncPingShutdown(() => {}),
+    /The sync ping shutdown handler is already registered./
+  );
+  await TelemetryController.testReset();
+});
+
+add_task(async function test_reset_clears_handler() {
+  await TelemetryController.testSetup();
+  TelemetryController.registerSyncPingShutdown(() => {});
+  await TelemetryController.testReset();
+  // If this works the reset must have cleared it.
+  TelemetryController.registerSyncPingShutdown(() => {});
+  await TelemetryController.testReset();
+});
+
+add_task(async function test_shutdown_handler_submits() {
+  let handlerCalled = false;
+  await TelemetryController.testSetup();
+  TelemetryController.registerSyncPingShutdown(() => {
+    handlerCalled = true;
+    // and submit a ping.
+    let ping = {
+      why: "shutdown",
+    };
+    TelemetryController.submitExternalPing("sync", ping);
+  });
+
+  await TelemetryController.testShutdown();
+  Assert.ok(handlerCalled);
+  await TelemetryController.testReset();
+});
+
+add_task(async function test_shutdown_handler_no_submit() {
+  let handlerCalled = false;
+  await TelemetryController.testSetup();
+  TelemetryController.registerSyncPingShutdown(() => {
+    handlerCalled = true;
+    // but don't submit a ping.
+  });
+
+  await TelemetryController.testShutdown();
+  Assert.ok(handlerCalled);
+  // and check we didn't record our scalar.
+  let snapshot = Telemetry.getSnapshotForScalars("main", true).parent || {};
+  Assert.ok(
+    !("telemetry.sync_shutdown_ping_sent" in snapshot),
+    "should not have recorded we sent a ping"
+  );
+  await TelemetryController.testReset();
+});