Websocket Test: wait until message becomes available

jknack · jknack · commit 601fe28e7a01 · 2019-11-21T23:59:29.000-03:00
diff --git a/tests/src/test/java/io/jooby/WebClient.java b/tests/src/test/java/io/jooby/WebClient.java
@@ -58,7 +58,7 @@ private class SyncWebSocketListener extends WebSocketListener {
 
     public String lastMessage() {
       try {
-        return  (String) messages.poll(5, TimeUnit.SECONDS);
+        return  (String) messages.take();
       } catch (Exception x) {
         throw SneakyThrows.propagate(x);
       }

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ private class SyncWebSocketListener extends WebSocketListener {`
`58`	`58`
`59`	`59`	`public String lastMessage() {`
`60`	`60`	`try {`
`61`		`- return (String) messages.poll(5, TimeUnit.SECONDS);`
	`61`	`+ return (String) messages.take();`
`62`	`62`	`} catch (Exception x) {`
`63`	`63`	`throw SneakyThrows.propagate(x);`
`64`	`64`	`}`