From 6b8f4a5e11296e3ff6f0b426ede8f1d2a4ac8b09 Mon Sep 17 00:00:00 2001
From: David Ho <hdavidh@amazon.com>
Date: Tue, 19 May 2026 15:11:49 -0700
Subject: [PATCH] Fix Sts flakey test

---
 .../awssdk/services/sts/internal/StsRetryStrategyTest.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/services/sts/src/test/java/software/amazon/awssdk/services/sts/internal/StsRetryStrategyTest.java b/services/sts/src/test/java/software/amazon/awssdk/services/sts/internal/StsRetryStrategyTest.java
index 8535205bf93..4ce965f1b61 100644
--- a/services/sts/src/test/java/software/amazon/awssdk/services/sts/internal/StsRetryStrategyTest.java
+++ b/services/sts/src/test/java/software/amazon/awssdk/services/sts/internal/StsRetryStrategyTest.java
@@ -18,7 +18,6 @@
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.Mockito.mock;
 
-import java.time.Duration;
 import java.util.stream.Stream;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
@@ -98,7 +97,8 @@ void assertRetriesOnIdpCommunicationException(RetryStrategy strategy) {
                                                                    .token(token)
                                                                    .failure(failure)
                                                                    .build();
-        assertThat(strategy.refreshRetryToken(refresh).delay()).isGreaterThan(Duration.ZERO);
+        // Should not throw TokenAcquisitionFailedException
+        assertThat(strategy.refreshRetryToken(refresh)).isNotNull();
     }
 
     private static Stream<Arguments> retryModeResolutionCases() {