test_mobile_metrics_match_desktop.py

Enable keyboard shortcuts

# 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

# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from pathlib import Path

import mozunit

import yaml

DESKTOP_METRICS_FILE = Path(__file__).parents[2].joinpath("metrics.yaml").absolute()

MOBILE_METRICS_FILE = (

    Path(__file__)

    .parents[5]

    .joinpath("mobile", "android", "fenix", "app", "metrics.yaml")

    .absolute()

def get_relevant_metrics_from_yaml_file(filename: str):

    metrics = set()

    for group_key, group_value in yaml.safe_load(open(filename)).items():

        if type(group_value) is not dict:

            continue

        for metric_key, metric_value in group_value.items():

            if "broken-site-report" in metric_value.get("send_in_pings", []):

                metrics.add(".".join([group_key, metric_key]))

    return metrics

def test_mobile_metrics_match_desktop():

    desktop_metrics = get_relevant_metrics_from_yaml_file(DESKTOP_METRICS_FILE)

    mobile_metrics = get_relevant_metrics_from_yaml_file(MOBILE_METRICS_FILE)

    assert desktop_metrics == mobile_metrics

if __name__ == "__main__":

    mozunit.main()