From 5fd600d2adeb0c0b4cdab5d3a1ec55ce6a10e849 Mon Sep 17 00:00:00 2001
From: Rossen Stoyanchev <rstoyanchev@pivotal.io>
Date: Mon, 19 Dec 2016 15:29:00 -0500
Subject: [PATCH] Fix failing test

---
 .../socket/server/WebSocketIntegrationTests.java         | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java b/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
index e860f10d31..08ee46096a 100644
--- a/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
+++ b/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
@@ -29,6 +29,7 @@ import org.springframework.context.annotation.Configuration;
 import org.springframework.web.reactive.HandlerMapping;
 import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping;
 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.client.ReactorNettyWebSocketClient;
 import org.springframework.web.reactive.socket.client.RxNettyWebSocketClient;
@@ -68,13 +69,7 @@ public class WebSocketIntegrationTests extends AbstractWebSocketIntegrationTests
 		client.execute(getUrl("/echo"),
 				session -> session
 						.send(input.map(session::textMessage))
-						.thenMany(session.receive()
-								.take(count)
-								.map(message -> {
-									String text = message.getPayloadAsText();
-									message.release();
-									return text;
-								}))
+						.thenMany(session.receive().take(count).map(WebSocketMessage::getPayloadAsText))
 						.subscribeWith(output)
 						.then())
 				.blockMillis(5000);