Examples with Response org.apache.apex.shaded.ning19.com.ning.http.client.Response used on opensource projects

Example 11 with Response

use of org.apache.apex.shaded.ning19.com.ning.http.client.Response in project riposte by Nike-Inc.

the class AsyncCompletionHandlerWithTracingAndMdcSupportTest method onCompleted_handles_circuit_breaker_but_does_nothing_else_if_completableFutureResponse_is_already_completed.

@Test
public void onCompleted_handles_circuit_breaker_but_does_nothing_else_if_completableFutureResponse_is_already_completed() throws Throwable {
    // given
    CompletableFuture<String> cfMock = mock(CompletableFuture.class);
    Whitebox.setInternalState(handlerSpy, "completableFutureResponse", cfMock);
    doReturn(true).when(cfMock).isDone();
    // when
    Response ignoredResult = handlerSpy.onCompleted(responseMock);
    // then
    verify(circuitBreakerManualTaskMock).handleEvent(responseMock);
    verify(cfMock).isDone();
    verifyNoMoreInteractions(cfMock);
    assertThat(ignoredResult).isEqualTo(responseMock);
    verifyZeroInteractions(responseMock);
}

Example 12 with Response

use of org.apache.apex.shaded.ning19.com.ning.http.client.Response in project riposte by Nike-Inc.

the class AsyncCompletionHandlerWithTracingAndMdcSupportTest method onCompleted_completes_completableFutureResponse_with_result_of_responseHandlerFunction.

@DataProvider(value = { "true", "false" })
@Test
public void onCompleted_completes_completableFutureResponse_with_result_of_responseHandlerFunction(boolean throwException) throws Throwable {
    // given
    Exception ex = new Exception("kaboom");
    if (throwException)
        doThrow(ex).when(responseHandlerFunctionMock).handleResponse(responseMock);
    // when
    Response ignoredResult = handlerSpy.onCompleted(responseMock);
    // then
    verify(responseHandlerFunctionMock).handleResponse(responseMock);
    if (throwException) {
        assertThat(completableFutureResponse).isCompletedExceptionally();
        assertThat(completableFutureResponse).hasFailedWithThrowableThat().isEqualTo(ex);
    } else {
        assertThat(completableFutureResponse).isCompleted();
        assertThat(completableFutureResponse.get()).isEqualTo(responseHandlerResult);
    }
    assertThat(ignoredResult).isEqualTo(responseMock);
    verifyZeroInteractions(responseMock);
}

Example 13 with Response

use of org.apache.apex.shaded.ning19.com.ning.http.client.Response in project apex-core by apache.

the class PubSubWebSocketClient method openConnectionAsync.

public void openConnectionAsync() throws IOException {
    throwable.set(null);
    if (loginUrl != null && userName != null && password != null) {
        // get the session key first before attempting web socket
        JSONObject json = new JSONObject();
        try {
            json.put("userName", userName);
            json.put("password", password);
        } catch (JSONException ex) {
            throw new RuntimeException(ex);
        }
        client.preparePost(loginUrl).setHeader("Content-Type", "application/json").setBody(json.toString()).execute(new AsyncCompletionHandler<Response>() {

            @Override
            public Response onCompleted(Response response) throws Exception {
                List<Cookie> cookies = response.getCookies();
                BoundRequestBuilder brb = client.prepareGet(uri.toString());
                if (cookies != null) {
                    for (Cookie cookie : cookies) {
                        brb.addCookie(cookie);
                    }
                }
                connection = brb.execute(new WebSocketUpgradeHandler.Builder().addWebSocketListener(new PubSubWebSocket()).build()).get();
                return response;
            }
        });
    } else {
        final PubSubWebSocket webSocket = new PubSubWebSocket() {

            @Override
            public void onOpen(WebSocket ws) {
                connection = ws;
                super.onOpen(ws);
            }
        };
        client.prepareGet(uri.toString()).execute(new WebSocketUpgradeHandler.Builder().addWebSocketListener(webSocket).build());
    }
}

Example 14 with Response

use of org.apache.apex.shaded.ning19.com.ning.http.client.Response in project apex-core by apache.

the class PubSubWebSocketClient method openConnection.

/**
   * <p>openConnection.</p>
   *
   * @param timeoutMillis
   * @throws IOException
   * @throws ExecutionException
   * @throws InterruptedException
   * @throws TimeoutException
   */
public void openConnection(long timeoutMillis) throws IOException, ExecutionException, InterruptedException, TimeoutException {
    throwable.set(null);
    List<Cookie> cookies = null;
    if (loginUrl != null && userName != null && password != null) {
        // get the session key first before attempting web socket
        JSONObject json = new JSONObject();
        try {
            json.put("userName", userName);
            json.put("password", password);
        } catch (JSONException ex) {
            throw new RuntimeException(ex);
        }
        Response response = client.preparePost(loginUrl).setHeader("Content-Type", "application/json").setBody(json.toString()).execute().get();
        cookies = response.getCookies();
    }
    BoundRequestBuilder brb = client.prepareGet(uri.toString());
    if (cookies != null) {
        for (Cookie cookie : cookies) {
            brb.addCookie(cookie);
        }
    }
    connection = brb.execute(new WebSocketUpgradeHandler.Builder().addWebSocketListener(new PubSubWebSocket()).build()).get(timeoutMillis, TimeUnit.MILLISECONDS);
}

Example 15 with Response

use of org.apache.apex.shaded.ning19.com.ning.http.client.Response in project elasticsearch-suggest-plugin by spinscale.

the class RestSuggestActionTest method refreshFieldSuggesters.

@Override
public void refreshFieldSuggesters(String index, String field) throws Exception {
    String jsonBody = String.format(Locale.ROOT, "{ \"field\": \"%s\" } ", field);
    Response r = httpClient.preparePost("http://localhost:" + port + "/" + index + "/product/__suggestRefresh").setBody(jsonBody).execute().get();
    assertThat(r.getStatusCode(), is(200));
}