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
|
import os
import pathlib
import sys
import pytest
from pr2test.marks import require_root
pytestmark = [require_root()]
def get_examples(*argv):
root = pathlib.Path(os.environ['WORKSPACE'])
examples = [
file
for file in root.joinpath(*argv).iterdir()
if not file.name.endswith('.swp')
]
return {
'argnames': 'example',
'argvalues': examples,
'ids': [x.name for x in examples],
}
@pytest.mark.parametrize(**get_examples('examples', 'pyroute2-cli'))
def test_cli_examples(example, pytester, context):
with example.open('r') as text:
result = pytester.run('pyroute2-cli', stdin=text)
assert result.ret == 0
@pytest.mark.parametrize(**get_examples('examples', 'ndb'))
def test_ndb_examples(example, pytester, context):
argv = []
with example.open('r') as text:
for line in text.readlines():
line = line.strip()
if line == ':notest:':
pytest.skip()
elif line.startswith(':test:argv:'):
argv.append(line.split(':')[-1])
elif line.startswith(':test:environ:'):
key, value = line.split(':')[-1].split('=')
os.environ[key] = value
result = pytester.run(sys.executable, example.as_posix(), *argv)
assert result.ret == 0
def test_basic(tmpdir, pytester, context):
pytester.makefile('.pr2', test='interfaces lo mtu')
with open('test.pr2', 'r') as text:
result = pytester.run("pyroute2-cli", stdin=text)
assert result.ret == 0
assert result.outlines == ['65536']
|
