|
@@ -21,6 +21,7 @@ import (
|
|
|
"time"
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/coreos/etcd/clientv3"
|
|
"github.com/coreos/etcd/clientv3"
|
|
|
|
|
+ "github.com/coreos/etcd/etcdserver/api/v3rpc/rpctypes"
|
|
|
"github.com/coreos/etcd/integration"
|
|
"github.com/coreos/etcd/integration"
|
|
|
"github.com/coreos/etcd/pkg/testutil"
|
|
"github.com/coreos/etcd/pkg/testutil"
|
|
|
)
|
|
)
|
|
@@ -99,7 +100,7 @@ func TestBalancerUnderServerShutdownWatch(t *testing.T) {
|
|
|
if err == nil {
|
|
if err == nil {
|
|
|
break
|
|
break
|
|
|
}
|
|
}
|
|
|
- if err == context.DeadlineExceeded {
|
|
|
|
|
|
|
+ if err == context.DeadlineExceeded || err == rpctypes.ErrTimeout || err == rpctypes.ErrTimeoutDueToLeaderFail {
|
|
|
continue
|
|
continue
|
|
|
}
|
|
}
|
|
|
t.Fatal(err)
|
|
t.Fatal(err)
|