Search in sources :

Example 1 with TestHttpSockJsSession

use of org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession in project spring-framework by spring-projects.

the class HttpReceivingTransportHandlerTests method handleRequestAndExpectFailure.

private void handleRequestAndExpectFailure() throws Exception {
    resetResponse();
    WebSocketHandler wsHandler = mock(WebSocketHandler.class);
    AbstractSockJsSession session = new TestHttpSockJsSession("1", new StubSockJsServiceConfig(), wsHandler, null);
    new XhrReceivingTransportHandler().handleRequest(this.request, this.response, wsHandler, session);
    assertEquals(500, this.servletResponse.getStatus());
    verifyNoMoreInteractions(wsHandler);
}
Also used : TestHttpSockJsSession(org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession) WebSocketHandler(org.springframework.web.socket.WebSocketHandler) AbstractSockJsSession(org.springframework.web.socket.sockjs.transport.session.AbstractSockJsSession) StubSockJsServiceConfig(org.springframework.web.socket.sockjs.transport.session.StubSockJsServiceConfig)

Example 2 with TestHttpSockJsSession

use of org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession in project spring-framework by spring-projects.

the class HttpReceivingTransportHandlerTests method handleRequest.

private void handleRequest(AbstractHttpReceivingTransportHandler transportHandler) throws Exception {
    WebSocketHandler wsHandler = mock(WebSocketHandler.class);
    AbstractSockJsSession session = new TestHttpSockJsSession("1", new StubSockJsServiceConfig(), wsHandler, null);
    transportHandler.initialize(new StubSockJsServiceConfig());
    transportHandler.handleRequest(this.request, this.response, wsHandler, session);
    assertEquals("text/plain;charset=UTF-8", this.response.getHeaders().getContentType().toString());
    verify(wsHandler).handleMessage(session, new TextMessage("x"));
}
Also used : TestHttpSockJsSession(org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession) WebSocketHandler(org.springframework.web.socket.WebSocketHandler) AbstractSockJsSession(org.springframework.web.socket.sockjs.transport.session.AbstractSockJsSession) StubSockJsServiceConfig(org.springframework.web.socket.sockjs.transport.session.StubSockJsServiceConfig) TextMessage(org.springframework.web.socket.TextMessage)

Example 3 with TestHttpSockJsSession

use of org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession in project spring-framework by spring-projects.

the class HttpReceivingTransportHandlerTests method delegateMessageException.

@Test
public void delegateMessageException() throws Exception {
    StubSockJsServiceConfig sockJsConfig = new StubSockJsServiceConfig();
    this.servletRequest.setContent("[\"x\"]".getBytes("UTF-8"));
    WebSocketHandler wsHandler = mock(WebSocketHandler.class);
    TestHttpSockJsSession session = new TestHttpSockJsSession("1", sockJsConfig, wsHandler, null);
    session.delegateConnectionEstablished();
    willThrow(new Exception()).given(wsHandler).handleMessage(session, new TextMessage("x"));
    try {
        XhrReceivingTransportHandler transportHandler = new XhrReceivingTransportHandler();
        transportHandler.initialize(sockJsConfig);
        transportHandler.handleRequest(this.request, this.response, wsHandler, session);
        fail("Expected exception");
    } catch (SockJsMessageDeliveryException ex) {
        assertNull(session.getCloseStatus());
    }
}
Also used : TestHttpSockJsSession(org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession) WebSocketHandler(org.springframework.web.socket.WebSocketHandler) SockJsMessageDeliveryException(org.springframework.web.socket.sockjs.SockJsMessageDeliveryException) StubSockJsServiceConfig(org.springframework.web.socket.sockjs.transport.session.StubSockJsServiceConfig) SockJsMessageDeliveryException(org.springframework.web.socket.sockjs.SockJsMessageDeliveryException) TextMessage(org.springframework.web.socket.TextMessage) Test(org.junit.Test)

Aggregations

WebSocketHandler (org.springframework.web.socket.WebSocketHandler)3 StubSockJsServiceConfig (org.springframework.web.socket.sockjs.transport.session.StubSockJsServiceConfig)3 TestHttpSockJsSession (org.springframework.web.socket.sockjs.transport.session.TestHttpSockJsSession)3 TextMessage (org.springframework.web.socket.TextMessage)2 AbstractSockJsSession (org.springframework.web.socket.sockjs.transport.session.AbstractSockJsSession)2 Test (org.junit.Test)1 SockJsMessageDeliveryException (org.springframework.web.socket.sockjs.SockJsMessageDeliveryException)1