Set GetOnFailure=true to better simulate k8s traffic

Signed-off-by: Aleksander Mistewicz <[email protected]>
etcd-io · Jan 7, 2025 · 45b2933 · 45b2933
1 parent 1270eed
commit 45b2933
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/robustness/traffic/kubernetes.go b/tests/robustness/traffic/kubernetes.go
@@ -146,7 +146,7 @@ func (t kubernetesTraffic) Write(ctx context.Context, kc kubernetes.Interface, i
 			return errors.New("storage empty")
 		}
 		if count > t.averageKeyCount*3/2 && nonUniqueWriteLimiter.Take() {
-			_, err = kc.OptimisticDelete(writeCtx, key, rev, kubernetes.DeleteOptions{})
+			_, err = kc.OptimisticDelete(writeCtx, key, rev, kubernetes.DeleteOptions{GetOnFailure: true})
 			nonUniqueWriteLimiter.Return()
 		} else {
 			shouldReturn := false
@@ -158,9 +158,9 @@ func (t kubernetesTraffic) Write(ctx context.Context, kc kubernetes.Interface, i
 			op := random.PickRandom(choices)
 			switch op {
 			case KubernetesDelete:
-				_, err = kc.OptimisticDelete(writeCtx, key, rev, kubernetes.DeleteOptions{})
+				_, err = kc.OptimisticDelete(writeCtx, key, rev, kubernetes.DeleteOptions{GetOnFailure: true})
 			case KubernetesUpdate:
-				_, err = kc.OptimisticPut(writeCtx, key, []byte(fmt.Sprintf("%d", ids.NewRequestID())), rev, kubernetes.PutOptions{})
+				_, err = kc.OptimisticPut(writeCtx, key, []byte(fmt.Sprintf("%d", ids.NewRequestID())), rev, kubernetes.PutOptions{GetOnFailure: true})
 			case KubernetesCreate:
 				_, err = kc.OptimisticPut(writeCtx, t.generateKey(), []byte(fmt.Sprintf("%d", ids.NewRequestID())), rev, kubernetes.PutOptions{})
 			default: