Examples with WebSocketClient org.jboss.pnc.restclient.websocket.WebSocketClient used on opensource projects

Example 6 with WebSocketClient

use of org.jboss.pnc.restclient.websocket.WebSocketClient in project pnc by project-ncl.

the class WebSocketClientTest method testConnection.

@Test
public void testConnection() throws Exception {
    // with
    WebSocketClient wsClient = new VertxWebSocketClient();
    // when
    CompletableFuture<Void> future = wsClient.connect(PNC_SOCKET_URL);
    // then
    assertThat(future).succeedsWithin(500, TimeUnit.MILLISECONDS);
    wsClient.disconnect();
}

Example 7 with WebSocketClient

use of org.jboss.pnc.restclient.websocket.WebSocketClient in project pnc by project-ncl.

the class WebSocketClientTest method shouldReconnectAfterBeingUnresponsive.

@Test
public void shouldReconnectAfterBeingUnresponsive() throws Exception {
    // with
    WebSocketSessionHandler handler = new WebSocketSessionHandler();
    Undertow wsServer = withHandler(handler);
    wsServer.start();
    // set max server unresponsiveness
    int maxUnresponsiveness = 1000;
    WebSocketClient wsClient = withMaxUnresponsiveness(1000);
    // make server unresponsive
    handler.blockPongs();
    // when
    wsClient.connect("ws://localhost:8082" + NOTIFICATION_PATH).get();
    // wait maxUnresponsiveness time plus a little more to give client time to reconnect
    Thread.sleep(maxUnresponsiveness + 500);
    // then
    assertThat(handler.getSessionCounter()).hasValue(2);
    wsClient.close();
    wsServer.stop();
}

Example 8 with WebSocketClient

use of org.jboss.pnc.restclient.websocket.WebSocketClient in project pnc by project-ncl.

the class WebSocketClientTest method testRestBuildFallback.

@Test
public void testRestBuildFallback() throws Exception {
    // with
    WebSocketSessionHandler handler = new WebSocketSessionHandler();
    Undertow wsServer = withHandler(handler);
    wsServer.start();
    WebSocketClient wsClient = new VertxWebSocketClient();
    wsClient.connect("ws://localhost:8082" + NOTIFICATION_PATH).join();
    AdvancedBuildConfigurationClient buildConfigurationClient = new AdvancedBuildConfigurationClient(RestClientConfiguration.asUser());
    // test the actual fallbackSupplier (it's private -> reflection unfortunately)
    Method supplier = buildConfigurationClient.getClass().getDeclaredMethod("fallbackSupplier", String.class);
    supplier.setAccessible(true);
    // when
    BuildConfiguration bc = buildConfigurationClient.getAll().iterator().next();
    CompletableFuture<BuildChangedNotification> future = wsClient.catchBuildChangedNotification(() -> invokeMethod(supplier, Build.class, buildConfigurationClient, bc.getId()), withBuildConfiguration(bc.getId()), withBuildCompleted());
    buildConfigurationClient.trigger(bc.getId(), new BuildParameters());
    // make client reconnect and use REST fallback
    handler.closeSession();
    // then
    assertThat(future).succeedsWithin(1000, TimeUnit.MILLISECONDS);
    wsClient.close();
    wsServer.stop();
}

Example 9 with WebSocketClient

use of org.jboss.pnc.restclient.websocket.WebSocketClient in project pnc by project-ncl.

the class WebSocketClientTest method testNotificationCatcher.

@Test
public void testNotificationCatcher() throws Exception {
    // with
    WebSocketClient wsClient = new VertxWebSocketClient();
    wsClient.connect(PNC_SOCKET_URL).get();
    BuildConfigurationClient buildConfigurationClient = new BuildConfigurationClient(RestClientConfiguration.asUser());
    BuildConfiguration bc = buildConfigurationClient.getAll().iterator().next();
    // when
    CompletableFuture<BuildChangedNotification> future = wsClient.catchBuildChangedNotification(withBuildConfiguration(bc.getId()), withBuildCompleted());
    buildConfigurationClient.trigger(bc.getId(), new BuildParameters());
    // then
    assertThat(future).succeedsWithin(500, TimeUnit.MILLISECONDS);
    BuildChangedNotification bcn = future.get();
    assertThat(bcn).isNotNull();
    assertThat(bcn.getBuild()).isNotNull();
    assertThat(bcn.getBuild().getBuildConfigRevision().getId()).isEqualTo(bc.getId());
    assertThat(bcn.getBuild().getStatus().isFinal()).isTrue();
    wsClient.disconnect();
}