From 0329184c9499bb874541d6e52bfba296d031413b Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Sun, 18 Feb 2024 18:14:10 -0600
Subject: [PATCH] Fix flaky test in MetricsHttpChannelListenerIntegrationTest

---
 .../MetricsHttpChannelListenerIntegrationTest.java        | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/metrics/MetricsHttpChannelListenerIntegrationTest.java b/service/src/test/java/org/whispersystems/textsecuregcm/metrics/MetricsHttpChannelListenerIntegrationTest.java
index e58a5993a..e78b7998f 100644
--- a/service/src/test/java/org/whispersystems/textsecuregcm/metrics/MetricsHttpChannelListenerIntegrationTest.java
+++ b/service/src/test/java/org/whispersystems/textsecuregcm/metrics/MetricsHttpChannelListenerIntegrationTest.java
@@ -188,14 +188,18 @@ class MetricsHttpChannelListenerIntegrationTest {
               : mock(Counter.class))
           .thenReturn(COUNTER);
 
+      final CompletableFuture<Void> connectionComplete = new CompletableFuture<>();
+
       client.connect(new WebSocketListener() {
                        @Override
                        public void onWebSocketConnect(final Session session) {
                          session.close(1000, "OK");
+                         connectionComplete.complete(null);
                        }
                      },
-              URI.create(String.format("ws://localhost:%d%s", EXTENSION.getLocalPort(), "/v1/websocket")), upgradeRequest)
-          .get(1, TimeUnit.SECONDS);
+          URI.create(String.format("ws://localhost:%d%s", EXTENSION.getLocalPort(), "/v1/websocket")), upgradeRequest);
+
+      connectionComplete.get(1, TimeUnit.SECONDS);
 
       verify(METER_REGISTRY).counter(eq(MetricsHttpChannelListener.REQUEST_COUNTER_NAME), tagCaptor.capture());
       verify(COUNTER).increment();