chore: add more tests (#3187)

This commit is contained in:
Kevin Wan
2023-04-29 22:59:07 +08:00
committed by GitHub
parent 14caf5c799
commit a31256b327
4 changed files with 42 additions and 1 deletions

View File

@@ -76,3 +76,14 @@ func (b *Buffer) Reset() {
func (b *Buffer) String() string {
return b.buf.String()
}
func PanicOnFatal(t *testing.T) {
ok := logx.ExitOnFatal.CompareAndSwap(true, false)
if !ok {
return
}
t.Cleanup(func() {
logx.ExitOnFatal.CompareAndSwap(false, true)
})
}