How to use WebSocketServerEndpointConfiguration class of com.consol.citrus.websocket.server package

Best Citrus code snippet using com.consol.citrus.websocket.server.WebSocketServerEndpointConfiguration

Source:WebSocketEndpointTest.java

...19import com.consol.citrus.message.Message;20import com.consol.citrus.testng.AbstractTestNGUnitTest;21import com.consol.citrus.websocket.handler.CitrusWebSocketHandler;22import com.consol.citrus.websocket.message.WebSocketMessage;23import com.consol.citrus.websocket.server.WebSocketServerEndpointConfiguration;24import org.mockito.Mockito;25import org.mockito.invocation.InvocationOnMock;26import org.mockito.stubbing.Answer;27import org.springframework.web.socket.*;28import org.testng.Assert;29import org.testng.annotations.Test;30import static org.mockito.Mockito.*;31public class WebSocketEndpointTest extends AbstractTestNGUnitTest {32    private WebSocketSession session = Mockito.mock(WebSocketSession.class);33    private WebSocketSession session2 = Mockito.mock(WebSocketSession.class);34    private WebSocketSession session3 = Mockito.mock(WebSocketSession.class);35    @Test36    public void testWebSocketEndpoint() throws Exception {37        WebSocketServerEndpointConfiguration endpointConfiguration = new WebSocketServerEndpointConfiguration();38        WebSocketEndpoint webSocketEndpoint = new WebSocketEndpoint(endpointConfiguration);39        String endpointUri = "/test";40        CitrusWebSocketHandler handler = new CitrusWebSocketHandler();41        endpointConfiguration.setHandler(handler);42        final String requestBody = "<TestRequest><Message>Hello World!</Message></TestRequest>";43        final Message responseMessage = new DefaultMessage("<TestResponse><Message>Hello World!</Message></TestResponse>");44        endpointConfiguration.setEndpointUri(endpointUri);45        reset(session);46        when(session.getId()).thenReturn("test-socket-1");47        when(session.isOpen()).thenReturn(true);48        doAnswer(new Answer() {49            @Override50            public Object answer(InvocationOnMock invocation) throws Throwable {51                org.springframework.web.socket.WebSocketMessage request = (org.springframework.web.socket.WebSocketMessage) invocation.getArguments()[0];52                Assert.assertTrue(TextMessage.class.isInstance(request));53                Assert.assertEquals(((TextMessage)request).getPayload(), responseMessage.getPayload(String.class));54                Assert.assertTrue(request.isLast());55                return null;56            }57        }).when(session).sendMessage(any(org.springframework.web.socket.WebSocketMessage.class));58        handler.afterConnectionEstablished(session);59        handler.handleMessage(session, new TextMessage(requestBody));60        WebSocketMessage requestMessage = (WebSocketMessage) webSocketEndpoint.createConsumer().receive(context);61        Assert.assertEquals(requestMessage.getPayload(), requestBody);62        Assert.assertTrue(requestMessage.isLast());63        webSocketEndpoint.createProducer().send(responseMessage, context);64    }65    @Test66    public void testWebSocketEndpointMultipleSessions() throws Exception {67        WebSocketServerEndpointConfiguration endpointConfiguration = new WebSocketServerEndpointConfiguration();68        WebSocketEndpoint webSocketEndpoint = new WebSocketEndpoint(endpointConfiguration);69        String endpointUri = "/test";70        CitrusWebSocketHandler handler = new CitrusWebSocketHandler();71        endpointConfiguration.setHandler(handler);72        final String requestBody = "<TestRequest><Message>Hello World!</Message></TestRequest>";73        final Message responseMessage = new DefaultMessage("<TestResponse><Message>Hello World!</Message></TestResponse>");74        endpointConfiguration.setEndpointUri(endpointUri);75        reset(session, session2, session3);76        when(session.getId()).thenReturn("test-socket-1");77        when(session2.getId()).thenReturn("test-socket-2");78        when(session3.getId()).thenReturn("test-socket-3");79        when(session.isOpen()).thenReturn(true);80        when(session2.isOpen()).thenReturn(true);81        doAnswer(new Answer() {82            @Override83            public Object answer(InvocationOnMock invocation) throws Throwable {84                org.springframework.web.socket.WebSocketMessage request = (org.springframework.web.socket.WebSocketMessage) invocation.getArguments()[0];85                Assert.assertTrue(TextMessage.class.isInstance(request));86                Assert.assertEquals(((TextMessage)request).getPayload(), responseMessage.getPayload(String.class));87                Assert.assertTrue(request.isLast());88                return null;89            }90        }).when(session).sendMessage(any(org.springframework.web.socket.WebSocketMessage.class));91        doAnswer(new Answer() {92            @Override93            public Object answer(InvocationOnMock invocation) throws Throwable {94                org.springframework.web.socket.WebSocketMessage request = (org.springframework.web.socket.WebSocketMessage) invocation.getArguments()[0];95                Assert.assertTrue(TextMessage.class.isInstance(request));96                Assert.assertEquals(((TextMessage)request).getPayload(), responseMessage.getPayload(String.class));97                Assert.assertTrue(request.isLast());98                return null;99            }100        }).when(session2).sendMessage(any(org.springframework.web.socket.WebSocketMessage.class));101        handler.afterConnectionEstablished(session);102        handler.afterConnectionEstablished(session2);103        handler.afterConnectionEstablished(session3);104        handler.afterConnectionClosed(session3, CloseStatus.NORMAL);105        handler.handleMessage(session, new TextMessage(requestBody));106        WebSocketMessage requestMessage = (WebSocketMessage) webSocketEndpoint.createConsumer().receive(context);107        Assert.assertEquals(requestMessage.getPayload(), requestBody);108        Assert.assertTrue(requestMessage.isLast());109        webSocketEndpoint.createProducer().send(responseMessage, context);110    }111    @Test112    public void testWebSocketEndpointTimeout() throws Exception {113        WebSocketServerEndpointConfiguration endpointConfiguration = new WebSocketServerEndpointConfiguration();114        WebSocketEndpoint webSocketEndpoint = new WebSocketEndpoint(endpointConfiguration);115        String endpointUri = "/test";116        CitrusWebSocketHandler handler = new CitrusWebSocketHandler();117        endpointConfiguration.setHandler(handler);118        endpointConfiguration.setEndpointUri(endpointUri);119        endpointConfiguration.setTimeout(1000L);120        reset(session);121        when(session.getId()).thenReturn("test-socket-1");122        handler.afterConnectionEstablished(session);123        try {124            webSocketEndpoint.createConsumer().receive(context, endpointConfiguration.getTimeout());125            Assert.fail("Missing timeout exception on web socket endpoint");126        } catch (ActionTimeoutException e) {127            Assert.assertTrue(e.getMessage().contains(endpointUri));...

Source:WebSocketEndpointParser.java

...18import com.consol.citrus.config.xml.AbstractEndpointParser;19import com.consol.citrus.endpoint.Endpoint;20import com.consol.citrus.endpoint.EndpointConfiguration;21import com.consol.citrus.websocket.endpoint.WebSocketEndpoint;22import com.consol.citrus.websocket.server.WebSocketServerEndpointConfiguration;23import org.springframework.beans.factory.support.BeanDefinitionBuilder;24import org.springframework.beans.factory.xml.ParserContext;25import org.w3c.dom.Element;26/**27 * Used for parsing server WebSocket endpoints and respective configurations.28 *29 * @author Martin Maher30 * @since 2.331 */32public class WebSocketEndpointParser extends AbstractEndpointParser {33    @Override34    protected void parseEndpointConfiguration(BeanDefinitionBuilder endpointConfiguration, Element element, ParserContext parserContext) {35        super.parseEndpointConfiguration(endpointConfiguration, element, parserContext);36        BeanDefinitionParserUtils.setPropertyValue(endpointConfiguration, element.getAttribute("path"), "endpointUri");37        BeanDefinitionParserUtils.setPropertyReference(endpointConfiguration, element.getAttribute("message-converter"), "messageConverter");38    }39    @Override40    protected Class<? extends Endpoint> getEndpointClass() {41        return WebSocketEndpoint.class;42    }43    @Override44    protected Class<? extends EndpointConfiguration> getEndpointConfigurationClass() {45        return WebSocketServerEndpointConfiguration.class;46    }47}...

Source:WebSocketServerEndpointConfiguration.java

...22 * Web socket endpoint configuration for server side web socket communication.23 * @author Martin Maher24 * @since 2.325 */26public class WebSocketServerEndpointConfiguration extends AbstractWebSocketEndpointConfiguration {27    /** Logger */28    private static final Logger LOG = LoggerFactory.getLogger(WebSocketServerEndpointConfiguration.class);29    /** Web socket handler */30    private CitrusWebSocketHandler handler;31    @Override32    public CitrusWebSocketHandler getHandler() {33        return handler;34    }35    @Override36    public void setHandler(CitrusWebSocketHandler handler) {37        if (this.handler != null) {38            LOG.warn(String.format("Handler already set for Web Socket endpoint (path='%s'). Check configuration to ensure that the Web Socket endpoint is not being used by multiple http-servers", getEndpointUri()));39        }40        this.handler = handler;41    }42}...