From 7a810e890e883aa935698182728265023080e80f Mon Sep 17 00:00:00 2001
From: Prudhvi Godithi <pgodithi@amazon.com>
Date: Thu, 25 Jan 2024 14:42:58 -0800
Subject: [PATCH] Fix flaky tests

Signed-off-by: Prudhvi Godithi <pgodithi@amazon.com>
---
 spi/build.gradle | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/spi/build.gradle b/spi/build.gradle
index b75797f2..e83781ee 100644
--- a/spi/build.gradle
+++ b/spi/build.gradle
@@ -11,6 +11,7 @@ plugins {
     id 'jacoco'
     id 'maven-publish'
     id 'signing'
+    id "org.gradle.test-retry" version "1.5.8"
 }
 
 apply plugin: 'opensearch.java'
@@ -70,6 +71,10 @@ shadowJar {
 }
 
 test {
+    retry {
+        failOnPassedAfterRetry = false
+        maxRetries = 5
+    }
     doFirst {
         // reverse operation of https://github.com/elastic/elasticsearch/blob/7.6/buildSrc/src/main/groovy/org/elasticsearch/gradle/BuildPlugin.groovy#L736-L743
         // to fix the classpath for unit tests