diff options
author | 2018-05-27 16:45:59 +0300 | |
---|---|---|
committer | 2018-05-27 16:50:16 +0300 | |
commit | 15c52d91884988dc3c714739f901b510eb6ffc56 (patch) | |
tree | 8e9c690c7c6d227fb6bd80d6cf061f0c58d0c9f9 /juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java | |
parent | 7f181f590fbc7b2062a5348a3b050d7cf884a135 (diff) |
ws: again, accept commands
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 | 22 |
1 files changed, 20 insertions, 2 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 index 7d896a04..ae92ccf0 100644 --- a/juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java +++ b/juick-server/src/test/java/com/juick/server/tests/WebsocketClientHandler.java @@ -1,18 +1,36 @@ package com.juick.server.tests; +import org.springframework.web.socket.TextMessage; import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.handler.TextWebSocketHandler; +import java.io.IOException; import java.util.concurrent.CountDownLatch; class WebsocketClientHandler extends TextWebSocketHandler { private final CountDownLatch latch; + private WebSocketSession session; + private String lastMessage; public WebsocketClientHandler(CountDownLatch latch) { this.latch = latch; } @Override - public void afterConnectionEstablished(WebSocketSession session) throws Exception { - super.afterConnectionEstablished(session); + public void afterConnectionEstablished(WebSocketSession session) throws IOException { + this.session = session; + session.sendMessage(new TextMessage("PING")); + } + + @Override + protected void handleTextMessage(WebSocketSession session, TextMessage message) { + this.lastMessage = message.getPayload(); latch.countDown(); } + + public WebSocketSession getSession() { + return session; + } + + public String getLastMessage() { + return lastMessage; + } } |