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
|
#!/usr/bin/env python
# Copyright (c) 2015 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import robot
from robot.run import RobotFramework
from robot.conf.settings import RobotSettings
from robot.running.builder import TestSuiteBuilder
from robot.running.model import TestSuite
def get_suite_list(*datasources, **options):
class _MyRobotFramework(RobotFramework):
def main(self, datasources, **options):
# copied from robot.run.RobotFramework.main
settings = RobotSettings(options)
suite = TestSuiteBuilder(settings['SuiteNames'],
settings['WarnOnSkipped']).build(*datasources)
suite.configure(**settings.suite_config)
return suite
# Options are in robot.conf.settings
suite = _MyRobotFramework().execute(*datasources, **options)
if isinstance(suite, TestSuite):
suites = []
suites.append(suite)
append_new = True
while append_new:
append_new = False
tmp = []
for s in suites:
if len(s.suites._items) > 0:
for i in s.suites._items:
tmp.append(i)
append_new = True
else:
tmp.append(s)
suites = tmp
return suites
else:
# TODO: add from robot.errors typ error
return []
def run_suites(test_dir, suites, out_dir="./outputs", **options):
# TODO: add logic
try:
for f in os.listdir(out_dir):
os.remove('/'.join((out_dir, f)))
except OSError:
pass
if not os.path.exists(out_dir):
os.makedirs(out_dir)
for s in suites:
longname = s.longname
varfile=[]
varfile.append('resources/libraries/python/topology.py')
# TODO: check testcases Tags
with open('{}/{}.out'.format(out_dir, longname), 'w') as out, \
open('{}/{}.log'.format(out_dir, longname), 'w') as debug:
robot.run(test_dir,
suite=[longname],
output='{}/{}.xml'.format(out_dir, longname),
debugfile=debug,
log=None,
report=None,
stdout=out,
variablefile=varfile,
**options)
def parse_outputs(out_dir='./'):
outs = ['/'.join((out_dir, file)) for file in os.listdir(out_dir) if file.endswith('.xml')]
robot.rebot(*outs, merge=True)
if __name__ == "__main__":
i = []
e = []
# i = ['bd', 'ip']
# i = ['hw']
# e = ['hw']
test_dir = "./tests"
out_dir = "./outputs"
suite_list = get_suite_list(test_dir, include=i, exclude=e, output=None, dryrun=True)
run_suites(test_dir, suite_list, include=i, exclude=e, out_dir=out_dir)
parse_outputs(out_dir=out_dir)
|
