More blocking

k8ssandra · Aug 21, 2023 · f1521c2 · f1521c2
1 parent 193cb38
commit f1521c2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/management-api-server/src/test/java/com/datastax/mgmtapi/helpers/NettyHttpClient.java b/management-api-server/src/test/java/com/datastax/mgmtapi/helpers/NettyHttpClient.java
@@ -11,6 +11,7 @@
 import com.google.common.util.concurrent.Uninterruptibles;
 import io.netty.bootstrap.Bootstrap;
 import io.netty.channel.Channel;
+import io.netty.channel.ChannelFuture;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInitializer;
 import io.netty.channel.EventLoopGroup;
@@ -180,7 +181,9 @@ private CompletableFuture<FullHttpResponse> buildAndSendRequest(HttpMethod metho
     request.headers().set(HttpHeaderNames.HOST, url.getHost());
 
     // Send the HTTP request.
-    client.writeAndFlush(request).awaitUninterruptibly();
+    ChannelFuture channelFuture = client.writeAndFlush(request).awaitUninterruptibly();
+    channelFuture.syncUninterruptibly();
+    channelFuture.channel().closeFuture().syncUninterruptibly();
 
     return result;
   }