Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use WebSocketClientEndpointConfiguration class of com.consol.citrus.websocket.client package

Best Citrus code snippet using com.consol.citrus.websocket.client.WebSocketClientEndpointConfiguration

Source:WebSocketClientTest.java

...17import com.consol.citrus.exceptions.ActionTimeoutException;18import com.consol.citrus.message.DefaultMessage;19import com.consol.citrus.message.Message;20import com.consol.citrus.testng.AbstractTestNGUnitTest;21import com.consol.citrus.websocket.client.WebSocketClientEndpointConfiguration;22import com.consol.citrus.websocket.handler.CitrusWebSocketHandler;23import com.consol.citrus.websocket.message.WebSocketMessage;24import org.mockito.Mockito;25import org.mockito.invocation.InvocationOnMock;26import org.mockito.stubbing.Answer;27import org.springframework.util.concurrent.ListenableFuture;28import org.springframework.util.concurrent.SettableListenableFuture;29import org.springframework.web.socket.*;30import org.springframework.web.socket.client.WebSocketClient;31import org.testng.Assert;32import org.testng.annotations.Test;33import java.net.URI;34import static org.mockito.Mockito.*;35public class WebSocketClientTest extends AbstractTestNGUnitTest {36    private WebSocketClient client = Mockito.mock(WebSocketClient.class);37    private WebSocketSession session = Mockito.mock(WebSocketSession.class);38    @Test39    public void testWebSocketClient() throws Exception {40        WebSocketClientEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();41        WebSocketEndpoint webSocketEndpoint = new WebSocketEndpoint(endpointConfiguration);42        String endpointUri = "ws://localhost:8088/test";43        final String responseBody = "<TestResponse><Message>Hello World!</Message></TestResponse>";44        endpointConfiguration.setEndpointUri(endpointUri);45        final Message requestMessage = new DefaultMessage("<TestRequest><Message>Hello World!</Message></TestRequest>");46        endpointConfiguration.setClient(client);47        reset(client, session);48        final SettableListenableFuture<WebSocketSession> future = new SettableListenableFuture<>();49        future.set(session);50        doAnswer(new Answer<ListenableFuture<WebSocketSession>>() {51            @Override52            public ListenableFuture<WebSocketSession> answer(InvocationOnMock invocation) throws Throwable {53                CitrusWebSocketHandler handler = (CitrusWebSocketHandler) invocation.getArguments()[0];54                handler.afterConnectionEstablished(session);55                handler.handleMessage(session, new TextMessage(responseBody));56                return future;57            }58        }).when(client).doHandshake(any(CitrusWebSocketHandler.class), any(WebSocketHttpHeaders.class), any(URI.class));59        when(session.getId()).thenReturn("test-socket-1");60        when(session.isOpen()).thenReturn(true);61        doAnswer(new Answer() {62            @Override63            public Object answer(InvocationOnMock invocation) throws Throwable {64                org.springframework.web.socket.WebSocketMessage request = (org.springframework.web.socket.WebSocketMessage) invocation.getArguments()[0];65                Assert.assertTrue(TextMessage.class.isInstance(request));66                Assert.assertEquals(((TextMessage)request).getPayload(), requestMessage.getPayload(String.class));67                Assert.assertTrue(request.isLast());68                return null;69            }70        }).when(session).sendMessage(any(org.springframework.web.socket.WebSocketMessage.class));71        webSocketEndpoint.createProducer().send(requestMessage, context);72        WebSocketMessage responseMessage = (WebSocketMessage) webSocketEndpoint.createConsumer().receive(context, endpointConfiguration.getTimeout());73        Assert.assertEquals(responseMessage.getPayload(), responseBody);74        Assert.assertTrue(responseMessage.isLast());75    }76    @Test77    public void testWebSocketClientTimeout() throws Exception {78        WebSocketClientEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();79        WebSocketEndpoint webSocketEndpoint = new WebSocketEndpoint(endpointConfiguration);80        String endpointUri = "ws://localhost:8088/test";81        endpointConfiguration.setEndpointUri(endpointUri);82        final Message requestMessage = new DefaultMessage("<TestRequest><Message>Hello World!</Message></TestRequest>");83        endpointConfiguration.setClient(client);84        endpointConfiguration.setTimeout(1000L);85        reset(client, session);86        final SettableListenableFuture<WebSocketSession> future = new SettableListenableFuture<>();87        future.set(session);88        doAnswer(new Answer<ListenableFuture<WebSocketSession>>() {89            @Override90            public ListenableFuture<WebSocketSession> answer(InvocationOnMock invocation) throws Throwable {91                CitrusWebSocketHandler handler = (CitrusWebSocketHandler) invocation.getArguments()[0];92                handler.afterConnectionEstablished(session);...

Source:WebSocketMessageConverterTest.java

...15 */16package com.consol.citrus.websocket.message;17import com.consol.citrus.message.Message;18import com.consol.citrus.testng.AbstractTestNGUnitTest;19import com.consol.citrus.websocket.client.WebSocketClientEndpointConfiguration;20import com.consol.citrus.websocket.endpoint.WebSocketEndpointConfiguration;21import org.springframework.web.socket.BinaryMessage;22import org.springframework.web.socket.TextMessage;23import org.testng.Assert;24import org.testng.annotations.Test;25import java.nio.ByteBuffer;26import java.nio.charset.Charset;27/**28 * @author Christoph Deppisch29 * @since 2.330 */31public class WebSocketMessageConverterTest extends AbstractTestNGUnitTest {32    private WebSocketMessageConverter messageConverter = new WebSocketMessageConverter();33    @Test34    public void testConvertTextMessageOutbound() throws Exception {35        WebSocketEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();36        WebSocketMessage message = new WebSocketMessage("Hello WebSocket!");37        org.springframework.web.socket.WebSocketMessage result = messageConverter.convertOutbound(message, endpointConfiguration, context);38        Assert.assertTrue(TextMessage.class.isInstance(result));39        Assert.assertEquals(((TextMessage) result).getPayload(), "Hello WebSocket!");40        Assert.assertEquals(result.isLast(), true);41        message = new WebSocketMessage("Hello WebSocket - stay tuned!").last(false);42        result = messageConverter.convertOutbound(message, endpointConfiguration, context);43        Assert.assertTrue(TextMessage.class.isInstance(result));44        Assert.assertEquals(((TextMessage) result).getPayload(), "Hello WebSocket - stay tuned!");45        Assert.assertEquals(result.isLast(), false);46    }47    @Test48    public void testConvertBinaryMessageOutbound() throws Exception {49        WebSocketEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();50        WebSocketMessage message = new WebSocketMessage("Hello WebSocket!".getBytes(Charset.forName("UTF-8")));51        org.springframework.web.socket.WebSocketMessage result = messageConverter.convertOutbound(message, endpointConfiguration, context);52        Assert.assertTrue(BinaryMessage.class.isInstance(result));53        Assert.assertEquals(((BinaryMessage) result).getPayload().array(), "Hello WebSocket!".getBytes(Charset.forName("UTF-8")));54        Assert.assertEquals(result.isLast(), true);55    }56    @Test57    public void testConvertTextMessageInbound() throws Exception {58        WebSocketEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();59        org.springframework.web.socket.WebSocketMessage externalMessage = new TextMessage("Hello WebSocket!");60        Message internal = messageConverter.convertInbound(externalMessage, endpointConfiguration, context);61        Assert.assertTrue(WebSocketMessage.class.isInstance(internal));62        Assert.assertEquals(internal.getPayload(String.class), "Hello WebSocket!");63        Assert.assertEquals(((WebSocketMessage) internal).isLast(), true);64        externalMessage = new TextMessage("Hello WebSocket - stay tuned!", false);65        internal = messageConverter.convertInbound(externalMessage, endpointConfiguration, context);66        Assert.assertTrue(WebSocketMessage.class.isInstance(internal));67        Assert.assertEquals(internal.getPayload(String.class), "Hello WebSocket - stay tuned!");68        Assert.assertEquals(((WebSocketMessage) internal).isLast(), false);69    }70    @Test71    public void testConvertBinaryMessageInbound() throws Exception {72        WebSocketEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();73        org.springframework.web.socket.WebSocketMessage externalMessage = new BinaryMessage("Hello WebSocket!".getBytes(Charset.forName("UTF-8")));74        Message internal = messageConverter.convertInbound(externalMessage, endpointConfiguration, context);75        Assert.assertTrue(WebSocketMessage.class.isInstance(internal));76        Assert.assertEquals(internal.getPayload(ByteBuffer.class).array(), "Hello WebSocket!".getBytes(Charset.forName("UTF-8")));77        Assert.assertEquals(((WebSocketMessage) internal).isLast(), true);78    }79}...

Source:WebSocketEndpointComponent.java

...16package com.consol.citrus.websocket.endpoint;17import com.consol.citrus.context.TestContext;18import com.consol.citrus.endpoint.AbstractEndpointComponent;19import com.consol.citrus.endpoint.Endpoint;20import com.consol.citrus.websocket.client.WebSocketClientEndpointConfiguration;21import java.util.Map;22/**23 * Web Socket endpoint component is able to create Web Socket client endpoint from endpoint uri with parameters.24 *25 * @author Martin Maher26 * @since 2.327 */28public class WebSocketEndpointComponent extends AbstractEndpointComponent {29    @Override30    protected Endpoint createEndpoint(String resourcePath, Map<String, String> parameters, TestContext context) {31        WebSocketClientEndpointConfiguration endpointConfiguration = new WebSocketClientEndpointConfiguration();32        endpointConfiguration.setEndpointUri(String.format("ws://%s%s", resourcePath, getParameterString(parameters, WebSocketClientEndpointConfiguration.class)));33        WebSocketEndpoint endpoint = new WebSocketEndpoint(endpointConfiguration);34        enrichEndpointConfiguration(endpoint.getEndpointConfiguration(), getEndpointConfigurationParameters(parameters, WebSocketClientEndpointConfiguration.class), context);35        return endpoint;36    }37}...