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
|
# 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 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_initialize
from mach.config import ConfigSettings
from tryselect.tasks import build
mach = mach_initialize.initialize(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)
|
