You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
728 B
Go
43 lines
728 B
Go
package rescue
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func init() {
|
|
logx.Disable()
|
|
}
|
|
|
|
func TestRescue(t *testing.T) {
|
|
var count int32
|
|
assert.NotPanics(t, func() {
|
|
defer Recover(func() {
|
|
atomic.AddInt32(&count, 2)
|
|
}, func() {
|
|
atomic.AddInt32(&count, 3)
|
|
})
|
|
|
|
panic("hello")
|
|
})
|
|
assert.Equal(t, int32(5), atomic.LoadInt32(&count))
|
|
}
|
|
|
|
func TestRescueCtx(t *testing.T) {
|
|
var count int32
|
|
assert.NotPanics(t, func() {
|
|
defer RecoverCtx(context.Background(), func() {
|
|
atomic.AddInt32(&count, 2)
|
|
}, func() {
|
|
atomic.AddInt32(&count, 3)
|
|
})
|
|
|
|
panic("hello")
|
|
})
|
|
assert.Equal(t, int32(5), atomic.LoadInt32(&count))
|
|
}
|