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
146
147
148
149
150
|
package testutil
import (
"fmt"
"testing"
"time"
)
// AssertEventually retries an assertion until it passes or times out
func AssertEventually(t *testing.T, assertion func() error, timeout time.Duration, interval time.Duration, msgAndArgs ...interface{}) {
t.Helper()
deadline := time.Now().Add(timeout)
var lastErr error
for time.Now().Before(deadline) {
if err := assertion(); err == nil {
return // Success
} else {
lastErr = err
}
time.Sleep(interval)
}
// Format the failure message
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "assertion failed"
}
t.Fatalf("%s after %v: %v", msg, timeout, lastErr)
}
// AssertNoError fails the test if err is not nil
func AssertNoError(t *testing.T, err error, msgAndArgs ...interface{}) {
t.Helper()
if err != nil {
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "unexpected error"
}
t.Fatalf("%s: %v", msg, err)
}
}
// AssertError fails the test if err is nil
func AssertError(t *testing.T, err error, msgAndArgs ...interface{}) {
t.Helper()
if err == nil {
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "expected error but got nil"
}
t.Fatal(msg)
}
}
// AssertEqual fails the test if expected != actual
func AssertEqual(t *testing.T, expected, actual interface{}, msgAndArgs ...interface{}) {
t.Helper()
if expected != actual {
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "values not equal"
}
t.Fatalf("%s: expected %v, got %v", msg, expected, actual)
}
}
// AssertNotEqual fails the test if expected == actual
func AssertNotEqual(t *testing.T, expected, actual interface{}, msgAndArgs ...interface{}) {
t.Helper()
if expected == actual {
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "values should not be equal"
}
t.Fatalf("%s: both values are %v", msg, expected)
}
}
// AssertGreaterThan fails the test if actual <= expected
func AssertGreaterThan(t *testing.T, expected, actual int, msgAndArgs ...interface{}) {
t.Helper()
if actual <= expected {
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "value not greater than expected"
}
t.Fatalf("%s: expected > %d, got %d", msg, expected, actual)
}
}
// AssertContains fails the test if slice doesn't contain item
func AssertContains(t *testing.T, slice []string, item string, msgAndArgs ...interface{}) {
t.Helper()
for _, s := range slice {
if s == item {
return // Found it
}
}
var msg string
if len(msgAndArgs) > 0 {
if format, ok := msgAndArgs[0].(string); ok {
msg = fmt.Sprintf(format, msgAndArgs[1:]...)
} else {
msg = fmt.Sprint(msgAndArgs...)
}
} else {
msg = "item not found in slice"
}
t.Fatalf("%s: %q not found in %v", msg, item, slice)
}
|
