-
Notifications
You must be signed in to change notification settings - Fork 0
/
expect.go
47 lines (39 loc) · 892 Bytes
/
expect.go
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
package expect
import (
"testing"
)
type TestHandler struct {
tester *testing.T
rules *FormatRules
}
func Init(tester *testing.T, rules *FormatRules) *TestHandler {
if rules == nil {
rules = GlobalFormatRules
}
return &TestHandler{tester: tester, rules: rules}
}
type ExpectHandler struct {
tester *testing.T
rules *FormatRules
handler any
}
func (t *TestHandler) Expect(actual any) *ExpectHandler {
return &ExpectHandler{
tester: t.tester,
rules: t.rules,
handler: actual,
}
}
var prefix = len("expect.go:36: ")
func (e *ExpectHandler) Error(message string) {
e.tester.Error(backspace(prefix), message)
}
func (e *ExpectHandler) ToBe(expect any) {
position := Trace(2)
if e.handler != expect {
e.Error(yellow(position.String()) +
dim(": Expect: ") + green(format(expect, e.rules)) +
dim(", but get: ") + red(format(e.handler, e.rules)),
)
}
}