1 #!/usr/bin/env python3 1 #!/usr/bin/env python3
2 2
3 from enum import Enum 3 from enum import Enum
4 4
5 class ResultState(Enum): 5 class ResultState(Enum):
6 noresult = -1 6 noresult = -1
7 skip = 0 7 skip = 0
8 success = 1 8 success = 1
9 fail = 2 9 fail = 2
10 10
11 class TestResult: 11 class TestResult:
12 def __init__(self, test_id="", test_name=" 12 def __init__(self, test_id="", test_name=""):
13 self.test_id = test_id 13 self.test_id = test_id
14 self.test_name = test_name 14 self.test_name = test_name
15 self.result = ResultState.noresult 15 self.result = ResultState.noresult
16 self.failmsg = "" 16 self.failmsg = ""
17 self.errormsg = "" 17 self.errormsg = ""
18 self.steps = [] 18 self.steps = []
19 19
20 def set_result(self, result): 20 def set_result(self, result):
21 if (isinstance(result, ResultState)): 21 if (isinstance(result, ResultState)):
22 self.result = result 22 self.result = result
23 return True 23 return True
24 else: 24 else:
25 raise TypeError('Unknown result ty 25 raise TypeError('Unknown result type, must be type ResultState')
26 26
27 def get_result(self): 27 def get_result(self):
28 return self.result 28 return self.result
29 29
30 def set_errormsg(self, errormsg): 30 def set_errormsg(self, errormsg):
31 self.errormsg = errormsg 31 self.errormsg = errormsg
32 return True 32 return True
33 33
34 def append_errormsg(self, errormsg): 34 def append_errormsg(self, errormsg):
35 self.errormsg = '{}\n{}'.format(self.e 35 self.errormsg = '{}\n{}'.format(self.errormsg, errormsg)
36 36
37 def get_errormsg(self): 37 def get_errormsg(self):
38 return self.errormsg 38 return self.errormsg
39 39
40 def set_failmsg(self, failmsg): 40 def set_failmsg(self, failmsg):
41 self.failmsg = failmsg 41 self.failmsg = failmsg
42 return True 42 return True
43 43
44 def append_failmsg(self, failmsg): 44 def append_failmsg(self, failmsg):
45 self.failmsg = '{}\n{}'.format(self.fa 45 self.failmsg = '{}\n{}'.format(self.failmsg, failmsg)
46 46
47 def get_failmsg(self): 47 def get_failmsg(self):
48 return self.failmsg 48 return self.failmsg
49 49
50 def add_steps(self, newstep): 50 def add_steps(self, newstep):
51 if type(newstep) == list: 51 if type(newstep) == list:
52 self.steps.extend(newstep) 52 self.steps.extend(newstep)
53 elif type(newstep) == str: 53 elif type(newstep) == str:
54 self.steps.append(step) 54 self.steps.append(step)
55 else: 55 else:
56 raise TypeError('TdcResults.add_st 56 raise TypeError('TdcResults.add_steps() requires a list or str')
57 57
58 def get_executed_steps(self): 58 def get_executed_steps(self):
59 return self.steps 59 return self.steps
60 60
61 class TestSuiteReport(): 61 class TestSuiteReport():
62 def __init__(self): 62 def __init__(self):
63 self._testsuite = [] 63 self._testsuite = []
64 64
65 def add_resultdata(self, result_data): 65 def add_resultdata(self, result_data):
66 if isinstance(result_data, TestResult) 66 if isinstance(result_data, TestResult):
67 self._testsuite.append(result_data 67 self._testsuite.append(result_data)
68 return True 68 return True
69 69
70 def count_tests(self): 70 def count_tests(self):
71 return len(self._testsuite) 71 return len(self._testsuite)
72 72
73 def count_failures(self): 73 def count_failures(self):
74 return sum(1 for t in self._testsuite 74 return sum(1 for t in self._testsuite if t.result == ResultState.fail)
75 75
76 def count_skips(self): 76 def count_skips(self):
77 return sum(1 for t in self._testsuite 77 return sum(1 for t in self._testsuite if t.result == ResultState.skip)
78 78
79 def find_result(self, test_id): 79 def find_result(self, test_id):
80 return next((tr for tr in self._testsu 80 return next((tr for tr in self._testsuite if tr.test_id == test_id), None)
81 81
82 def update_result(self, result_data): 82 def update_result(self, result_data):
83 orig = self.find_result(result_data.te 83 orig = self.find_result(result_data.test_id)
84 if orig != None: 84 if orig != None:
85 idx = self._testsuite.index(orig) 85 idx = self._testsuite.index(orig)
86 self._testsuite[idx] = result_data 86 self._testsuite[idx] = result_data
87 else: 87 else:
88 self.add_resultdata(result_data) 88 self.add_resultdata(result_data)
89 89
90 def format_tap(self): 90 def format_tap(self):
91 ftap = "" 91 ftap = ""
92 ftap += '1..{}\n'.format(self.count_te 92 ftap += '1..{}\n'.format(self.count_tests())
93 index = 1 93 index = 1
94 for t in self._testsuite: 94 for t in self._testsuite:
95 if t.result == ResultState.fail: 95 if t.result == ResultState.fail:
96 ftap += 'not ' 96 ftap += 'not '
97 ftap += 'ok {} {} - {}'.format(str 97 ftap += 'ok {} {} - {}'.format(str(index), t.test_id, t.test_name)
98 if t.result == ResultState.skip or 98 if t.result == ResultState.skip or t.result == ResultState.noresult:
99 ftap += ' # skipped - {}\n'.fo 99 ftap += ' # skipped - {}\n'.format(t.errormsg)
100 elif t.result == ResultState.fail: 100 elif t.result == ResultState.fail:
101 if len(t.steps) > 0: 101 if len(t.steps) > 0:
102 ftap += '\tCommands execut 102 ftap += '\tCommands executed in this test case:'
103 for step in t.steps: 103 for step in t.steps:
104 ftap += '\n\t\t{}'.for 104 ftap += '\n\t\t{}'.format(step)
105 ftap += '\n\t{}'.format(t.fail 105 ftap += '\n\t{}'.format(t.failmsg)
106 ftap += '\n' 106 ftap += '\n'
107 index += 1 107 index += 1
108 return ftap 108 return ftap
109 109
110 def format_xunit(self): 110 def format_xunit(self):
111 from xml.sax.saxutils import escape 111 from xml.sax.saxutils import escape
112 xunit = "<testsuites>\n" 112 xunit = "<testsuites>\n"
113 xunit += '\t<testsuite tests=\"{}\" sk 113 xunit += '\t<testsuite tests=\"{}\" skips=\"{}\">\n'.format(self.count_tests(), self.count_skips())
114 for t in self._testsuite: 114 for t in self._testsuite:
115 xunit += '\t\t<testcase classname= 115 xunit += '\t\t<testcase classname=\"{}\" '.format(escape(t.test_id))
116 xunit += 'name=\"{}\">\n'.format(e 116 xunit += 'name=\"{}\">\n'.format(escape(t.test_name))
117 if t.failmsg: 117 if t.failmsg:
118 xunit += '\t\t\t<failure>\n' 118 xunit += '\t\t\t<failure>\n'
119 if len(t.steps) > 0: 119 if len(t.steps) > 0:
120 xunit += 'Commands execute 120 xunit += 'Commands executed in this test case:\n'
121 for step in t.steps: 121 for step in t.steps:
122 xunit += '\t{}\n'.form 122 xunit += '\t{}\n'.format(escape(step))
123 xunit += 'FAILURE: {}\n'.forma 123 xunit += 'FAILURE: {}\n'.format(escape(t.failmsg))
124 xunit += '\t\t\t</failure>\n' 124 xunit += '\t\t\t</failure>\n'
125 if t.errormsg: 125 if t.errormsg:
126 xunit += '\t\t\t<error>\n{}\n' 126 xunit += '\t\t\t<error>\n{}\n'.format(escape(t.errormsg))
127 xunit += '\t\t\t</error>\n' 127 xunit += '\t\t\t</error>\n'
128 if t.result == ResultState.skip: 128 if t.result == ResultState.skip:
129 xunit += '\t\t\t<skipped/>\n' 129 xunit += '\t\t\t<skipped/>\n'
130 xunit += '\t\t</testcase>\n' 130 xunit += '\t\t</testcase>\n'
131 xunit += '\t</testsuite>\n' 131 xunit += '\t</testsuite>\n'
132 xunit += '</testsuites>\n' 132 xunit += '</testsuites>\n'
133 return xunit 133 return xunit
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.