Browse Source

clientv3/integration: match more errors in put retries

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
Gyu-Ho Lee 8 years ago
parent
commit
0ca8f420d4
1 changed files with 2 additions and 1 deletions
  1. 2 1
      clientv3/integration/server_shutdown_test.go

+ 2 - 1
clientv3/integration/server_shutdown_test.go

@@ -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)