Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
from marionette_harness import parameterized
from telemetry_harness.testcase import TelemetryTestCase
class TestShutdownPingsSucced(TelemetryTestCase):
"""Test Firefox shutdown pings."""
def tearDown(self):
super(TestShutdownPingsSucced, self).tearDown()
# We need a fresh profile next run in order that the "new-profile" and
# "first-shutdown" pings are sent.
self.marionette.profile = None
@parameterized("pingsender1", pingsender_version=b"1.0")
@parameterized("pingsender2", pingsender_version=b"2.0")
def test_shutdown_pings_succeed(self, pingsender_version=b""):
"""Test that known Firefox shutdown pings are received, with the correct
X-PingSender-Version headers."""
pingsender2_enabled = {b"1.0": False, b"2.0": True}[pingsender_version]
self.marionette.set_pref(
"toolkit.telemetry.shutdownPingSender.backgroundtask.enabled",
pingsender2_enabled,
)
# Map ping type to expected X-PingSender-Version header. Not all pings
# will be sent via pingsender, so they might have an empty (binary)
# string version.
ping_types = {
"event": pingsender_version,
"first-shutdown": pingsender_version,
"main": b"",
"new-profile": pingsender_version,
}
# We don't need the browser after this, but the harness expects a
# running browser to clean up, so we `restart_browser` rather than
# `quit_browser`.
pings = self.wait_for_pings(
self.restart_browser,
lambda p: p["type"] in ping_types.keys(),
len(ping_types),
)
self.assertEqual(len(pings), len(ping_types))
self.assertEqual(set(ping_types.keys()), set(p["type"] for p in pings))
self.assertEqual(
ping_types, dict((p["type"], p["X-PingSender-Version"]) for p in pings)
)