Move RetryUtilsTest to langchain4j-core with RetryUtils. (langchain4j…

…#478)
hwy110 · Jan 9, 2024 · 4a7d14c · 4a7d14c
1 parent 23b676f
commit 4a7d14c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.../langchain4j/internal/RetryUtilsTest.java → .../langchain4j/internal/RetryUtilsTest.java b/.../langchain4j/internal/RetryUtilsTest.java → .../langchain4j/internal/RetryUtilsTest.java
@@ -13,6 +13,7 @@ class RetryUtilsTest {
 
     @Test
     void testSuccessfulCall() throws Exception {
+        @SuppressWarnings("unchecked")
         Callable<String> mockAction = mock(Callable.class);
         when(mockAction.call()).thenReturn("Success");
 
@@ -25,6 +26,7 @@ void testSuccessfulCall() throws Exception {
 
     @Test
     void testRetryThenSuccess() throws Exception {
+        @SuppressWarnings("unchecked")
         Callable<String> mockAction = mock(Callable.class);
         when(mockAction.call())
                 .thenThrow(new RuntimeException())
@@ -46,6 +48,7 @@ void testRetryThenSuccess() throws Exception {
 
     @Test
     void testMaxAttemptsReached() throws Exception {
+        @SuppressWarnings("unchecked")
         Callable<String> mockAction = mock(Callable.class);
         when(mockAction.call()).thenThrow(new RuntimeException());