Add method with configurable timeout

reportportal · Mar 6, 2024 · d1cce9f · d1cce9f
1 parent a1fcea1
commit d1cce9f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/java/com/epam/reportportal/util/test/SocketUtils.java b/src/main/java/com/epam/reportportal/util/test/SocketUtils.java
@@ -96,14 +96,19 @@ public static ServerSocket getServerSocketOnFreePort() throws IOException {
 		return new ServerSocket(0);
 	}
 
-	public static <T> Pair<List<String>, T> executeServerCallable(ServerCallable srvCall, Callable<T> clientCallable) throws Exception {
+	public static <T> Pair<List<String>, T> executeServerCallable(ServerCallable srvCall, Callable<T> clientCallable,
+			long timeoutSeconds) throws Exception {
 		ExecutorService serverExecutor = Executors.newSingleThreadExecutor();
 		Future<List<String>> future = serverExecutor.submit(srvCall);
 		T rs = clientCallable.call();
 		try {
-			return Pair.of(future.get(5, TimeUnit.SECONDS), rs);
+			return Pair.of(future.get(timeoutSeconds, TimeUnit.SECONDS), rs);
 		} finally {
 			CommonUtils.shutdownExecutorService(serverExecutor);
 		}
 	}
+
+	public static <T> Pair<List<String>, T> executeServerCallable(ServerCallable srvCall, Callable<T> clientCallable) throws Exception {
+		return executeServerCallable(srvCall, clientCallable, 5L);
+	}
 }