104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
|
||
|
|
var statsMu sync.Mutex
|
||
|
|
var stats = 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)
|
||
|
|
}
|