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.
115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package breaker
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/stat"
|
|
)
|
|
|
|
func init() {
|
|
stat.SetReporter(nil)
|
|
}
|
|
|
|
func TestBreakersDo(t *testing.T) {
|
|
assert.Nil(t, Do("any", func() error {
|
|
return nil
|
|
}))
|
|
|
|
errDummy := errors.New("any")
|
|
assert.Equal(t, errDummy, Do("any", func() error {
|
|
return errDummy
|
|
}))
|
|
}
|
|
|
|
func TestBreakersDoWithAcceptable(t *testing.T) {
|
|
errDummy := errors.New("anyone")
|
|
for i := 0; i < 10000; i++ {
|
|
assert.Equal(t, errDummy, GetBreaker("anyone").DoWithAcceptable(func() error {
|
|
return errDummy
|
|
}, func(err error) bool {
|
|
return err == nil || errors.Is(err, errDummy)
|
|
}))
|
|
}
|
|
verify(t, func() bool {
|
|
return Do("anyone", func() error {
|
|
return nil
|
|
}) == nil
|
|
})
|
|
|
|
for i := 0; i < 10000; i++ {
|
|
err := DoWithAcceptable("another", func() error {
|
|
return errDummy
|
|
}, func(err error) bool {
|
|
return err == nil
|
|
})
|
|
assert.True(t, errors.Is(err, errDummy) || errors.Is(err, ErrServiceUnavailable))
|
|
}
|
|
verify(t, func() bool {
|
|
return errors.Is(Do("another", func() error {
|
|
return nil
|
|
}), ErrServiceUnavailable)
|
|
})
|
|
}
|
|
|
|
func TestBreakersNoBreakerFor(t *testing.T) {
|
|
NoBreakerFor("any")
|
|
errDummy := errors.New("any")
|
|
for i := 0; i < 10000; i++ {
|
|
assert.Equal(t, errDummy, GetBreaker("any").Do(func() error {
|
|
return errDummy
|
|
}))
|
|
}
|
|
assert.Equal(t, nil, Do("any", func() error {
|
|
return nil
|
|
}))
|
|
}
|
|
|
|
func TestBreakersFallback(t *testing.T) {
|
|
errDummy := errors.New("any")
|
|
for i := 0; i < 10000; i++ {
|
|
err := DoWithFallback("fallback", func() error {
|
|
return errDummy
|
|
}, func(err error) error {
|
|
return nil
|
|
})
|
|
assert.True(t, err == nil || errors.Is(err, errDummy))
|
|
}
|
|
verify(t, func() bool {
|
|
return errors.Is(Do("fallback", func() error {
|
|
return nil
|
|
}), ErrServiceUnavailable)
|
|
})
|
|
}
|
|
|
|
func TestBreakersAcceptableFallback(t *testing.T) {
|
|
errDummy := errors.New("any")
|
|
for i := 0; i < 10000; i++ {
|
|
err := DoWithFallbackAcceptable("acceptablefallback", func() error {
|
|
return errDummy
|
|
}, func(err error) error {
|
|
return nil
|
|
}, func(err error) bool {
|
|
return err == nil
|
|
})
|
|
assert.True(t, err == nil || errors.Is(err, errDummy))
|
|
}
|
|
verify(t, func() bool {
|
|
return errors.Is(Do("acceptablefallback", func() error {
|
|
return nil
|
|
}), ErrServiceUnavailable)
|
|
})
|
|
}
|
|
|
|
func verify(t *testing.T, fn func() bool) {
|
|
var count int
|
|
for i := 0; i < 100; i++ {
|
|
if fn() {
|
|
count++
|
|
}
|
|
}
|
|
assert.True(t, count >= 80, fmt.Sprintf("should be greater than 80, actual %d", count))
|
|
}
|