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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
from __future__ import annotations
import pytest
@pytest.fixture(params=[None, 'inprocess', 'subprocess', 'both'])
def launch_mode_conf(request: pytest.FixtureRequest) -> str | None:
"""Configured launch mode (None|'inprocess'|'subprocess'|'both')."""
assert request.param is None or isinstance(request.param, str)
return request.param
@pytest.fixture
def launch_modes(launch_mode_conf: str | None) -> set[str]:
"""Set of launch modes in which the tests will actually be run.
The value of this fixture depends on the value of `launch_mode_conf`:
- 'inprocess' -> {'inprocess'}
- 'subprocess' -> {'subprocess'}
- 'both' -> {'inprocess', 'subprocess'}
- None -> {'inprocess'}
"""
if launch_mode_conf == 'both':
return {'inprocess', 'subprocess'}
if launch_mode_conf is not None:
return {launch_mode_conf}
return {'inprocess'}
class RunTest:
def __init__(self, testdir: pytest.Testdir) -> None:
self.testdir = testdir
def __call__(
self,
script: str,
passed: int = 1,
skipped: int = 0,
failed: int = 0,
launch_mode_conf: str | None = None
) -> pytest.RunResult:
self.testdir.makepyfile(script)
args = []
if launch_mode_conf is not None:
args.append('--script-launch-mode=' + launch_mode_conf)
result = self.testdir.runpytest(*args)
print('\n'.join(['pytest stdout:'] + result.outlines +
['pytest stderr:'] + result.errlines))
result.assert_outcomes(passed=passed, skipped=skipped, failed=failed)
return result
@pytest.fixture
def run_test(testdir: pytest.Testdir) -> RunTest:
return RunTest(testdir)
CHECK_LAUNCH_MODE = """
def test_both(script_runner, accumulator=set()):
assert script_runner.launch_mode in {}
assert script_runner.launch_mode not in accumulator
accumulator.add(script_runner.launch_mode)
"""
def test_command_line_option(
run_test: RunTest, launch_mode_conf: str | None, launch_modes: set[str]
) -> None:
run_test(
CHECK_LAUNCH_MODE.format(launch_modes),
passed=len(launch_modes),
launch_mode_conf=launch_mode_conf
)
def test_config_option(
run_test: RunTest,
testdir: pytest.Testdir,
launch_mode_conf: str | None,
launch_modes: set[str],
) -> None:
if launch_mode_conf is not None:
testdir.makeini(f"""
[pytest]
script_launch_mode = {launch_mode_conf}
""")
run_test(
CHECK_LAUNCH_MODE.format(launch_modes),
passed=len(launch_modes)
)
def test_override_launch_mode_with_mark(
run_test: RunTest, launch_mode_conf: str | None
) -> None:
run_test(
"""
import pytest
@pytest.mark.script_launch_mode('inprocess')
def test_inprocess(script_runner):
assert script_runner.launch_mode == 'inprocess'
@pytest.mark.script_launch_mode('subprocess')
def test_subprocess(script_runner):
assert script_runner.launch_mode == 'subprocess'
@pytest.mark.script_launch_mode('both')
def test_both(script_runner, accumulator=set()):
assert script_runner.launch_mode not in accumulator
accumulator.add(script_runner.launch_mode)
""",
passed=4,
launch_mode_conf=launch_mode_conf
)
def test_help_message(testdir: pytest.Testdir) -> None:
result = testdir.runpytest(
'--help',
)
# fnmatch_lines does an assertion internally
result.stdout.fnmatch_lines([
'console-scripts:',
'*--script-launch-mode=*',
'*--hide-run-results*',
])
|
