Improving test

kiegroup · Sep 19, 2024 · cb298d0 · cb298d0
1 parent 0294ac2
commit cb298d0
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/...java/org/uberfire/ext/metadata/backend/infinispan/provider/InfinispanPingServiceTest.java b/...java/org/uberfire/ext/metadata/backend/infinispan/provider/InfinispanPingServiceTest.java
@@ -25,35 +25,38 @@
 import org.mockito.Mock;
 import org.mockito.junit.MockitoJUnitRunner;
 
-import java.util.concurrent.ExecutionException;
+import java.util.concurrent.CompletableFuture;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
-import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.when;
+import static org.mockito.Mockito.*;
 
 @RunWith(MockitoJUnitRunner.class)
 public class InfinispanPingServiceTest {
 
     @Mock(answer = Answers.RETURNS_DEEP_STUBS)
-    private RemoteCacheImpl remoteCache;
+    private RemoteCacheImpl<?,?> remoteCache;
+
+    @Mock
+    PingResponse pingResponse;
 
     @Before
-    public void setUp() throws ExecutionException, InterruptedException {
-        when(((PingResponse) remoteCache.ping().toCompletableFuture().get()).isSuccess()).thenReturn(true);
+    public void setUp() {
+        when(pingResponse.isSuccess()).thenReturn(true);
+        when(remoteCache.ping().toCompletableFuture()).thenReturn(CompletableFuture.completedFuture(pingResponse));
     }
 
     @Test
-    public void testPingSuccess() throws ExecutionException, InterruptedException {
+    public void testPingSuccess() {
         {
             InfinispanPingService service = spy(new InfinispanPingService(remoteCache));
             assertTrue(service.ping());
             service.stop();
         }
 
         {
-            when(((PingResponse) remoteCache.ping().toCompletableFuture().get()).isSuccess()).thenReturn(false);
+            when(pingResponse.isSuccess()).thenReturn(false);
             InfinispanPingService service = spy(new InfinispanPingService(remoteCache));
             assertFalse(service.ping());
             service.stop();