diff options
Diffstat (limited to 'juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java')
-rw-r--r-- | juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java b/juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java new file mode 100644 index 00000000..7d896a04 --- /dev/null +++ b/juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java @@ -0,0 +1,18 @@ +package com.juick.server.tests; + +import org.springframework.web.socket.WebSocketSession; +import org.springframework.web.socket.handler.TextWebSocketHandler; + +import java.util.concurrent.CountDownLatch; + +class WebsocketClientHandler extends TextWebSocketHandler { + private final CountDownLatch latch; + public WebsocketClientHandler(CountDownLatch latch) { + this.latch = latch; + } + @Override + public void afterConnectionEstablished(WebSocketSession session) throws Exception { + super.afterConnectionEstablished(session); + latch.countDown(); + } +} |