From f1521c2efbd1966f18859b815a0f71d7e7969086 Mon Sep 17 00:00:00 2001
From: Michael Burman <michael.burman@datastax.com>
Date: Mon, 21 Aug 2023 15:22:40 +0300
Subject: [PATCH] More blocking

---
 .../java/com/datastax/mgmtapi/helpers/NettyHttpClient.java   | 5 ++++-
 1 file changed, 4 insertions(+), 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
index 94ecedc0..5657437c 100644
--- 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;
   }