From d3b902b410faa2c54824b16ad5decc3e213a605e Mon Sep 17 00:00:00 2001
From: Abhishek Pandey <abhishek@alcion.ai>
Date: Wed, 6 Dec 2023 18:20:48 -0800
Subject: [PATCH] Add chaos middleware

---
 src/pkg/services/m365/api/graph/service.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/pkg/services/m365/api/graph/service.go b/src/pkg/services/m365/api/graph/service.go
index e005030bd..169ebd75c 100644
--- a/src/pkg/services/m365/api/graph/service.go
+++ b/src/pkg/services/m365/api/graph/service.go
@@ -285,6 +285,18 @@ func kiotaMiddlewares(
 	cc *clientConfig,
 	counter *count.Bus,
 ) []khttp.Middleware {
+	// Add chaos handler to default middleware
+	chaosOpt := &khttp.ChaosHandlerOptions{
+		ChaosStrategy:   khttp.Random,
+		ChaosPercentage: 50,
+		StatusCode:      429,
+	}
+
+	chaosHandler, err := khttp.NewChaosHandlerWithOptions(chaosOpt)
+	if err != nil {
+		panic(err)
+	}
+
 	retryOptions := khttp.RetryHandlerOptions{
 		ShouldRetry: func(
 			delay time.Duration,
@@ -311,6 +323,7 @@ func kiotaMiddlewares(
 		khttp.NewParametersNameDecodingHandler(),
 		khttp.NewUserAgentHandler(),
 		&LoggingMiddleware{},
+		chaosHandler,
 	}
 
 	// Optionally add concurrency limiter middleware if it has been initialized.