From 2f805826153be1445cecc90f994c861c9e2501cf Mon Sep 17 00:00:00 2001
From: Greg Wilkins <gregw@webtide.com>
Date: Tue, 17 Aug 2021 14:05:16 +1000
Subject: [PATCH] Fix flaky test from #6562 (#6627)

Fix flaky test from #6562

Signed-off-by: Greg Wilkins <gregw@webtide.com>
---
 .../org/eclipse/jetty/server/HttpOutputTest.java    | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/jetty-server/src/test/java/org/eclipse/jetty/server/HttpOutputTest.java b/jetty-server/src/test/java/org/eclipse/jetty/server/HttpOutputTest.java
index af67693b8e5c..8e3c7e18a318 100644
--- a/jetty-server/src/test/java/org/eclipse/jetty/server/HttpOutputTest.java
+++ b/jetty-server/src/test/java/org/eclipse/jetty/server/HttpOutputTest.java
@@ -23,7 +23,6 @@
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.concurrent.TimeUnit;
-import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
 import javax.servlet.AsyncContext;
 import javax.servlet.ServletException;
@@ -818,7 +817,7 @@ public void handle(String target, Request baseRequest, HttpServletRequest reques
     @Test
     public void testEmptyBuffer() throws Exception
     {
-        AtomicBoolean committed = new AtomicBoolean();
+        FuturePromise<Boolean> committed = new FuturePromise<>();
         AbstractHandler handler = new AbstractHandler()
         {
             @Override
@@ -827,7 +826,7 @@ public void handle(String target, Request baseRequest, HttpServletRequest reques
                 baseRequest.setHandled(true);
                 response.setStatus(200);
                 ((HttpOutput)response.getOutputStream()).write(ByteBuffer.wrap(new byte[0]));
-                committed.set(response.isCommitted());
+                committed.succeeded(response.isCommitted());
             }
         };
 
@@ -835,13 +834,13 @@ public void handle(String target, Request baseRequest, HttpServletRequest reques
         handler.start();
         String response = _connector.getResponse("GET / HTTP/1.0\nHost: localhost:80\n\n");
         assertThat(response, containsString("HTTP/1.1 200 OK"));
-        assertThat(committed.get(), is(false));
+        assertThat(committed.get(10, TimeUnit.SECONDS), is(false));
     }
 
     @Test
     public void testEmptyBufferKnown() throws Exception
     {
-        AtomicBoolean committed = new AtomicBoolean();
+        FuturePromise<Boolean> committed = new FuturePromise<>();
         AbstractHandler handler = new AbstractHandler()
         {
             @Override
@@ -851,7 +850,7 @@ public void handle(String target, Request baseRequest, HttpServletRequest reques
                 response.setStatus(200);
                 response.setContentLength(0);
                 ((HttpOutput)response.getOutputStream()).write(ByteBuffer.wrap(new byte[0]));
-                committed.set(response.isCommitted());
+                committed.succeeded(response.isCommitted());
             }
         };
 
@@ -860,7 +859,7 @@ public void handle(String target, Request baseRequest, HttpServletRequest reques
         String response = _connector.getResponse("GET / HTTP/1.0\nHost: localhost:80\n\n");
         assertThat(response, containsString("HTTP/1.1 200 OK"));
         assertThat(response, containsString("Content-Length: 0"));
-        assertThat(committed.get(), is(true));
+        assertThat(committed.get(10, TimeUnit.SECONDS), is(true));
     }
 
     @Test