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
|
#!/usr/bin/python3
import argparse
import yaml
from fmf_metadata.base import (
yaml_fmf_output,
read_config,
debug_print,
dict_to_yaml,
)
from fmf_metadata.constants import MAIN_FMF, CONFIG_FMF_FILE
# disable references inside yaml files
setattr(yaml.SafeDumper, "ignore_aliases", lambda *args: True)
def arg_parser():
parser = argparse.ArgumentParser(
description="FMF formatter and wrapper for running tests under pytest"
)
parser.add_argument(
"--file",
dest="fmf_file",
action="store",
help="Use this FMF file (input and output when option --update)",
)
parser.add_argument(
"-u",
"--update",
dest="fmf_update",
action="store_true",
help="Update the selected FMF file",
)
parser.add_argument(
"--path",
dest="fmf_path",
action="store",
help="root path to test",
)
parser.add_argument(
"--config",
dest="config",
action="store",
help="Config file for fmf formatter",
)
parser.add_argument(
"--merge-plus",
dest="merge_plus",
action="append",
help="override post_mark for item elements (change to +)",
)
parser.add_argument(
"--merge-minus",
dest="merge_minus",
action="append",
help="override post_mark for item elements (change to -)",
)
parser.add_argument("tests", nargs="*")
return parser
def run():
opts = arg_parser().parse_args()
config = dict()
if opts.config:
config = read_config(opts.config)
fmf_file = opts.fmf_file or config.get(CONFIG_FMF_FILE, MAIN_FMF)
data = yaml_fmf_output(
fmf_file=opts.fmf_file,
path=opts.fmf_path,
testfile_globs=opts.tests,
config=config,
merge_minus_list=opts.merge_minus,
merge_plus_list=opts.merge_plus,
)
if opts.fmf_update:
debug_print(f"Update FMF file: {fmf_file}")
with open(fmf_file, "w") as fd:
fd.write(dict_to_yaml(data))
else:
print(dict_to_yaml(data))
if __name__ == "__main__":
run()
|
