diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/util/VirtualThreadPinEventMonitorTest.java b/service/src/test/java/org/whispersystems/textsecuregcm/util/VirtualThreadPinEventMonitorTest.java index 4c93704eb..209a9d5b5 100644 --- a/service/src/test/java/org/whispersystems/textsecuregcm/util/VirtualThreadPinEventMonitorTest.java +++ b/service/src/test/java/org/whispersystems/textsecuregcm/util/VirtualThreadPinEventMonitorTest.java @@ -53,8 +53,9 @@ public class VirtualThreadPinEventMonitorTest { exec.submit(() -> synchronizedSleep1()).get(); eventMonitor.stop(); - final Pair nxt = bq.take(); - assertThat(nxt.getRight()).isFalse(); + final Pair event = bq.poll(1, TimeUnit.SECONDS); + assertThat(event).isNotNull(); + assertThat(event.getRight()).isFalse(); assertThat(bq.isEmpty()); exec.shutdown(); exec.awaitTermination(1, TimeUnit.MILLISECONDS); @@ -73,10 +74,12 @@ public class VirtualThreadPinEventMonitorTest { exec.submit(() -> synchronizedSleep2()).get(); eventMonitor.stop(); - final boolean sleep1Allowed = bq.take().getRight(); - final boolean sleep2Allowed = bq.take().getRight(); - assertThat(sleep1Allowed).isTrue(); - assertThat(sleep2Allowed).isFalse(); + final Pair sleep1Event = bq.poll(1, TimeUnit.SECONDS); + final Pair sleep2Event = bq.poll(1, TimeUnit.SECONDS); + assertThat(sleep1Event).isNotNull(); + assertThat(sleep2Event).isNotNull(); + assertThat(sleep1Event.getRight()).isTrue(); + assertThat(sleep2Event.getRight()).isFalse(); assertThat(bq.isEmpty()); exec.shutdown(); exec.awaitTermination(1, TimeUnit.MILLISECONDS);