initial import

core/stores/sqlx/mysql_test.go

@@ -0,0 +1,56 @@
+package sqlx
+
+import (
+	"testing"
+
+	"zero/core/breaker"
+	"zero/core/logx"
+	"zero/core/stat"
+
+	"github.com/go-sql-driver/mysql"
+	"github.com/stretchr/testify/assert"
+)
+
+func init() {
+	stat.SetReporter(nil)
+}
+
+func TestBreakerOnDuplicateEntry(t *testing.T) {
+	logx.Disable()
+
+	err := tryOnDuplicateEntryError(t, mysqlAcceptable)
+	assert.Equal(t, duplicateEntryCode, err.(*mysql.MySQLError).Number)
+}
+
+func TestBreakerOnNotHandlingDuplicateEntry(t *testing.T) {
+	logx.Disable()
+
+	var found bool
+	for i := 0; i < 100; i++ {
+		if tryOnDuplicateEntryError(t, nil) == breaker.ErrServiceUnavailable {
+			found = true
+		}
+	}
+	assert.True(t, found)
+}
+
+func tryOnDuplicateEntryError(t *testing.T, accept func(error) bool) error {
+	logx.Disable()
+
+	conn := commonSqlConn{
+		brk:    breaker.NewBreaker(),
+		accept: accept,
+	}
+	for i := 0; i < 1000; i++ {
+		assert.NotNil(t, conn.brk.DoWithAcceptable(func() error {
+			return &mysql.MySQLError{
+				Number: duplicateEntryCode,
+			}
+		}, conn.acceptable))
+	}
+	return conn.brk.DoWithAcceptable(func() error {
+		return &mysql.MySQLError{
+			Number: duplicateEntryCode,
+		}
+	}, conn.acceptable)
+}