initial import

core/breaker/breakers_test.go

@@ -0,0 +1,115 @@
+package breaker
+
+import (
+	"errors"
+	"fmt"
+	"testing"
+
+	"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) {
+	NoBreakFor("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))
+}