From 7b5012ea6e3796851040a1ba316018d43343682a Mon Sep 17 00:00:00 2001
From: Clint Wylie <cwylie@apache.org>
Date: Tue, 22 Aug 2023 22:01:47 -0700
Subject: [PATCH] override retry attempts for InputEntityIteratingReaderTest
 for much faster test run (#14897)

---
 .../data/input/impl/InputEntityIteratingReaderTest.java  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/processing/src/test/java/org/apache/druid/data/input/impl/InputEntityIteratingReaderTest.java b/processing/src/test/java/org/apache/druid/data/input/impl/InputEntityIteratingReaderTest.java
index 41d191b36b3..9e175b2a3df 100644
--- a/processing/src/test/java/org/apache/druid/data/input/impl/InputEntityIteratingReaderTest.java
+++ b/processing/src/test/java/org/apache/druid/data/input/impl/InputEntityIteratingReaderTest.java
@@ -125,6 +125,15 @@ public class InputEntityIteratingReaderTest extends InitializedNullHandlingTest
         ),
         ImmutableList.of(
             new HttpEntity(new URI("testscheme://some/path"), null, null)
+            {
+              @Override
+              protected int getMaxRetries()
+              {
+                // override so this test does not take like 4 minutes to run
+                return 2;
+              }
+            }
+
         ).iterator(),
         temporaryFolder.newFolder()
     );