From 16e38e49a9b36e945630eead9686ef782be015b1 Mon Sep 17 00:00:00 2001
From: tangcong <tangcong506@gmail.com>
Date: Mon, 26 Apr 2021 12:07:05 +0800
Subject: [PATCH] functional: add FAILPOINTS_WITH_DISK_IO_LATENCY case

---
 tests/functional/rpcpb/rpc.proto              |  3 +
 .../tester/case_failpoints_disk_io.go         | 71 +++++++++++++++++++
 tests/functional/tester/cluster.go            |  7 ++
 3 files changed, 81 insertions(+)
 create mode 100644 tests/functional/tester/case_failpoints_disk_io.go

diff --git a/tests/functional/rpcpb/rpc.proto b/tests/functional/rpcpb/rpc.proto
index 79e0d352b..7d0fc6f31 100644
--- a/tests/functional/rpcpb/rpc.proto
+++ b/tests/functional/rpcpb/rpc.proto
@@ -625,6 +625,9 @@ enum Case {
   // in critical code paths.
   FAILPOINTS = 400;
 
+  // FAILPOINTS_WITH_DISK_IO_LATENCY injects high disk I/O latency failure in raftAfterSave code paths.
+  FAILPOINTS_WITH_DISK_IO_LATENCY = 401;
+
   // EXTERNAL runs external failure injection scripts.
   EXTERNAL = 500;
 }
diff --git a/tests/functional/tester/case_failpoints_disk_io.go b/tests/functional/tester/case_failpoints_disk_io.go
new file mode 100644
index 000000000..4cc2396b6
--- /dev/null
+++ b/tests/functional/tester/case_failpoints_disk_io.go
@@ -0,0 +1,71 @@
+// Copyright 2018 The etcd Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package tester
+
+import (
+	"fmt"
+	"strings"
+
+	"go.etcd.io/etcd/tests/v3/functional/rpcpb"
+)
+
+const (
+	diskIOFailpoint = "raftAfterSave"
+)
+
+func failpointDiskIOFailures(clus *Cluster) (ret []Case, err error) {
+	fps, err := failpointPaths(clus.Members[0].FailpointHTTPAddr)
+	if err != nil {
+		return nil, err
+	}
+	var detailDiskIOLatencyFailpointPath string
+	for i := 0; i < len(fps); i++ {
+		if strings.HasSuffix(fps[i], diskIOFailpoint) {
+			detailDiskIOLatencyFailpointPath = fps[i]
+			break
+		}
+	}
+	// create failure objects for diskIOFailpoint
+	fpFails := casesFromDiskIOFailpoint(detailDiskIOLatencyFailpointPath, clus.Tester.FailpointCommands)
+	// wrap in delays so failpoint has time to trigger
+	for i, fpf := range fpFails {
+		fpFails[i] = &caseDelay{
+			Case:          fpf,
+			delayDuration: clus.GetCaseDelayDuration(),
+		}
+	}
+	ret = append(ret, fpFails...)
+	return ret, nil
+}
+
+func casesFromDiskIOFailpoint(fp string, failpointCommands []string) (fs []Case) {
+	recov := makeRecoverFailpoint(fp)
+	for _, fcmd := range failpointCommands {
+		inject := makeInjectFailpoint(fp, fcmd)
+		fs = append(fs, []Case{
+			&caseLeader{
+				caseByFunc: caseByFunc{
+					desc:          fmt.Sprintf("failpoint %q (leader: %q)", fp, fcmd),
+					rpcpbCase:     rpcpb.Case_FAILPOINTS,
+					injectMember:  inject,
+					recoverMember: recov,
+				},
+				last: -1,
+				lead: -1,
+			},
+		}...)
+	}
+	return fs
+}
diff --git a/tests/functional/tester/cluster.go b/tests/functional/tester/cluster.go
index 44cff91ea..25d478a14 100644
--- a/tests/functional/tester/cluster.go
+++ b/tests/functional/tester/cluster.go
@@ -259,6 +259,13 @@ func (clus *Cluster) updateCases() {
 			}
 			clus.cases = append(clus.cases,
 				fpFailures...)
+		case "FAILPOINTS_WITH_DISK_IO_LATENCY":
+			fpFailures, fperr := failpointDiskIOFailures(clus)
+			if len(fpFailures) == 0 {
+				clus.lg.Info("no failpoints found!", zap.Error(fperr))
+			}
+			clus.cases = append(clus.cases,
+				fpFailures...)
 		}
 	}
 }