package breaker import ( "errors" "fmt" "testing" "git.i2edu.net/i2/go-zero/core/stat" "github.com/stretchr/testify/assert" ) 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 || 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, err == errDummy || err == ErrServiceUnavailable) } verify(t, func() bool { return ErrServiceUnavailable == Do("another", func() error { return nil }) }) } 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 || err == errDummy) } verify(t, func() bool { return ErrServiceUnavailable == Do("fallback", func() error { return nil }) }) } 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 || err == errDummy) } verify(t, func() bool { return ErrServiceUnavailable == Do("acceptablefallback", func() error { return nil }) }) } 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)) }