diff options
author | Vitaly Takmazov | 2018-05-27 16:45:59 +0300 |
---|---|---|
committer | Vitaly Takmazov | 2018-05-27 16:50:16 +0300 |
commit | 15c52d91884988dc3c714739f901b510eb6ffc56 (patch) | |
tree | 8e9c690c7c6d227fb6bd80d6cf061f0c58d0c9f9 /juick-server/src/test/java/com/juick/server/tests/ServerTests.java | |
parent | 7f181f590fbc7b2062a5348a3b050d7cf884a135 (diff) |
ws: again, accept commands
Diffstat (limited to 'juick-server/src/test/java/com/juick/server/tests/ServerTests.java')
-rw-r--r-- | juick-server/src/test/java/com/juick/server/tests/ServerTests.java | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/juick-server/src/test/java/com/juick/server/tests/ServerTests.java b/juick-server/src/test/java/com/juick/server/tests/ServerTests.java index 8e3e5dfd..9a78eac9 100644 --- a/juick-server/src/test/java/com/juick/server/tests/ServerTests.java +++ b/juick-server/src/test/java/com/juick/server/tests/ServerTests.java @@ -57,6 +57,9 @@ import org.springframework.util.LinkedMultiValueMap; import org.springframework.util.MultiValueMap; import org.springframework.web.client.RestTemplate; import org.springframework.web.context.WebApplicationContext; +import org.springframework.web.socket.TextMessage; +import org.springframework.web.socket.client.WebSocketConnectionManager; +import org.springframework.web.socket.client.standard.StandardWebSocketClient; import org.springframework.web.util.UriComponents; import org.springframework.web.util.UriComponentsBuilder; import org.w3c.dom.Document; @@ -841,6 +844,8 @@ public class ServerTests { WebsocketClientConfig.class, PropertyPlaceholderAutoConfiguration.class) .run("--spring.main.web_environment=false"); long count = context.getBean(WebsocketClientConfig.class).getLatch().getCount(); + WebsocketClientHandler handler = context.getBean(WebsocketClientHandler.class); + assertThat(handler.getLastMessage(), is("PONG")); context.close(); Assert.assertThat(count, equalTo(0L)); mockMvc.perform(get("/ws/_all")).andExpect(status().isBadRequest()); |