From 0a830c871ecb8029712e5f6434f3866bcc08e9f7 Mon Sep 17 00:00:00 2001
From: tangcong <tangcong506@gmail.com>
Date: Tue, 20 Oct 2020 23:07:21 +0800
Subject: [PATCH] clientv3test: fix network partition flaky test

---
 tests/integration/clientv3/network_partition_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integration/clientv3/network_partition_test.go b/tests/integration/clientv3/network_partition_test.go
index af2927b13..bc2849d5a 100644
--- a/tests/integration/clientv3/network_partition_test.go
+++ b/tests/integration/clientv3/network_partition_test.go
@@ -74,7 +74,7 @@ func TestBalancerUnderNetworkPartitionTxn(t *testing.T) {
 func TestBalancerUnderNetworkPartitionLinearizableGetWithLongTimeout(t *testing.T) {
 	testBalancerUnderNetworkPartition(t, func(cli *clientv3.Client, ctx context.Context) error {
 		_, err := cli.Get(ctx, "a")
-		if err == rpctypes.ErrTimeout {
+		if isClientTimeout(err) || isServerCtxTimeout(err) || err == rpctypes.ErrTimeout {
 			return errExpected
 		}
 		return err