Implement terminate failpoint

Signed-off-by: Marek Siarkowicz <[email protected]>
etcd-io · Oct 17, 2023 · 5201d18 · 5201d18
1 parent be83fff
commit 5201d18
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/tests/robustness/failpoint/failpoint.go b/tests/robustness/failpoint/failpoint.go
@@ -47,6 +47,7 @@ var (
 		RaftBeforeSaveSnapPanic, RaftAfterSaveSnapPanic, BlackholeUntilSnapshot,
 		beforeApplyOneConfChangeSleep,
 		MemberReplace,
+		TerminateFailpoint,
 	}
 )
 

diff --git a/tests/robustness/failpoint/kill.go b/tests/robustness/failpoint/kill.go
@@ -27,12 +27,13 @@ import (
 )
 
 var (
-	KillFailpoint Failpoint = killFailpoint{}
+	KillFailpoint      Failpoint = sigkillFailpoint{}
+	TerminateFailpoint Failpoint = sigtermFailpoint{}
 )
 
-type killFailpoint struct{}
+type sigkillFailpoint struct{}
 
-func (f killFailpoint) Inject(ctx context.Context, t *testing.T, lg *zap.Logger, clus *e2e.EtcdProcessCluster) error {
+func (f sigkillFailpoint) Inject(ctx context.Context, t *testing.T, lg *zap.Logger, clus *e2e.EtcdProcessCluster) error {
 	member := clus.Procs[rand.Int()%len(clus.Procs)]
 
 	for member.IsRunning() {
@@ -60,10 +61,25 @@ func (f killFailpoint) Inject(ctx context.Context, t *testing.T, lg *zap.Logger,
 	return nil
 }
 
-func (f killFailpoint) Name() string {
+func (f sigkillFailpoint) Name() string {
 	return "Kill"
 }
 
-func (f killFailpoint) Available(e2e.EtcdProcessClusterConfig, e2e.EtcdProcess) bool {
+func (f sigkillFailpoint) Available(e2e.EtcdProcessClusterConfig, e2e.EtcdProcess) bool {
+	return true
+}
+
+type sigtermFailpoint struct{}
+
+func (f sigtermFailpoint) Inject(ctx context.Context, t *testing.T, lg *zap.Logger, clus *e2e.EtcdProcessCluster) error {
+	member := clus.Procs[rand.Int()%len(clus.Procs)]
+	return member.Restart(ctx)
+}
+
+func (f sigtermFailpoint) Name() string {
+	return "Terminate"
+}
+
+func (f sigtermFailpoint) Available(e2e.EtcdProcessClusterConfig, e2e.EtcdProcess) bool {
 	return true
 }