Examples with WebSocketHandler org.springframework.web.socket.WebSocketHandler used on opensource projects

Example 21 with WebSocketHandler

use of org.springframework.web.socket.WebSocketHandler in project spring-framework by spring-projects.

the class OriginHandshakeInterceptorTests method originMatchAll.

@Test
public void originMatchAll() throws Exception {
    Map<String, Object> attributes = new HashMap<>();
    WebSocketHandler wsHandler = Mockito.mock(WebSocketHandler.class);
    this.servletRequest.addHeader(HttpHeaders.ORIGIN, "http://mydomain1.com");
    OriginHandshakeInterceptor interceptor = new OriginHandshakeInterceptor();
    interceptor.setAllowedOrigins(Collections.singletonList("*"));
    assertTrue(interceptor.beforeHandshake(request, response, wsHandler, attributes));
    assertNotEquals(servletResponse.getStatus(), HttpStatus.FORBIDDEN.value());
}

Example 22 with WebSocketHandler

use of org.springframework.web.socket.WebSocketHandler in project spring-framework by spring-projects.

the class OriginHandshakeInterceptorTests method originListMatch.

@Test
public void originListMatch() throws Exception {
    Map<String, Object> attributes = new HashMap<>();
    WebSocketHandler wsHandler = Mockito.mock(WebSocketHandler.class);
    this.servletRequest.addHeader(HttpHeaders.ORIGIN, "http://mydomain2.com");
    List<String> allowed = Arrays.asList("http://mydomain1.com", "http://mydomain2.com", "http://mydomain3.com");
    OriginHandshakeInterceptor interceptor = new OriginHandshakeInterceptor(allowed);
    assertTrue(interceptor.beforeHandshake(request, response, wsHandler, attributes));
    assertNotEquals(servletResponse.getStatus(), HttpStatus.FORBIDDEN.value());
}

Example 23 with WebSocketHandler

use of org.springframework.web.socket.WebSocketHandler in project spring-framework by spring-projects.

the class OriginHandshakeInterceptorTests method originNoMatchWithNullHostileCollection.

@Test
public void originNoMatchWithNullHostileCollection() throws Exception {
    Map<String, Object> attributes = new HashMap<>();
    WebSocketHandler wsHandler = Mockito.mock(WebSocketHandler.class);
    this.servletRequest.addHeader(HttpHeaders.ORIGIN, "http://mydomain4.com");
    OriginHandshakeInterceptor interceptor = new OriginHandshakeInterceptor();
    Set<String> allowedOrigins = new ConcurrentSkipListSet<>();
    allowedOrigins.add("http://mydomain1.com");
    interceptor.setAllowedOrigins(allowedOrigins);
    assertFalse(interceptor.beforeHandshake(request, response, wsHandler, attributes));
    assertEquals(servletResponse.getStatus(), HttpStatus.FORBIDDEN.value());
}

Example 24 with WebSocketHandler

use of org.springframework.web.socket.WebSocketHandler in project spring-framework by spring-projects.

the class OriginHandshakeInterceptorTests method sameOriginNoMatch.

@Test
public void sameOriginNoMatch() throws Exception {
    Map<String, Object> attributes = new HashMap<>();
    WebSocketHandler wsHandler = Mockito.mock(WebSocketHandler.class);
    this.servletRequest.addHeader(HttpHeaders.ORIGIN, "http://mydomain3.com");
    this.servletRequest.setServerName("mydomain2.com");
    OriginHandshakeInterceptor interceptor = new OriginHandshakeInterceptor(Collections.emptyList());
    assertFalse(interceptor.beforeHandshake(request, response, wsHandler, attributes));
    assertEquals(servletResponse.getStatus(), HttpStatus.FORBIDDEN.value());
}

Example 25 with WebSocketHandler

use of org.springframework.web.socket.WebSocketHandler in project spring-framework by spring-projects.

the class WebSocketStompClientTests method getTcpConnection.

@SuppressWarnings("unchecked")
private TcpConnection<byte[]> getTcpConnection() throws Exception {
    WebSocketHandler webSocketHandler = connect();
    webSocketHandler.afterConnectionEstablished(this.webSocketSession);
    return (TcpConnection<byte[]>) webSocketHandler;
}