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
|
package check
import (
"fmt"
"os"
"sort"
"strings"
"sync"
"testing"
)
type counter struct {
name string
value int
force bool
color string
size int
}
func (c counter) String() (s string) {
if c.value != 0 || c.force {
color := c.color
if c.value == 0 {
color = ansiReset
}
s = fmt.Sprintf("%s%*d %s%s", color, c.size, c.value, c.name, ansiReset)
} else {
s = strings.Repeat(" ", c.size+1+len(c.name))
}
return s
}
type testStat struct {
name string
passed counter
forged counter
failed counter
}
func newTestStat(desc string, force bool) *testStat {
return &testStat{
name: desc,
passed: counter{force: force, name: "passed", color: ansiGreen},
forged: counter{force: force, name: "todo", color: ansiYellow},
failed: counter{force: force, name: "failed", color: ansiRed},
}
}
func (c testStat) String() string {
return fmt.Sprintf("checks: %s %s %s\t%s", c.passed, c.forged, c.failed, c.name)
}
//nolint:gochecknoglobals // By design.
var (
statsMu sync.Mutex
stats = make(map[*testing.T]*testStat)
)
// Report output statistics about passed/failed checks.
// It should be called from TestMain after m.Run(), for ex.:
//
// func TestMain(m *testing.M) {
// code := m.Run()
// check.Report()
// os.Exit(code)
// }
//
// If this is all you need - just use TestMain instead.
func Report() {
statsMu.Lock()
defer statsMu.Unlock()
total := newTestStat("(total)", true)
ts := make([]*testing.T, 0, len(stats))
for t := range stats {
ts = append(ts, t)
total.passed.value += stats[t].passed.value
total.forged.value += stats[t].forged.value
total.failed.value += stats[t].failed.value
}
total.passed.size = digits(total.passed.value)
total.forged.size = digits(total.forged.value)
total.failed.size = digits(total.failed.value)
if testing.Verbose() {
sort.Slice(ts, func(a, b int) bool { return ts[a].Name() < ts[b].Name() })
for _, t := range ts {
stats[t].passed.size = total.passed.size
stats[t].forged.size = total.forged.size
stats[t].failed.size = total.failed.size
fmt.Printf(" %s\n", stats[t])
}
}
fmt.Printf(" %s\n", total)
}
// TestMain provides same default implementation as used by testing
// package with extra Report call to output statistics. Usage:
//
// func TestMain(m *testing.M) { check.TestMain(m) }
func TestMain(m *testing.M) {
code := m.Run()
Report()
os.Exit(code) //nolint:revive // By design.
}
|
