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
137
138
139
140
141
142
143
144
145
|
// SPDX-FileCopyrightText: © 2022 Grégoire Duchêne <gduchene@awhk.org>
// SPDX-License-Identifier: ISC
package core
import (
"errors"
"sync"
"testing"
"github.com/google/go-cmp/cmp"
)
// T is a wrapper around the standard testing.T. It adds a few helper
// functions, but behaves otherwise like testing.T.
type T struct {
*testing.T
Options cmp.Options
wg sync.WaitGroup
}
func (t *T) Assert(b bool) bool {
t.Helper()
if !b {
t.Error("\nexpected value to be true")
}
return b
}
func (t *T) AssertEqual(exp, actual any) bool {
t.Helper()
diff := cmp.Diff(exp, actual, t.Options...)
if diff == "" {
return true
}
t.Errorf("\nexpected %#v, got %#v\n%s", exp, actual, diff)
return false
}
func (t *T) AssertErrorAs(target any, err error) bool {
t.Helper()
if errors.As(err, target) {
return true
}
t.Errorf("\nexpected error chain to contain %#v, got %#v", target, err)
return false
}
func (t *T) AssertErrorIs(target, err error) bool {
t.Helper()
if errors.Is(err, target) {
return true
}
t.Errorf("\nexpected error chain to contain %#v, got %#v", target, err)
return false
}
func (t *T) AssertPanics(f func()) bool {
t.Helper()
return t.AssertPanicsWith(f, nil)
}
func (t *T) AssertPanicsWith(f func(), exp any) (b bool) {
t.Helper()
defer func() {
t.Helper()
actual := recover()
switch {
case actual == nil:
t.Errorf("\nexpected panic")
b = false
case exp == nil:
default:
b = t.AssertEqual(exp, actual)
}
}()
f()
return true
}
func (t *T) AssertNot(b bool) bool {
t.Helper()
if b {
t.Error("\nexpected value to be false")
}
return !b
}
func (t *T) AssertNotEqual(notExp, actual any) bool {
t.Helper()
if !cmp.Equal(notExp, actual, t.Options...) {
return true
}
t.Errorf("\nunexpected %#v", actual)
return false
}
func (t *T) AssertNotPanics(f func()) (b bool) {
t.Helper()
defer func() {
if actual := recover(); actual != nil {
t.Errorf("\nunexpected panic with %#v", actual)
b = false
}
}()
f()
return true
}
func (t *T) Go(f func()) {
t.wg.Add(1)
go func() {
defer t.wg.Done()
f()
}()
}
func (t *T) Must(b bool) {
if !b {
t.FailNow()
}
}
func (t *T) Run(name string, f func(t *T)) {
t.T.Run(name, func(s *testing.T) {
o := &T{T: s, Options: make(cmp.Options, len(t.Options))}
copy(o.Options, t.Options)
f(o)
o.wg.Wait()
})
}
func (t *T) Wait() { t.wg.Wait() }
|
