Fix failing test

This commit is contained in:
Rossen Stoyanchev
2016-12-19 15:29:00 -05:00
parent 1243556047
commit 5fd600d2ad

View File

@@ -29,6 +29,7 @@ import org.springframework.context.annotation.Configuration;
import org.springframework.web.reactive.HandlerMapping; import org.springframework.web.reactive.HandlerMapping;
import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping; import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping;
import org.springframework.web.reactive.socket.WebSocketHandler; import org.springframework.web.reactive.socket.WebSocketHandler;
import org.springframework.web.reactive.socket.WebSocketMessage;
import org.springframework.web.reactive.socket.WebSocketSession; import org.springframework.web.reactive.socket.WebSocketSession;
import org.springframework.web.reactive.socket.client.ReactorNettyWebSocketClient; import org.springframework.web.reactive.socket.client.ReactorNettyWebSocketClient;
import org.springframework.web.reactive.socket.client.RxNettyWebSocketClient; import org.springframework.web.reactive.socket.client.RxNettyWebSocketClient;
@@ -68,13 +69,7 @@ public class WebSocketIntegrationTests extends AbstractWebSocketIntegrationTests
client.execute(getUrl("/echo"), client.execute(getUrl("/echo"),
session -> session session -> session
.send(input.map(session::textMessage)) .send(input.map(session::textMessage))
.thenMany(session.receive() .thenMany(session.receive().take(count).map(WebSocketMessage::getPayloadAsText))
.take(count)
.map(message -> {
String text = message.getPayloadAsText();
message.release();
return text;
}))
.subscribeWith(output) .subscribeWith(output)
.then()) .then())
.blockMillis(5000); .blockMillis(5000);