test_raptor_profiling.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/.

import json

import os

import sys

from pathlib import Path

from subprocess import CompletedProcess

from unittest import mock

import mozunit

import pytest

here = os.path.abspath(os.path.dirname(__file__))

raptor_dir = os.path.join(os.path.dirname(here), "raptor")

sys.path.insert(0, raptor_dir)

from raptor_profiling import RaptorProfiling

@pytest.fixture

def toolchain_dir(tmp_path, monkeypatch):

    toolchain_dir = tmp_path / "toolchains"

    for tool in [

        Path("node", "node.exe"),

        Path("node", "bin", "node"),

        Path("profiler-node-tools", "profiler-edit.js"),

]:

        (toolchain_dir / tool).parent.mkdir(parents=True, exist_ok=True)

        (toolchain_dir / tool).touch()

    monkeypatch.setenv("MOZBUILD_STATE_PATH", str(toolchain_dir))

    return toolchain_dir

def test_source_path_mappings_are_passed_to_profiler_edit(tmp_path, toolchain_dir):

    source_path_mappings = [

        {"urlPrefix": "http://127.0.0.1:62763/", "localPath": str(tmp_path)}

    upload_dir = tmp_path / "upload"

    upload_dir.mkdir()

    profiler = RaptorProfiling(

        upload_dir,

        {"run_local": True},

        {"name": "speedometer3"},

        source_path_mappings,

    profiler.profile = upload_dir / "profile_unprocessed.jslb.gz"

    profiler.profile.touch()

    profiler_edit_runs = []

    def run_profiler_edit(cmd, **kwargs):

        mapping_path = Path(cmd[cmd.index("--map-source-paths") + 1])

        profiler_edit_runs.append(json.loads(mapping_path.read_text()))

        Path(cmd[cmd.index("-o") + 1]).touch()

        return CompletedProcess(cmd, 0, stdout="", stderr="")

    with mock.patch("raptor_profiling.subprocess.run", side_effect=run_profiler_edit):

        profiles = profiler.post_process_profiles()

    profiler.clean()

    assert profiles

    assert profiler_edit_runs == [{"mappings": source_path_mappings}] * len(profiles)

if __name__ == "__main__":

    mozunit.main()