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.
29 lines
405 B
Go
29 lines
405 B
Go
4 years ago
|
package rescue
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync/atomic"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"zero/core/logx"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
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))
|
||
|
|
}
|