forked from mirrors/gecko-dev
7e9c59f478
This adds the `-E/--use-existing-tasks` flag to the `fuzzy` or `chooser` selectors. For example: ./mach try fuzzy -E This queries the Treeherder API to find your latest try push, then any tasks that exist on that push, but aren't explicitly selected this time, will be replaced with those older tasks. You can also run: ./mach try fuzzy -E task-id=<task id> Where `<task id>` is the id of the Decision task from the push you would like to use as the base. Differential Revision: https://phabricator.services.mozilla.com/D200380
106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
# 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 os
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
import yaml
|
|
from moztest.resolve import TestResolver
|
|
from responses import RequestsMock
|
|
from taskgraph.graph import Graph
|
|
from taskgraph.task import Task
|
|
from taskgraph.taskgraph import TaskGraph
|
|
from tryselect import push
|
|
|
|
|
|
@pytest.fixture
|
|
def responses():
|
|
with RequestsMock() as rsps:
|
|
yield rsps
|
|
|
|
|
|
@pytest.fixture
|
|
def tg(request):
|
|
if not hasattr(request.module, "TASKS"):
|
|
pytest.fail(
|
|
"'tg' fixture used from a module that didn't define the TASKS variable"
|
|
)
|
|
|
|
tasks = request.module.TASKS
|
|
for task in tasks:
|
|
task.setdefault("task", {})
|
|
task["task"].setdefault("tags", {})
|
|
|
|
tasks = {t["label"]: Task(**t) for t in tasks}
|
|
return TaskGraph(tasks, Graph(tasks.keys(), set()))
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_resolver(monkeypatch):
|
|
def inner(suites, tests):
|
|
def fake_test_metadata(*args, **kwargs):
|
|
return suites, tests
|
|
|
|
monkeypatch.setattr(TestResolver, "resolve_metadata", fake_test_metadata)
|
|
|
|
return inner
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def patch_vcs(monkeypatch):
|
|
attrs = {
|
|
"path": push.vcs.path,
|
|
}
|
|
mock = MagicMock()
|
|
mock.configure_mock(**attrs)
|
|
monkeypatch.setattr(push, "vcs", mock)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def run_mach():
|
|
import mach_initialize
|
|
from tryselect.tasks import build
|
|
|
|
mach = mach_initialize.initialize(build.topsrcdir)
|
|
|
|
def inner(args):
|
|
return mach.run(args)
|
|
|
|
return inner
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
if all(
|
|
fixture in metafunc.fixturenames
|
|
for fixture in ("task_config", "args", "expected")
|
|
):
|
|
|
|
def load_tests():
|
|
for task_config, tests in metafunc.module.TASK_CONFIG_TESTS.items():
|
|
for args, expected in tests:
|
|
yield (task_config, args, expected)
|
|
|
|
tests = list(load_tests())
|
|
ids = ["{} {}".format(t[0], " ".join(t[1])).strip() for t in tests]
|
|
metafunc.parametrize("task_config,args,expected", tests, ids=ids)
|
|
|
|
elif all(
|
|
fixture in metafunc.fixturenames for fixture in ("shared_name", "shared_preset")
|
|
):
|
|
preset_path = os.path.join(
|
|
push.build.topsrcdir, "tools", "tryselect", "try_presets.yml"
|
|
)
|
|
with open(preset_path, "r") as fh:
|
|
presets = list(yaml.safe_load(fh).items())
|
|
|
|
ids = [p[0] for p in presets]
|
|
|
|
# Mark fuzzy presets on Windows xfail due to fzf not being installed.
|
|
if os.name == "nt":
|
|
for i, preset in enumerate(presets):
|
|
if preset[1]["selector"] == "fuzzy":
|
|
presets[i] = pytest.param(*preset, marks=pytest.mark.xfail)
|
|
|
|
metafunc.parametrize("shared_name,shared_preset", presets, ids=ids)
|