Fix test

lightsail-network · Apr 28, 2023 · b04dd35 · b04dd35
1 parent d8415f1
commit b04dd35
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/test/java/org/stellar/sdk/requests/ResponseHandlerTest.java b/src/test/java/org/stellar/sdk/requests/ResponseHandlerTest.java
@@ -3,6 +3,7 @@
 import static org.junit.Assert.assertEquals;
 
 import java.io.IOException;
+import java.util.Optional;
 
 import org.junit.Assert;
 import org.junit.Test;
@@ -30,7 +31,7 @@ public void testTooManyRequests() throws IOException, InterruptedException {
             AccountsRequestBuilder.execute(okHttpClient, mockWebServer.url("/"));
             Assert.fail();
         } catch (TooManyRequestsException tmre) {
-            assertEquals(10, tmre.getRetryAfter());
+            assertEquals(Optional.of(10), tmre.getRetryAfter());
         } finally {
 
             mockWebServer.shutdown();
@@ -55,7 +56,7 @@ public void testTooManyRequestsNoHeader() throws IOException, InterruptedExcepti
             AccountsRequestBuilder.execute(okHttpClient, mockWebServer.url("/"));
             Assert.fail();
         } catch (TooManyRequestsException tmre) {
-            assertEquals(-1, tmre.getRetryAfter());
+            assertEquals(Optional.empty(), tmre.getRetryAfter());
         } finally {
 
             mockWebServer.shutdown();