Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: add TestIsUnavailableErr

Browse Source 
Signed-off-by: wafuwafu13 <mariobaske@i.softbank.jp>
This commit is contained in:
wafuwafu13 2022-12-19 22:18:03 +09:00
parent 77e4e87ee7
commit dc88d90763
1 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
client/v3/client_test.go
View File

@ -158,29 +158,59 @@ func TestIsHaltErr(t *testing.T) {
assert.Equal(t, assert.Equal(t,
isHaltErr(context.TODO(), errors.New("etcdserver: some etcdserver error")), isHaltErr(context.TODO(), errors.New("etcdserver: some etcdserver error")),
true, true,
`error prefixed with "etcdserver: " should be Halted by default`, `error prefixed with "etcdserver: " should be halt error by default`,
) )
assert.Equal(t, assert.Equal(t,
isHaltErr(context.TODO(), rpctypes.ErrGRPCStopped), isHaltErr(context.TODO(), rpctypes.ErrGRPCStopped),
false, false,
fmt.Sprintf("error %v should not halt", rpctypes.ErrGRPCStopped), fmt.Sprintf(`error "%v" should not be halt error`, rpctypes.ErrGRPCStopped),
) )
assert.Equal(t, assert.Equal(t,
isHaltErr(context.TODO(), rpctypes.ErrGRPCNoLeader), isHaltErr(context.TODO(), rpctypes.ErrGRPCNoLeader),
false, false,
fmt.Sprintf("error %v should not halt", rpctypes.ErrGRPCNoLeader), fmt.Sprintf(`error "%v" should not be halt error`, rpctypes.ErrGRPCNoLeader),
) )
ctx, cancel := context.WithCancel(context.TODO()) ctx, cancel := context.WithCancel(context.TODO())
assert.Equal(t, assert.Equal(t,
isHaltErr(ctx, nil), isHaltErr(ctx, nil),
false, false,
"no error and active context should not be Halted", "no error and active context should be halt error",
) )
cancel() cancel()
assert.Equal(t, assert.Equal(t,
isHaltErr(ctx, nil), isHaltErr(ctx, nil),
true, true,
"cancel on context should be Halted", "cancel on context should be halte error",
)
}
func TestIsUnavailableErr(t *testing.T) {
assert.Equal(t,
isUnavailableErr(context.TODO(), errors.New("etcdserver: some etcdserver error")),
false,
`error prefixed with "etcdserver: " should not be unavailable error by default`,
)
assert.Equal(t,
isUnavailableErr(context.TODO(), rpctypes.ErrGRPCStopped),
true,
fmt.Sprintf(`error "%v" should be unavailable error`, rpctypes.ErrGRPCStopped),
)
assert.Equal(t,
isUnavailableErr(context.TODO(), rpctypes.ErrGRPCNotCapable),
false,
fmt.Sprintf("error %v should not be unavailable error", rpctypes.ErrGRPCNotCapable),
)
ctx, cancel := context.WithCancel(context.TODO())
assert.Equal(t,
isUnavailableErr(ctx, nil),
false,
"no error and active context should not be unavailable error",
)
cancel()
assert.Equal(t,
isUnavailableErr(ctx, nil),
false,
"cancel on context should not be unavailable error",
) )
} }