mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-23 17:20:24 +08:00
28 lines
428 B
Go
28 lines
428 B
Go
package rescue
|
|
|
|
import (
|
|
"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))
|
|
}
|