diff --git a/spring-websocket/src/test/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecoratorTests.java b/spring-websocket/src/test/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecoratorTests.java index 0f06aba3a4..937e1d992a 100644 --- a/spring-websocket/src/test/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecoratorTests.java +++ b/spring-websocket/src/test/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecoratorTests.java @@ -203,6 +203,8 @@ public class ConcurrentWebSocketSessionDecoratorTests { } private void sendBlockingMessage(ConcurrentWebSocketSessionDecorator session) throws InterruptedException { + BlockingSession delegate = (BlockingSession) session.getDelegate(); + CountDownLatch sentMessageLatch = delegate.getSentMessageLatch(); Executors.newSingleThreadExecutor().submit(() -> { TextMessage message = new TextMessage("slow message"); try { @@ -212,17 +214,16 @@ public class ConcurrentWebSocketSessionDecoratorTests { e.printStackTrace(); } }); - BlockingSession delegate = (BlockingSession) session.getDelegate(); - assertThat(delegate.getSentMessageLatch().await(5, TimeUnit.SECONDS)).isTrue(); + assertThat(sentMessageLatch.await(5, TimeUnit.SECONDS)).isTrue(); } private static class BlockingSession extends TestWebSocketSession { - private AtomicReference nextMessageLatch = new AtomicReference<>(); + private final AtomicReference nextMessageLatch = new AtomicReference<>(); - private AtomicReference releaseLatch = new AtomicReference<>(); + private final AtomicReference releaseLatch = new AtomicReference<>(); public CountDownLatch getSentMessageLatch() {