Merge pull request #1844 from aaronlehmann/speed-up-TestRestartLeader

integration: Speed up TestRestartLeader

Author: LK4D4

integration/integration_test.go

@@ -374,7 +374,7 @@ func TestDemoteDownedManager(t *testing.T) {
 }
 
 func TestRestartLeader(t *testing.T) {
-	numWorker, numManager := 10, 5
+	numWorker, numManager := 5, 3
 	cl := newCluster(t, numWorker, numManager)
 	defer func() {
 		require.NoError(t, cl.Stop())
@@ -386,19 +386,6 @@ func TestRestartLeader(t *testing.T) {
 
 	require.NoError(t, leader.Pause())
 
-	require.NoError(t, raftutils.PollFuncWithTimeout(nil, func() error {
-		resp, err := cl.api.GetNode(context.Background(), &api.GetNodeRequest{
-			NodeID: origLeaderID,
-		})
-		if err != nil {
-			return err
-		}
-		if resp.Node.Status.State != api.NodeStatus_DOWN {
-			return errors.Errorf("node %s is still not down", origLeaderID)
-		}
-		return nil
-	}, opsTimeout))
-
 	require.NoError(t, raftutils.PollFuncWithTimeout(nil, func() error {
 		resp, err := cl.api.ListNodes(context.Background(), &api.ListNodesRequest{})
 		if err != nil {