File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
spring-websocket/src/test/java/org/springframework/web/socket/messaging Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 46
46
import org .springframework .web .socket .WebSocketHandler ;
47
47
import org .springframework .web .socket .WebSocketSession ;
48
48
import org .springframework .web .socket .client .WebSocketClient ;
49
+ import org .springframework .web .socket .handler .WebSocketHandlerDecorator ;
49
50
50
51
import static org .assertj .core .api .Assertions .assertThat ;
51
52
import static org .assertj .core .api .Assertions .assertThatIllegalStateException ;
@@ -319,9 +320,12 @@ private WebSocketHandler connect() {
319
320
320
321
@ SuppressWarnings ("unchecked" )
321
322
private TcpConnection <byte []> getTcpConnection () throws Exception {
322
- WebSocketHandler webSocketHandler = connect ();
323
- webSocketHandler .afterConnectionEstablished (this .webSocketSession );
324
- return (TcpConnection <byte []>) webSocketHandler ;
323
+ WebSocketHandler handler = connect ();
324
+ handler .afterConnectionEstablished (this .webSocketSession );
325
+ if (handler instanceof WebSocketHandlerDecorator ) {
326
+ handler = ((WebSocketHandlerDecorator ) handler ).getLastHandler ();
327
+ }
328
+ return (TcpConnection <byte []>) handler ;
325
329
}
326
330
327
331
private void testInactivityTaskScheduling (Runnable runnable , long delay , long sleepTime )
You can’t perform that action at this time.
0 commit comments