test_mozilla_raptor.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 sys

from pathlib import Path

from unittest.mock import Mock, patch

import mozunit

import pytest

from mozharness.base.python import VirtualenvMixin

from mozharness.mozilla.testing.raptor import Raptor

@pytest.mark.parametrize("run_local", [False, True])

def test_raptor_package_pins_cover_setup(tmp_path, run_local):

    harness = Raptor.__new__(Raptor)

    harness.config = {"virtualenv_path": str(tmp_path / "venv")}

    harness.clean = False

    harness.run_local = run_local

    harness.raptor_path = str(tmp_path / "raptor")

    harness.query_exe = Mock(return_value=sys.executable)

    harness.platform_name = Mock(return_value="linux")

    harness.query_abs_dirs = Mock(return_value={"abs_test_install_dir": str(tmp_path)})

    harness.register_virtualenv_module = Mock()

    harness.setup_local_ffmpeg = Mock()

    harness.install_module = Mock()

    harness.package_versions = Mock()

    with patch.object(VirtualenvMixin, "create_virtualenv") as create:

        harness.create_virtualenv()

    assert harness.query_python_constraints() == [

        str(tmp_path / "config" / "requirements-raptor.txt")

    common = str(tmp_path / "raptor" / "requirements-common.txt")

    assert create.call_args.kwargs["requirements"] == [common]

    requirements = "source_requirements.txt" if run_local else "requirements.txt"

    harness.register_virtualenv_module.assert_called_once_with(

        requirements=[str(tmp_path / "raptor" / requirements)]

    harness.install_module.assert_not_called()

@pytest.mark.parametrize("harness", ["raptor", "talos"])

@pytest.mark.parametrize("source", [False, True])

def test_performance_requirements_use_local_mozbase(harness, source):

    testing = Path(__file__).resolve().parents[2]

    filename = "source_requirements.txt" if source else "requirements.txt"

    requirements = testing / harness / filename

    def local_paths(path):

        result = set()

        for line in path.read_text().splitlines():

            if line.startswith("-r "):

                result.update(local_paths(path.parent / line[3:]))

            else:

                requirement = line.removeprefix("--editable ")

                if requirement.startswith("../"):

                    result.add(

                        (requirements.parent / requirement.split(";")[0]).resolve()

        return result

    expected = {path.parent for path in (testing / "mozbase").glob("*/setup.py")}

    assert expected

    assert expected <= local_paths(requirements)

if __name__ == "__main__":

    mozunit.main()