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
129
130
131
132
133
134
135
136
|
# ############################################################################ #
# #
# ::: :::::::: #
# suite.py :+: :+: :+: #
# +:+ +:+ +:+ #
# By: charles <charles.cabergs@gmail.com> +#+ +:+ +#+ #
# +#+#+#+#+#+ +#+ #
# Created: 2020/07/15 18:24:29 by charles #+# #+# #
# Updated: 2020/09/12 10:53:42 by charles ### ########.fr #
# #
# ############################################################################ #
import sys
import config
class Suite:
available = []
@classmethod
def run_all(cls):
"""Run all available suites"""
for s in cls.available:
s.run()
@classmethod
def setup(cls, asked_names: [str]):
"""Remove not asked suite from available suites"""
if not config.BONUS:
cls.available = [s for s in cls.available if not s.bonus]
if len(asked_names) == 0:
asked_names = [s.name for s in cls.available]
suite_names = [s.name for s in cls.available]
names = []
for i, name in enumerate(asked_names):
if name in suite_names:
names.append(name)
continue
matches = [n for n in suite_names
if n.find("/") != -1
and n[n.find("/") + 1:].startswith(name)
or n.startswith(name)]
if len(matches) == 1:
names.append(matches[0])
elif len(matches) != 0 and all([n.startswith(name) for n in matches]):
names.extend(matches)
elif len(matches) > 2:
print(("Ambiguous name `{}` match the following suites\n\t{}\n"
"Try to run with -l to see the available suites")
.format(name, ', '.join(matches)))
sys.exit(1)
elif len(matches) == 0:
print(("Name `{}` doesn't match any suite/group name\n\t"
"Try to run with -l to see the available suites")
.format(name))
sys.exit(1)
cls.available = list(set(
[s for s in cls.available if s.name in names]
+ [s for s in cls.available if any([g for g in s.groups if g in names])]
))
for s in cls.available:
s.generator_func()
@classmethod
def available_names(cls) -> [str]:
"""List of available suites names"""
return [s.name for s in cls.available]
def __init__(self, name: str, groups: [str], bonus: bool = False):
"""Suite class
name: suite id
groups: list of suite groups
bonus: is this suite bonus
"""
self.name = name
self.groups = groups
self.bonus = bonus
self.generator_func = None
self.tests = []
def add(self, test):
"""Append a test to the suite"""
self.tests.append(test)
def run(self):
"""Run all test in the suite"""
if config.VERBOSE_LEVEL == 0:
print(self.name + ": ", end="")
else:
print("{} {:#<41}".format("#" * 39, self.name + " "))
for t in self.tests:
t.run()
if config.VERBOSE_LEVEL == 0:
print()
def total(self) -> (int, int):
"""Returns the total of passed and failed tests"""
passed_total = 0
for t in self.tests:
if t.result is None:
return (-1, -1)
if t.result.passed:
passed_total += 1
return (passed_total, len(self.tests) - passed_total)
@classmethod
def summarize(cls):
"""Print a summary of all runned suites"""
pass_sum = 0
fail_sum = 0
print("\nSummary:")
for s in cls.available:
(pass_total, fail_total) = s.total()
if pass_total == -1:
continue
pass_sum += pass_total
fail_sum += fail_total
print("{:<30} \033[32m{:3} [PASS]\033[0m \033[31m{:3} [FAIL]\033[0m"
.format(s.name, pass_total, fail_total))
print("{:<30} \033[32m{:3} [PASS]\033[0m \033[31m{:3} [FAIL]\033[0m"
.format("TOTAL", pass_sum, fail_sum))
@classmethod
def save_log(cls):
"""Save the result of all suites to a file"""
with open(config.LOG_PATH, "w") as log_file:
for s in cls.available:
for t in s.tests:
if t.result is not None and t.result.failed:
t.result.colored = False
t.result.set_colors()
log_file.write(t.result.full_diff() + '\n')
|
