Corrected InMemoryRetryingActivityHandler reference

finos · May 2, 2023 · 7e21035 · 7e21035
1 parent 784b2c7
commit 7e21035
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/...st/java/org/finos/springbot/teams/handlers/retry/InMemoryRetryingActivityHandlerTest.java b/...st/java/org/finos/springbot/teams/handlers/retry/InMemoryRetryingActivityHandlerTest.java
@@ -1,7 +1,6 @@
 package org.finos.springbot.teams.handlers.retry;
 
 import java.util.HashSet;
-import java.util.Iterator;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
@@ -115,7 +114,7 @@ public void testWorksFirstTime() throws InterruptedException, ExecutionException
 		passEvery = 1;	// always succeeds
 		long now = System.currentTimeMillis();
 
-		RetryingActivityHandler retry = new RetryingActivityHandler(conv);
+		InMemoryRetryingActivityHandler retry = new InMemoryRetryingActivityHandler(conv);
 
 		CompletableFuture<ResourceResponse> cf = retry.handleActivity(new Activity("dummy"), dummyChat1);
 
@@ -125,8 +124,8 @@ public void testWorksFirstTime() throws InterruptedException, ExecutionException
 
 	@Test
 	public void testMultipleTimes() throws Exception {
-		for (int i = 0; i <10; i++) {
-			RetryingActivityHandler retry = new RetryingActivityHandler(conv);
+		for (int i = 0; i <5; i++) {
+			InMemoryRetryingActivityHandler retry = new InMemoryRetryingActivityHandler(conv);
 			CompletableFuture<ResourceResponse> cf = retry.handleActivity(new Activity("dummy"), dummyChat1);
 			ResourceResponse rr = cf.get();
 		}