root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Ensure WebSocketStompClientTests compiles in Gradle build as well

This commit is contained in:
Sam Brannen 2015-03-17 18:47:24 +01:00
parent ffff596ae6
commit 8ee0e98540
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spring-websocket/src/test/java/org/springframework/web/socket/messaging/WebSocketStompClientTests.java
View File

@ -125,12 +125,12 @@ public class WebSocketStompClientTests {
} }
@Test @Test
@SuppressWarnings("unchecked") @SuppressWarnings({ "unchecked", "rawtypes" })
public void handleWebSocketMessage() throws Exception { public void handleWebSocketMessage() throws Exception {
String text = "SEND\na:alpha\n\nMessage payload\0"; String text = "SEND\na:alpha\n\nMessage payload\0";
connect().handleMessage(this.webSocketSession, new TextMessage(text)); connect().handleMessage(this.webSocketSession, new TextMessage(text));
ArgumentCaptor<? extends Message<byte[]>> captor = (ArgumentCaptor<? extends Message<byte[]>>) ArgumentCaptor.forClass(Message.class); ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
verify(this.stompSession).handleMessage(captor.capture()); verify(this.stompSession).handleMessage(captor.capture());
Message<byte[]> message = captor.getValue(); Message<byte[]> message = captor.getValue();
assertNotNull(message); assertNotNull(message);
@ -143,7 +143,7 @@ public class WebSocketStompClientTests {
} }
@Test @Test
@SuppressWarnings("unchecked") @SuppressWarnings({ "unchecked", "rawtypes" })
public void handleWebSocketMessageSplitAcrossTwoMessage() throws Exception { public void handleWebSocketMessageSplitAcrossTwoMessage() throws Exception {
WebSocketHandler webSocketHandler = connect(); WebSocketHandler webSocketHandler = connect();
@ -155,7 +155,7 @@ public class WebSocketStompClientTests {
String part2 = " payload\0"; String part2 = " payload\0";
webSocketHandler.handleMessage(this.webSocketSession, new TextMessage(part2)); webSocketHandler.handleMessage(this.webSocketSession, new TextMessage(part2));
ArgumentCaptor<? extends Message<byte[]>> captor = (ArgumentCaptor<? extends Message<byte[]>>) ArgumentCaptor.forClass(Message.class); ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
verify(this.stompSession).handleMessage(captor.capture()); verify(this.stompSession).handleMessage(captor.capture());
Message<byte[]> message = captor.getValue(); Message<byte[]> message = captor.getValue();
assertNotNull(message); assertNotNull(message);
@ -168,12 +168,12 @@ public class WebSocketStompClientTests {
} }
@Test @Test
@SuppressWarnings("unchecked") @SuppressWarnings({ "unchecked", "rawtypes" })
public void handleWebSocketMessageBinary() throws Exception { public void handleWebSocketMessageBinary() throws Exception {
String text = "SEND\na:alpha\n\nMessage payload\0"; String text = "SEND\na:alpha\n\nMessage payload\0";
connect().handleMessage(this.webSocketSession, new BinaryMessage(text.getBytes(UTF_8))); connect().handleMessage(this.webSocketSession, new BinaryMessage(text.getBytes(UTF_8)));
ArgumentCaptor<? extends Message<byte[]>> captor = (ArgumentCaptor<? extends Message<byte[]>>) ArgumentCaptor.forClass(Message.class); ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
verify(this.stompSession).handleMessage(captor.capture()); verify(this.stompSession).handleMessage(captor.capture());
Message<byte[]> message = captor.getValue(); Message<byte[]> message = captor.getValue();
assertNotNull(message); assertNotNull(message);