コード例 #1
0
  @Test(groups = "standalone")
  public void asyncStreamReusePOSTTest() throws Exception {

    final AtomicReference<HttpHeaders> responseHeaders = new AtomicReference<>();
    try (AsyncHttpClient c = asyncHttpClient()) {
      BoundRequestBuilder rb =
          c.preparePost(getTargetUrl()) //
              .setHeader("Content-Type", "application/x-www-form-urlencoded")
              .addFormParam("param_1", "value_1");

      Future<String> f =
          rb.execute(
              new AsyncHandlerAdapter() {
                private StringBuilder builder = new StringBuilder();

                @Override
                public State onHeadersReceived(HttpResponseHeaders content) throws Exception {
                  responseHeaders.set(content.getHeaders());
                  return State.CONTINUE;
                }

                @Override
                public State onBodyPartReceived(HttpResponseBodyPart content) throws Exception {
                  builder.append(new String(content.getBodyPartBytes()));
                  return State.CONTINUE;
                }

                @Override
                public String onCompleted() throws Exception {
                  return builder.toString();
                }
              });

      String r = f.get(5, TimeUnit.SECONDS);
      HttpHeaders h = responseHeaders.get();
      assertNotNull(h, "Should receive non null headers");
      assertEquals(
          h.get(HttpHeaders.Names.CONTENT_TYPE).toLowerCase(Locale.ENGLISH),
          TEXT_HTML_CONTENT_TYPE_WITH_UTF_8_CHARSET.toLowerCase(Locale.ENGLISH),
          "Unexpected content-type");
      assertNotNull(r, "No response body");
      assertEquals(r.trim(), RESPONSE, "Unexpected response body");

      responseHeaders.set(null);

      // Let do the same again
      f =
          rb.execute(
              new AsyncHandlerAdapter() {
                private StringBuilder builder = new StringBuilder();

                @Override
                public State onHeadersReceived(HttpResponseHeaders content) throws Exception {
                  responseHeaders.set(content.getHeaders());
                  return State.CONTINUE;
                }

                @Override
                public State onBodyPartReceived(HttpResponseBodyPart content) throws Exception {
                  builder.append(new String(content.getBodyPartBytes()));
                  return State.CONTINUE;
                }

                @Override
                public String onCompleted() throws Exception {
                  return builder.toString();
                }
              });

      f.get(5, TimeUnit.SECONDS);
      h = responseHeaders.get();
      assertNotNull(h, "Should receive non null headers");
      assertEquals(
          h.get(HttpHeaders.Names.CONTENT_TYPE).toLowerCase(Locale.ENGLISH),
          TEXT_HTML_CONTENT_TYPE_WITH_UTF_8_CHARSET.toLowerCase(Locale.ENGLISH),
          "Unexpected content-type");
      assertNotNull(r, "No response body");
      assertEquals(r.trim(), RESPONSE, "Unexpected response body");
    }
  }
コード例 #2
0
  @Test
  public void asyncStreamReusePOSTTest() throws Throwable {

    withClient()
        .run(
            client -> {
              withServer(server)
                  .run(
                      server -> {
                        server.enqueueEcho();

                        final AtomicReference<HttpHeaders> responseHeaders =
                            new AtomicReference<>();

                        BoundRequestBuilder rb =
                            client
                                .preparePost(getTargetUrl()) //
                                .setHeader(
                                    CONTENT_TYPE,
                                    HttpHeaderValues.APPLICATION_X_WWW_FORM_URLENCODED) //
                                .addFormParam("param_1", "value_1");

                        Future<String> f =
                            rb.execute(
                                new AsyncHandlerAdapter() {
                                  private StringBuilder builder = new StringBuilder();

                                  @Override
                                  public State onHeadersReceived(HttpResponseHeaders content)
                                      throws Exception {
                                    responseHeaders.set(content.getHeaders());
                                    return State.CONTINUE;
                                  }

                                  @Override
                                  public State onBodyPartReceived(HttpResponseBodyPart content)
                                      throws Exception {
                                    builder.append(new String(content.getBodyPartBytes()));
                                    return State.CONTINUE;
                                  }

                                  @Override
                                  public String onCompleted() throws Exception {
                                    return builder.toString();
                                  }
                                });

                        String r = f.get(5, TimeUnit.SECONDS);
                        HttpHeaders h = responseHeaders.get();
                        assertNotNull(h, "Should receive non null headers");
                        assertContentTypesEquals(
                            h.get(CONTENT_TYPE), TEXT_HTML_CONTENT_TYPE_WITH_UTF_8_CHARSET);
                        assertNotNull(r, "No response body");
                        assertEquals(r.trim(), RESPONSE, "Unexpected response body");

                        responseHeaders.set(null);

                        server.enqueueEcho();

                        // Let do the same again
                        f =
                            rb.execute(
                                new AsyncHandlerAdapter() {
                                  private StringBuilder builder = new StringBuilder();

                                  @Override
                                  public State onHeadersReceived(HttpResponseHeaders content)
                                      throws Exception {
                                    responseHeaders.set(content.getHeaders());
                                    return State.CONTINUE;
                                  }

                                  @Override
                                  public State onBodyPartReceived(HttpResponseBodyPart content)
                                      throws Exception {
                                    builder.append(new String(content.getBodyPartBytes()));
                                    return State.CONTINUE;
                                  }

                                  @Override
                                  public String onCompleted() throws Exception {
                                    return builder.toString();
                                  }
                                });

                        f.get(5, TimeUnit.SECONDS);
                        h = responseHeaders.get();
                        assertNotNull(h, "Should receive non null headers");
                        assertContentTypesEquals(
                            h.get(CONTENT_TYPE), TEXT_HTML_CONTENT_TYPE_WITH_UTF_8_CHARSET);
                        assertNotNull(r, "No response body");
                        assertEquals(r.trim(), RESPONSE, "Unexpected response body");
                      });
            });
  }