зеркало из https://github.com/github/vitess-gh.git
Merge pull request #580 from yaoshengzhe/fix_add_tablet_error_unit_test
add tablet error unit test
This commit is contained in:
Shengzhe
Коммит
3fb7e03eaf
|
|
@ -0,0 +1,233 @@
|
|||
// Copyright 2015, Google Inc. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package tabletserver
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/youtube/vitess/go/mysql"
|
||||
"github.com/youtube/vitess/go/sqldb"
|
||||
"golang.org/x/net/context"
|
||||
)
|
||||
|
||||
func TestTabletErrorRetriableErrorTypeOverwrite(t *testing.T) {
|
||||
sqlErr := sqldb.NewSqlError(mysql.ErrOptionPreventsStatement, "read-only")
|
||||
tabletErr := NewTabletErrorSql(ErrFatal, sqlErr)
|
||||
if tabletErr.ErrorType != ErrRetry {
|
||||
t.Fatalf("tablet error should have error type: ErrRetry")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTabletErrorMsgTooLong(t *testing.T) {
|
||||
buf := make([]byte, 2*maxErrLen)
|
||||
for i := 0; i < 2*maxErrLen; i++ {
|
||||
buf[i] = 'a'
|
||||
}
|
||||
msg := string(buf)
|
||||
sqlErr := sqldb.NewSqlError(mysql.ErrDupEntry, msg)
|
||||
tabletErr := NewTabletErrorSql(ErrFatal, sqlErr)
|
||||
if tabletErr.ErrorType != ErrFatal {
|
||||
t.Fatalf("tablet error should have error type: ErrFatal")
|
||||
}
|
||||
if tabletErr.Message != string(buf[:maxErrLen]) {
|
||||
t.Fatalf("message should be capped, only %s character will be shown", maxErrLen)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTabletErrorConnError(t *testing.T) {
|
||||
tabletErr := NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(1999, "test"))
|
||||
if IsConnErr(tabletErr) {
|
||||
t.Fatalf("table error: %v is not a connection error", tabletErr)
|
||||
}
|
||||
tabletErr = NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(2000, "test"))
|
||||
if !IsConnErr(tabletErr) {
|
||||
t.Fatalf("table error: %v is a connection error", tabletErr)
|
||||
}
|
||||
tabletErr = NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(2013, "test"))
|
||||
if IsConnErr(tabletErr) {
|
||||
t.Fatalf("table error: %v is not a connection error", tabletErr)
|
||||
}
|
||||
sqlErr := sqldb.NewSqlError(1998, "test")
|
||||
if IsConnErr(sqlErr) {
|
||||
t.Fatalf("sql error: %v is not a connection error", sqlErr)
|
||||
}
|
||||
sqlErr = sqldb.NewSqlError(2001, "test")
|
||||
if !IsConnErr(sqlErr) {
|
||||
t.Fatalf("sql error: %v is a connection error", sqlErr)
|
||||
}
|
||||
|
||||
err := fmt.Errorf("(errno 2005)")
|
||||
if !IsConnErr(err) {
|
||||
t.Fatalf("error: %v is a connection error", err)
|
||||
}
|
||||
|
||||
err = fmt.Errorf("(errno 123456789012345678901234567890)")
|
||||
if IsConnErr(err) {
|
||||
t.Fatalf("error: %v is not a connection error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTabletErrorPrefix(t *testing.T) {
|
||||
tabletErr := NewTabletErrorSql(ErrRetry, sqldb.NewSqlError(2000, "test"))
|
||||
if tabletErr.Prefix() != "retry: " {
|
||||
t.Fatalf("tablet error with error type: ErrRetry should has prefix: 'retry: '")
|
||||
}
|
||||
tabletErr = NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(2000, "test"))
|
||||
if tabletErr.Prefix() != "fatal: " {
|
||||
t.Fatalf("tablet error with error type: ErrFatal should has prefix: 'fatal: '")
|
||||
}
|
||||
tabletErr = NewTabletErrorSql(ErrTxPoolFull, sqldb.NewSqlError(2000, "test"))
|
||||
if tabletErr.Prefix() != "tx_pool_full: " {
|
||||
t.Fatalf("tablet error with error type: ErrTxPoolFull should has prefix: 'tx_pool_full: '")
|
||||
}
|
||||
tabletErr = NewTabletErrorSql(ErrNotInTx, sqldb.NewSqlError(2000, "test"))
|
||||
if tabletErr.Prefix() != "not_in_tx: " {
|
||||
t.Fatalf("tablet error with error type: ErrNotInTx should has prefix: 'not_in_tx: '")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTabletErrorRecordStats(t *testing.T) {
|
||||
tabletErr := NewTabletErrorSql(ErrRetry, sqldb.NewSqlError(2000, "test"))
|
||||
retryCounterBefore := infoErrors.Counts()["Retry"]
|
||||
tabletErr.RecordStats()
|
||||
retryCounterAfter := infoErrors.Counts()["Retry"]
|
||||
if retryCounterAfter-retryCounterBefore != 1 {
|
||||
t.Fatalf("tablet error with error type ErrRetry should increase Retry error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(2000, "test"))
|
||||
fatalCounterBefore := infoErrors.Counts()["Fatal"]
|
||||
tabletErr.RecordStats()
|
||||
fatalCounterAfter := infoErrors.Counts()["Fatal"]
|
||||
if fatalCounterAfter-fatalCounterBefore != 1 {
|
||||
t.Fatalf("tablet error with error type ErrFatal should increase Fatal error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrTxPoolFull, sqldb.NewSqlError(2000, "test"))
|
||||
txPoolFullCounterBefore := errorStats.Counts()["TxPoolFull"]
|
||||
tabletErr.RecordStats()
|
||||
txPoolFullCounterAfter := errorStats.Counts()["TxPoolFull"]
|
||||
if txPoolFullCounterAfter-txPoolFullCounterBefore != 1 {
|
||||
t.Fatalf("tablet error with error type ErrTxPoolFull should increase TxPoolFull error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrNotInTx, sqldb.NewSqlError(2000, "test"))
|
||||
notInTxCounterBefore := errorStats.Counts()["NotInTx"]
|
||||
tabletErr.RecordStats()
|
||||
notInTxCounterAfter := errorStats.Counts()["NotInTx"]
|
||||
if notInTxCounterAfter-notInTxCounterBefore != 1 {
|
||||
t.Fatalf("tablet error with error type ErrNotInTx should increase NotInTx error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrFail, sqldb.NewSqlError(mysql.ErrDupEntry, "test"))
|
||||
dupKeyCounterBefore := infoErrors.Counts()["DupKey"]
|
||||
tabletErr.RecordStats()
|
||||
dupKeyCounterAfter := infoErrors.Counts()["DupKey"]
|
||||
if dupKeyCounterAfter-dupKeyCounterBefore != 1 {
|
||||
t.Fatalf("sql error with error type mysql.ErrDupEntry should increase DupKey error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrFail, sqldb.NewSqlError(mysql.ErrLockWaitTimeout, "test"))
|
||||
lockWaitTimeoutCounterBefore := errorStats.Counts()["Deadlock"]
|
||||
tabletErr.RecordStats()
|
||||
lockWaitTimeoutCounterAfter := errorStats.Counts()["Deadlock"]
|
||||
if lockWaitTimeoutCounterAfter-lockWaitTimeoutCounterBefore != 1 {
|
||||
t.Fatalf("sql error with error type mysql.ErrLockWaitTimeout should increase Deadlock error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrFail, sqldb.NewSqlError(mysql.ErrLockDeadlock, "test"))
|
||||
deadlockCounterBefore := errorStats.Counts()["Deadlock"]
|
||||
tabletErr.RecordStats()
|
||||
deadlockCounterAfter := errorStats.Counts()["Deadlock"]
|
||||
if deadlockCounterAfter-deadlockCounterBefore != 1 {
|
||||
t.Fatalf("sql error with error type mysql.ErrLockDeadlock should increase Deadlock error count by 1")
|
||||
}
|
||||
|
||||
tabletErr = NewTabletErrorSql(ErrFail, sqldb.NewSqlError(mysql.ErrOptionPreventsStatement, "test"))
|
||||
failCounterBefore := errorStats.Counts()["Fail"]
|
||||
tabletErr.RecordStats()
|
||||
failCounterAfter := errorStats.Counts()["Fail"]
|
||||
if failCounterAfter-failCounterBefore != 1 {
|
||||
t.Fatalf("sql error with error type mysql.ErrOptionPreventsStatement should increase Fail error count by 1")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTabletErrorHandleUncaughtError(t *testing.T) {
|
||||
var err error
|
||||
logStats := newSqlQueryStats("TestTabletErrorHandleError", context.Background())
|
||||
defer func() {
|
||||
_, ok := err.(*TabletError)
|
||||
if !ok {
|
||||
t.Fatalf("error should be a TabletError, but got error: %v", err)
|
||||
}
|
||||
}()
|
||||
defer handleError(&err, logStats)
|
||||
panic("unknown error")
|
||||
}
|
||||
|
||||
func TestTabletErrorHandleRetryError(t *testing.T) {
|
||||
var err error
|
||||
tabletErr := NewTabletErrorSql(ErrRetry, sqldb.NewSqlError(1000, "test"))
|
||||
logStats := newSqlQueryStats("TestTabletErrorHandleError", context.Background())
|
||||
defer func() {
|
||||
_, ok := err.(*TabletError)
|
||||
if !ok {
|
||||
t.Fatalf("error should be a TabletError, but got error: %v", err)
|
||||
}
|
||||
}()
|
||||
defer handleError(&err, logStats)
|
||||
panic(tabletErr)
|
||||
}
|
||||
|
||||
func TestTabletErrorHandleTxPoolFullError(t *testing.T) {
|
||||
var err error
|
||||
tabletErr := NewTabletErrorSql(ErrTxPoolFull, sqldb.NewSqlError(1000, "test"))
|
||||
logStats := newSqlQueryStats("TestTabletErrorHandleError", context.Background())
|
||||
defer func() {
|
||||
_, ok := err.(*TabletError)
|
||||
if !ok {
|
||||
t.Fatalf("error should be a TabletError, but got error: %v", err)
|
||||
}
|
||||
}()
|
||||
defer handleError(&err, logStats)
|
||||
panic(tabletErr)
|
||||
}
|
||||
|
||||
func TestTabletErrorLogUncaughtErr(t *testing.T) {
|
||||
panicCountBefore := internalErrors.Counts()["Panic"]
|
||||
defer func() {
|
||||
panicCountAfter := internalErrors.Counts()["Panic"]
|
||||
if panicCountAfter-panicCountBefore != 1 {
|
||||
t.Fatalf("Panic count should increase by 1 for uncaught panic")
|
||||
}
|
||||
}()
|
||||
defer logError()
|
||||
panic("unknown error")
|
||||
}
|
||||
|
||||
func TestTabletErrorTxPoolFull(t *testing.T) {
|
||||
tabletErr := NewTabletErrorSql(ErrTxPoolFull, sqldb.NewSqlError(1000, "test"))
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
t.Fatalf("error should have been handled already")
|
||||
}
|
||||
}()
|
||||
defer logError()
|
||||
panic(tabletErr)
|
||||
}
|
||||
|
||||
func TestTabletErrorFatal(t *testing.T) {
|
||||
tabletErr := NewTabletErrorSql(ErrFatal, sqldb.NewSqlError(1000, "test"))
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
t.Fatalf("error should have been handled already")
|
||||
}
|
||||
}()
|
||||
defer logError()
|
||||
panic(tabletErr)
|
||||
}
|
||||
Загрузка…
Ссылка в новой задаче