Fix tests - one more time.

Signed-off-by: Yury-Fridlyand <[email protected]>
Bit-Quill · Feb 7, 2024 · 4cdfd33 · 4cdfd33
1 parent d08bec0
commit 4cdfd33
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/java/client/src/test/java/glide/connection/ConnectionWithGlideMockTests.java b/java/client/src/test/java/glide/connection/ConnectionWithGlideMockTests.java
@@ -178,8 +178,6 @@ public byte[] handle(byte[] request) {
     public void rethrow_error_if_UDS_channel_closed() {
         var client = new TestClient(channelHandler);
         stopRustCoreLibMock();
-        // If we don't sleep, we will get another error - why?
-        Thread.sleep(200);
         try {
             var exception =
                     assertThrows(

diff --git a/java/client/src/test/java/glide/utils/RustCoreMock.java b/java/client/src/test/java/glide/utils/RustCoreMock.java
@@ -19,6 +19,7 @@
 import io.netty.handler.codec.protobuf.ProtobufVarint32FrameDecoder;
 import io.netty.handler.codec.protobuf.ProtobufVarint32LengthFieldPrepender;
 import java.nio.file.Files;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicBoolean;
 import lombok.SneakyThrows;
 import redis_request.RedisRequestOuterClass.RedisRequest;
@@ -175,10 +176,11 @@ public static String start(GlideMock messageProcessor) {
         return instance.socketPath;
     }
 
+    @SneakyThrows
     public static void stop() {
         if (instance != null) {
             instance.channel.close().syncUninterruptibly();
-            instance.group.shutdownGracefully();
+            instance.group.shutdownGracefully().get(5, TimeUnit.SECONDS);
             instance = null;
         }
     }