Examples with ClientEndpointConfig jakarta.websocket.ClientEndpointConfig used on opensource projects

Example 11 with ClientEndpointConfig

use of jakarta.websocket.ClientEndpointConfig in project tomcat by apache.

the class TestWebSocketFrameClientSSL method testBug56032Legacy.

@SuppressWarnings("removal")
@Test
public void testBug56032Legacy() throws Exception {
    Tomcat tomcat = getTomcatInstance();
    // No file system docBase required
    Context ctx = tomcat.addContext("", null);
    ctx.addApplicationListener(TesterFirehoseServer.ConfigInline.class.getName());
    Tomcat.addServlet(ctx, "default", new DefaultServlet());
    ctx.addServletMappingDecoded("/", "default");
    tomcat.start();
    WebSocketContainer wsContainer = ContainerProvider.getWebSocketContainer();
    ClientEndpointConfig clientEndpointConfig = ClientEndpointConfig.Builder.create().build();
    clientEndpointConfig.getUserProperties().put(Constants.SSL_TRUSTSTORE_PROPERTY, TesterSupport.CA_JKS);
    Session wsSession = wsContainer.connectToServer(TesterProgrammaticEndpoint.class, clientEndpointConfig, new URI("wss://localhost:" + getPort() + TesterFirehoseServer.PATH));
    // Process incoming messages very slowly
    MessageHandler handler = new SleepingText(5000);
    wsSession.addMessageHandler(handler);
    wsSession.getBasicRemote().sendText("Hello");
    // Wait long enough for the buffers to fill and the send to timeout
    int count = 0;
    int limit = TesterFirehoseServer.WAIT_TIME_MILLIS / 100;
    System.out.println("Waiting for server to report an error");
    while (TesterFirehoseServer.Endpoint.getErrorCount() == 0 && count < limit) {
        Thread.sleep(100);
        count++;
    }
    if (TesterFirehoseServer.Endpoint.getErrorCount() == 0) {
        Assert.fail("No error reported by Endpoint when timeout was expected");
    }
    // Wait up to another 10 seconds for the connection to be closed -
    // should be a lot faster.
    System.out.println("Waiting for connection to be closed");
    count = 0;
    limit = (TesterFirehoseServer.SEND_TIME_OUT_MILLIS * 2) / 100;
    while (TesterFirehoseServer.Endpoint.getOpenConnectionCount() != 0 && count < limit) {
        Thread.sleep(100);
        count++;
    }
    int openConnectionCount = TesterFirehoseServer.Endpoint.getOpenConnectionCount();
    if (openConnectionCount != 0) {
        Assert.fail("There are [" + openConnectionCount + "] connections still open");
    }
    // Close the client session.
    wsSession.close();
}

Example 12 with ClientEndpointConfig

use of jakarta.websocket.ClientEndpointConfig in project tomcat by apache.

the class TestWebSocketFrameClientSSL method testConnectToServerEndpointLegacy.

@SuppressWarnings("removal")
@Test
public void testConnectToServerEndpointLegacy() throws Exception {
    Tomcat tomcat = getTomcatInstance();
    // No file system docBase required
    Context ctx = tomcat.addContext("", null);
    ctx.addApplicationListener(TesterFirehoseServer.ConfigInline.class.getName());
    Tomcat.addServlet(ctx, "default", new DefaultServlet());
    ctx.addServletMappingDecoded("/", "default");
    tomcat.start();
    WebSocketContainer wsContainer = ContainerProvider.getWebSocketContainer();
    ClientEndpointConfig clientEndpointConfig = ClientEndpointConfig.Builder.create().build();
    clientEndpointConfig.getUserProperties().put(Constants.SSL_TRUSTSTORE_PROPERTY, TesterSupport.CA_JKS);
    Session wsSession = wsContainer.connectToServer(TesterProgrammaticEndpoint.class, clientEndpointConfig, new URI("wss://localhost:" + getPort() + TesterFirehoseServer.PATH));
    CountDownLatch latch = new CountDownLatch(TesterFirehoseServer.MESSAGE_COUNT);
    BasicText handler = new BasicText(latch);
    wsSession.addMessageHandler(handler);
    wsSession.getBasicRemote().sendText("Hello");
    System.out.println("Sent Hello message, waiting for data");
    // Ignore the latch result as the message count test below will tell us
    // if the right number of messages arrived
    handler.getLatch().await(TesterFirehoseServer.WAIT_TIME_MILLIS, TimeUnit.MILLISECONDS);
    Queue<String> messages = handler.getMessages();
    Assert.assertEquals(TesterFirehoseServer.MESSAGE_COUNT, messages.size());
    for (String message : messages) {
        Assert.assertEquals(TesterFirehoseServer.MESSAGE, message);
    }
}

Example 13 with ClientEndpointConfig

use of jakarta.websocket.ClientEndpointConfig in project spring-framework by spring-projects.

the class StandardWebSocketClientTests method clientEndpointConfig.

@Test
public void clientEndpointConfig() throws Exception {
    URI uri = new URI("ws://localhost/abc");
    List<String> protocols = Collections.singletonList("abc");
    this.headers.setSecWebSocketProtocol(protocols);
    this.wsClient.doHandshake(this.wsHandler, this.headers, uri).get();
    ArgumentCaptor<ClientEndpointConfig> captor = ArgumentCaptor.forClass(ClientEndpointConfig.class);
    verify(this.wsContainer).connectToServer(any(Endpoint.class), captor.capture(), any(URI.class));
    ClientEndpointConfig endpointConfig = captor.getValue();
    assertThat(endpointConfig.getPreferredSubprotocols()).isEqualTo(protocols);
}

Example 14 with ClientEndpointConfig

use of jakarta.websocket.ClientEndpointConfig in project spring-framework by spring-projects.

the class StandardWebSocketClientTests method clientEndpointConfigWithUserProperties.

@Test
public void clientEndpointConfigWithUserProperties() throws Exception {
    Map<String, Object> userProperties = Collections.singletonMap("foo", "bar");
    URI uri = new URI("ws://localhost/abc");
    this.wsClient.setUserProperties(userProperties);
    this.wsClient.doHandshake(this.wsHandler, this.headers, uri).get();
    ArgumentCaptor<ClientEndpointConfig> captor = ArgumentCaptor.forClass(ClientEndpointConfig.class);
    verify(this.wsContainer).connectToServer(any(Endpoint.class), captor.capture(), any(URI.class));
    ClientEndpointConfig endpointConfig = captor.getValue();
    assertThat(endpointConfig.getUserProperties()).isEqualTo(userProperties);
}

Example 15 with ClientEndpointConfig

use of jakarta.websocket.ClientEndpointConfig in project spring-framework by spring-projects.

the class StandardWebSocketClientTests method standardWebSocketClientConfiguratorInsertsHandshakeHeaders.

@Test
public void standardWebSocketClientConfiguratorInsertsHandshakeHeaders() throws Exception {
    URI uri = new URI("ws://localhost/abc");
    this.headers.add("foo", "bar");
    this.wsClient.doHandshake(this.wsHandler, this.headers, uri).get();
    ArgumentCaptor<ClientEndpointConfig> captor = ArgumentCaptor.forClass(ClientEndpointConfig.class);
    verify(this.wsContainer).connectToServer(any(Endpoint.class), captor.capture(), any(URI.class));
    ClientEndpointConfig endpointConfig = captor.getValue();
    Map<String, List<String>> headers = new HashMap<>();
    endpointConfig.getConfigurator().beforeRequest(headers);
    assertThat(headers.size()).isEqualTo(1);
}