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
|
import config
from test import Test
class Suite:
available = []
@classmethod
def run_all(cls):
for s in cls.available:
s.run()
@classmethod
def setup(cls, asked_names: [str]):
if len(asked_names) == 0:
asked_names = [s.name for s in cls.available]
cls.available = [s for s in cls.available if s.name in asked_names]
for s in cls.available:
s.generate()
@classmethod
def available_names(cls) -> [str]:
return [s.name for s in cls.available]
def __init__(self, name: str):
self.name = name
self.generator_func = None
self.tests = []
def add(self, test):
self.tests.append(test)
def add_generator(self, generator):
self.generator_func = generator
def run(self):
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 generate(self):
self.generator_func()
def total(self) -> (int, int):
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("\nSummary:")
for s in cls.available:
(pass_total, fail_total) = s.total()
if pass_total == -1:
continue
print("{:<15} \033[32m{:2} [PASS]\033[0m \033[31m{:2} [FAIL]\033[0m"
.format(s.name, pass_total, fail_total))
@classmethod
def save_log(cls):
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')
def suite(origin):
""" decorator for a suite function (fmt: suite_[name]) """
name = origin.__name__[len("suite_"):]
s = Suite(name)
def test_generator():
def test(cmd: str, setup: str = "", files: [str] = [], exports: {str, str} = {}):
s.add(Test(cmd, setup, files, exports))
origin(test)
s.add_generator(test_generator)
Suite.available.append(s)
return test_generator
|
