1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
# 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 __future__ import absolute_import, print_function, unicode_literals
import os
import pytest
import yaml
from mock import MagicMock
from moztest.resolve import TestResolver
from taskgraph.graph import Graph
from taskgraph.task import Task
from taskgraph.taskgraph import TaskGraph
from tryselect import push
@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_bootstrap
from mach.config import ConfigSettings
from tryselect.tasks import build
mach = mach_bootstrap.bootstrap(build.topsrcdir)
def inner(args):
mach.settings = ConfigSettings()
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)
|
