switch from ChronoUnit to TimeUnit

Author: crazyrokr

rlib-common/src/test/java/javasabr/rlib/common/util/AwaitUtilsTest.java

@@ -2,7 +2,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 
-import java.time.temporal.ChronoUnit;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicBoolean;
 import org.junit.jupiter.api.Test;
 
@@ -28,7 +28,7 @@ void shouldAwaitCondition() throws InterruptedException {
 
     // when
     thread.start();
-    boolean result = AwaitUtils.await(500, ChronoUnit.MILLIS, condition::get);
+    boolean result = AwaitUtils.await(500, TimeUnit.MILLISECONDS, condition::get);
 
     // then
     assertThat(result).isTrue();
@@ -40,7 +40,7 @@ void shouldTimeoutIfConditionNotMet() throws InterruptedException {
     var condition = new AtomicBoolean(false);
 
     // when
-    boolean result = AwaitUtils.await(100, ChronoUnit.MILLIS, condition::get);
+    boolean result = AwaitUtils.await(100, TimeUnit.MILLISECONDS, condition::get);
 
     // then
     assertThat(result).isFalse();

rlib-common/src/testFixtures/java/javasabr/rlib/common/util/AwaitUtils.java

@@ -1,6 +1,6 @@
 package javasabr.rlib.common.util;
 
-import java.time.temporal.ChronoUnit;
+import java.util.concurrent.TimeUnit;
 import java.util.function.Supplier;
 
 /**
@@ -19,11 +19,11 @@ public final class AwaitUtils {
    * @return true if the condition was met.
    * @throws InterruptedException if the current thread was interrupted.
    */
-  public static boolean await(long amount, ChronoUnit unit, Supplier<Boolean> condition) throws InterruptedException {
+  public static boolean await(long amount, TimeUnit unit, Supplier<Boolean> condition) throws InterruptedException {
     if (condition.get()) {
       return true;
     }
-    var timeoutMillis = unit.getDuration().toMillis() * amount;
+    var timeoutMillis = unit.toMillis(amount);
     var endTime = System.currentTimeMillis() + timeoutMillis;
     while (System.currentTimeMillis() < endTime) {
       if (condition.get()) {

rlib-network/src/test/java/javasabr/rlib/network/ConnectionCloseTest.java

@@ -4,7 +4,6 @@
 
 import java.io.InputStream;
 import java.net.InetSocketAddress;
-import java.time.temporal.ChronoUnit;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import javasabr.rlib.common.util.AwaitUtils;
@@ -77,11 +76,11 @@ void shouldCloseServerConnectionWhenClientClosesTcpChannelAbruptly() {
 
       // when
       clientConnection.channel().close();
-      assertThat(AwaitUtils.await(5000, ChronoUnit.MILLIS, clientConnection::closed))
+      assertThat(AwaitUtils.await(5, TimeUnit.SECONDS, clientConnection::closed))
           .as("Client connection should be closed prior server side verification").isTrue();
 
       // then
-      assertThat(AwaitUtils.await(5000, ChronoUnit.MILLIS, serverConnection::closed))
+      assertThat(AwaitUtils.await(5, TimeUnit.SECONDS, serverConnection::closed))
           .as("Server connection should be closed after receiving EOF from abruptly closed client channel")
           .isTrue();
     }