From 15c52d91884988dc3c714739f901b510eb6ffc56 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Sun, 27 May 2018 16:45:59 +0300
Subject: ws: again, accept commands

---
 juick-server/src/test/java/com/juick/server/tests/ServerTests.java | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'juick-server/src/test/java/com/juick/server/tests/ServerTests.java')

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());
-- 
cgit v1.2.3