From 472e716d06b7b1af18841f270b4d82c53f706351 Mon Sep 17 00:00:00 2001
From: "tung.tq" <quangtung29121997@gmail.com>
Date: Tue, 9 Apr 2024 09:23:16 +0700
Subject: [PATCH] Add Operation Error Probability

---
 curator/fake_property.go      | 57 +++++++++++++++++++++++++++++++++--
 curator/fake_property_test.go | 41 +++++++++++++++++++++++++
 2 files changed, 95 insertions(+), 3 deletions(-)

diff --git a/curator/fake_property.go b/curator/fake_property.go
index 25f90ec..07221e0 100644
--- a/curator/fake_property.go
+++ b/curator/fake_property.go
@@ -90,12 +90,51 @@ func (f *FakeZookeeperTester) doConnectionError(client FakeClientID) {
 	}
 }
 
+type runConfig struct {
+	opsErrorPercent float64
+}
+
+func (c runConfig) operationShouldError(randSource *rand.Rand) bool {
+	if c.opsErrorPercent == 0 {
+		return false
+	}
+	n := randSource.Intn(randMax)
+	end := int(c.opsErrorPercent / 100.0 * randMax)
+	if n < end {
+		return true
+	}
+	return false
+}
+
+func newRunConfig(options ...RunOption) runConfig {
+	conf := runConfig{
+		opsErrorPercent: 0.0,
+	}
+	for _, fn := range options {
+		fn(&conf)
+	}
+	return conf
+}
+
+// RunOption ...
+type RunOption func(conf *runConfig)
+
+// WithRunOperationErrorPercentage ...
+func WithRunOperationErrorPercentage(percent float64) RunOption {
+	return func(conf *runConfig) {
+		conf.opsErrorPercent = percent
+	}
+}
+
 // RunSessionExpiredAndConnectionError ...
 func (f *FakeZookeeperTester) RunSessionExpiredAndConnectionError(
 	sessionExpiredPercentage float64,
 	connectionErrorPercentage float64,
 	numSteps int,
+	options ...RunOption,
 ) int {
+	conf := newRunConfig(options...)
+
 	sessionExpiredEnd := int(sessionExpiredPercentage / 100.0 * randMax)
 	connectionErrorEnd := int(connectionErrorPercentage / 100.0 * randMax)
 
@@ -123,15 +162,27 @@ func (f *FakeZookeeperTester) RunSessionExpiredAndConnectionError(
 		genericCmd := f.store.Pending[client][0]
 		switch genericCmd.(type) {
 		case CreateInput:
-			f.store.CreateApply(client)
+			if conf.operationShouldError(f.rand) {
+				f.store.CreateApplyError(client)
+			} else {
+				f.store.CreateApply(client)
+			}
 		case GetInput:
 			f.store.GetApply(client)
 		case ChildrenInput:
 			f.store.ChildrenApply(client)
 		case DeleteInput:
-			f.store.DeleteApply(client)
+			if conf.operationShouldError(f.rand) {
+				f.store.DeleteApplyError(client)
+			} else {
+				f.store.DeleteApply(client)
+			}
 		case SetInput:
-			f.store.SetApply(client)
+			if conf.operationShouldError(f.rand) {
+				f.store.SetApplyError(client)
+			} else {
+				f.store.SetApply(client)
+			}
 		case RetryInput:
 			f.store.Retry(client)
 		default:
diff --git a/curator/fake_property_test.go b/curator/fake_property_test.go
index 760f5cc..6ad2c1a 100644
--- a/curator/fake_property_test.go
+++ b/curator/fake_property_test.go
@@ -3,6 +3,7 @@ package curator
 import (
 	"errors"
 	"fmt"
+	"math/rand"
 	"strconv"
 	"testing"
 	"time"
@@ -188,3 +189,43 @@ func TestFakeZookeeperTester_Master_Lock__Multi_Times(t *testing.T) {
 		)
 	}
 }
+
+func TestFakeZookeeperTester_Master_Lock__Multi_Times__With_Ops_Error(t *testing.T) {
+	for i := 0; i < 1000; i++ {
+		seed := time.Now().UnixNano()
+		fmt.Println("SEED:", seed)
+
+		store := NewFakeZookeeper()
+		tester := NewFakeZookeeperTester(store, []FakeClientID{
+			client1,
+			client2,
+			client3,
+		}, seed)
+
+		newSimpleLock(store, client1)
+		newSimpleLock(store, client2)
+		newSimpleLock(store, client3)
+
+		tester.Begin()
+
+		tester.RunSessionExpiredAndConnectionError(
+			10,
+			10,
+			1000,
+			WithRunOperationErrorPercentage(15),
+		)
+	}
+}
+
+func TestRunConfig(t *testing.T) {
+	c := newRunConfig(WithRunOperationErrorPercentage(20))
+
+	source := rand.New(rand.NewSource(1234))
+	trueCount := 0
+	for i := 0; i < 1000; i++ {
+		if c.operationShouldError(source) {
+			trueCount++
+		}
+	}
+	assert.Equal(t, 214, trueCount)
+}