41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
package connector
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/ezoic/go-kinesis"
|
|
)
|
|
|
|
func Test_isRecoverableError(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
err error
|
|
isRecoverable bool
|
|
}{
|
|
{err: &kinesis.Error{Code: "ProvisionedThroughputExceededException"}, isRecoverable: true},
|
|
{err: &kinesis.Error{Code: "Throttling"}, isRecoverable: true},
|
|
{err: &kinesis.Error{Code: "ServiceUnavailable"}, isRecoverable: true},
|
|
{err: &kinesis.Error{Code: "ExpiredIteratorException"}, isRecoverable: false},
|
|
{err: &net.OpError{Err: fmt.Errorf("connection reset by peer")}, isRecoverable: true},
|
|
{err: &net.OpError{Err: fmt.Errorf("unexpected error")}, isRecoverable: false},
|
|
{err: fmt.Errorf("an arbitrary error"), isRecoverable: false},
|
|
|
|
//"InternalFailure": true,
|
|
//"Throttling": true,
|
|
//"ServiceUnavailable": true,
|
|
////"ExpiredIteratorException": true,
|
|
//{err: *kinesis.Error{Code:""}}
|
|
}
|
|
|
|
for idx, tc := range testCases {
|
|
|
|
p := Pipeline{}
|
|
isRecoverable := p.isRecoverableError(tc.err)
|
|
if isRecoverable != tc.isRecoverable {
|
|
t.Errorf("test case %d: isRecoverable expected %t, actual %t, for error %+v", idx, tc.isRecoverable, isRecoverable, tc.err)
|
|
}
|
|
|
|
}
|
|
}
|