/** @author Federico Recio */
public class SimpleServerTest extends DropwizardTest {

  @ClassRule
  public static final DropwizardAppRule<TestConfiguration> RULE =
      new DropwizardAppRule<>(
          TestApplication.class, ResourceHelpers.resourceFilePath("test-simple.yaml"));

  public SimpleServerTest() {
    super(55668, "/application");
  }
}
public class DropwizardApacheConnectorTest {

  private static final int SLEEP_TIME_IN_MILLIS = 1000;
  private static final int DEFAULT_CONNECT_TIMEOUT_IN_MILLIS = 500;
  private static final int ERROR_MARGIN_IN_MILLIS = 300;
  private static final int INCREASE_IN_MILLIS = 100;
  private static final URI NON_ROUTABLE_ADDRESS = URI.create("http://10.255.255.1");

  @ClassRule
  public static final DropwizardAppRule<Configuration> APP_RULE =
      new DropwizardAppRule<>(
          TestApplication.class,
          ResourceHelpers.resourceFilePath("yaml/dropwizardApacheConnectorTest.yml"));

  @Rule public ExpectedException thrown = ExpectedException.none();

  private final URI testUri = URI.create("http://localhost:" + APP_RULE.getLocalPort());

  private JerseyClient client;
  private Environment environment;

  @Before
  public void setup() throws Exception {
    JerseyClientConfiguration clientConfiguration = new JerseyClientConfiguration();
    clientConfiguration.setConnectionTimeout(Duration.milliseconds(SLEEP_TIME_IN_MILLIS / 2));
    clientConfiguration.setTimeout(Duration.milliseconds(DEFAULT_CONNECT_TIMEOUT_IN_MILLIS));

    environment =
        new Environment(
            "test-dropwizard-apache-connector",
            Jackson.newObjectMapper(),
            Validators.newValidator(),
            new MetricRegistry(),
            getClass().getClassLoader());
    client =
        (JerseyClient)
            new JerseyClientBuilder(environment).using(clientConfiguration).build("test");
    for (LifeCycle lifeCycle : environment.lifecycle().getManagedObjects()) {
      lifeCycle.start();
    }
  }

  @After
  public void tearDown() throws Exception {
    for (LifeCycle lifeCycle : environment.lifecycle().getManagedObjects()) {
      lifeCycle.stop();
    }
    assertThat(client.isClosed()).isTrue();
  }

  @Test
  public void when_no_read_timeout_override_then_client_request_times_out() {
    thrown.expect(ProcessingException.class);
    thrown.expectCause(any(SocketTimeoutException.class));

    client.target(testUri + "/long_running").request().get();
  }

  @Test
  public void when_read_timeout_override_created_then_client_requests_completes_successfully() {
    client
        .target(testUri + "/long_running")
        .property(ClientProperties.READ_TIMEOUT, SLEEP_TIME_IN_MILLIS * 2)
        .request()
        .get();
  }

  /**
   * In first assertion we prove, that a request takes no longer than: <em>request_time <
   * connect_timeout + error_margin</em> (1)
   *
   * <p>In the second we show that if we set <b>connect_timeout</b> to <b>set_connect_timeout +
   * increase + error_margin</b> then <em>request_time > connect_timeout + increase +
   * error_margin</em> (2)
   *
   * <p>
   *
   * <p>Now, (1) and (2) can hold at the same time if then connect_timeout update was successful.
   */
  @Test
  public void connect_timeout_override_changes_how_long_it_takes_for_a_connection_to_timeout() {
    // before override
    WebTarget target = client.target(NON_ROUTABLE_ADDRESS);

    // This can't be tested without a real connection
    try {
      target.request().get(Response.class);
    } catch (ProcessingException e) {
      if (e.getCause() instanceof HttpHostConnectException) {
        return;
      }
    }

    assertThatConnectionTimeoutFor(target)
        .isLessThan(DEFAULT_CONNECT_TIMEOUT_IN_MILLIS + ERROR_MARGIN_IN_MILLIS);

    // after override
    final int newTimeout =
        DEFAULT_CONNECT_TIMEOUT_IN_MILLIS + INCREASE_IN_MILLIS + ERROR_MARGIN_IN_MILLIS;
    final WebTarget newTarget = target.property(ClientProperties.CONNECT_TIMEOUT, newTimeout);
    assertThatConnectionTimeoutFor(newTarget).isGreaterThan(newTimeout);
  }

  @Test
  public void when_no_override_then_redirected_request_successfully_redirected() {
    assertThat(client.target(testUri + "/redirect").request().get(String.class))
        .isEqualTo("redirected");
  }

  @Test
  public void
      when_configuration_overridden_to_disallow_redirects_temporary_redirect_status_returned() {
    assertThat(
            client
                .target(testUri + "/redirect")
                .property(ClientProperties.FOLLOW_REDIRECTS, false)
                .request()
                .get(Response.class)
                .getStatus())
        .isEqualTo(HttpStatus.SC_TEMPORARY_REDIRECT);
  }

  @Test
  public void when_jersey_client_runtime_is_garbage_collected_apache_client_is_not_closed() {
    for (int j = 0; j < 5; j++) {
      System.gc(); // We actually want GC here
      final String response =
          client
              .target(testUri + "/long_running")
              .property(ClientProperties.READ_TIMEOUT, SLEEP_TIME_IN_MILLIS * 2)
              .request()
              .get(String.class);
      assertThat(response).isEqualTo("success");
    }
  }

  @Test
  public void multiple_headers_with_the_same_name_are_processed_successfully() throws Exception {

    final CloseableHttpClient client = mock(CloseableHttpClient.class);
    final DropwizardApacheConnector dropwizardApacheConnector =
        new DropwizardApacheConnector(client, null, false);
    final Header[] apacheHeaders = {
      new BasicHeader("Set-Cookie", "test1"), new BasicHeader("Set-Cookie", "test2")
    };

    final CloseableHttpResponse apacheResponse = mock(CloseableHttpResponse.class);
    when(apacheResponse.getStatusLine())
        .thenReturn(new BasicStatusLine(new ProtocolVersion("HTTP", 1, 1), 200, "OK"));
    when(apacheResponse.getAllHeaders()).thenReturn(apacheHeaders);
    when(client.execute(Matchers.any())).thenReturn(apacheResponse);

    final ClientRequest jerseyRequest = mock(ClientRequest.class);
    when(jerseyRequest.getUri()).thenReturn(URI.create("http://localhost"));
    when(jerseyRequest.getMethod()).thenReturn("GET");
    when(jerseyRequest.getHeaders()).thenReturn(new MultivaluedHashMap<>());

    final ClientResponse jerseyResponse = dropwizardApacheConnector.apply(jerseyRequest);

    assertThat(jerseyResponse.getStatus())
        .isEqualTo(apacheResponse.getStatusLine().getStatusCode());
  }

  @Path("/")
  public static class TestResource {

    @GET
    @Path("/long_running")
    public String getWithSleep() throws InterruptedException {
      TimeUnit.MILLISECONDS.sleep(SLEEP_TIME_IN_MILLIS);
      return "success";
    }

    @GET
    @Path("redirect")
    public Response getWithRedirect() {
      return Response.temporaryRedirect(URI.create("/redirected")).build();
    }

    @GET
    @Path("redirected")
    public String redirectedGet() {
      return "redirected";
    }
  }

  public static class TestApplication extends Application<Configuration> {
    public static void main(String[] args) throws Exception {
      new TestApplication().run(args);
    }

    @Override
    public void run(Configuration configuration, Environment environment) throws Exception {
      environment.jersey().register(TestResource.class);
      environment
          .healthChecks()
          .register(
              "dummy",
              new HealthCheck() {
                @Override
                protected Result check() throws Exception {
                  return Result.healthy();
                }
              });
    }
  }

  private static AbstractLongAssert<?> assertThatConnectionTimeoutFor(WebTarget webTarget) {
    final long startTime = System.nanoTime();
    try {
      webTarget.request().get(Response.class);
    } catch (ProcessingException e) {
      final long endTime = System.nanoTime();
      assertThat(e).isNotNull();
      //noinspection ConstantConditions
      assertThat(e.getCause()).isNotNull();
      assertThat(e.getCause())
          .isInstanceOfAny(ConnectTimeoutException.class, NoRouteToHostException.class);
      return assertThat(TimeUnit.MILLISECONDS.convert(endTime - startTime, TimeUnit.NANOSECONDS));
    }
    throw new AssertionError("ProcessingException expected but not thrown");
  }
}
public class CustomIntegrationTest {

  private static final String TMP_FILE = createTempFile();
  private static final String CONFIG_PATH = ResourceHelpers.resourceFilePath("test-example.yml");

  @ClassRule
  public static final DropwizardAppRule<HelloWorldConfiguration> RULE =
      new DropwizardAppRule<>(
          HelloWorldApplication.class,
          CONFIG_PATH,
          ConfigOverride.config("database.url", "jdbc:h2:" + TMP_FILE));

  private Client client;

  @BeforeClass
  public static void migrateDb() throws Exception {
    RULE.getApplication().run("db", "migrate", CONFIG_PATH);
  }

  @Before
  public void setUp() throws Exception {
    client = ClientBuilder.newClient();
  }

  @After
  public void tearDown() throws Exception {
    client.close();
  }

  private static String createTempFile() {
    try {
      return File.createTempFile("test-example", null).getAbsolutePath();
    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
  }

  /*@Test
  public void postProcess() throws Exception {
      final Process process = new Process("First process");
      final ProcessDTO processDTO = client.target("http://localhost:" + RULE.getLocalPort() + "/processes")
              .request()
              .post(Entity.entity(process, APPLICATION_JSON_TYPE))
              .readEntity(ProcessDTO.class);
      assertThat(processDTO.getId()).isNotNull();
      assertThat(processDTO.getName()).isEqualTo(process.getName());
  }

  @Test
  public void postTask() throws Exception {
      final Task task = new Task("First task");
      final TaskDTO taskDTO = client.target("http://localhost:" + RULE.getLocalPort() + "/tasks")
              .request()
              .post(Entity.entity(task, APPLICATION_JSON_TYPE))
              .readEntity(TaskDTO.class);
      assertThat(taskDTO.getId()).isNotNull();
      assertThat(taskDTO.getName()).isEqualTo(task.getName());
  }*/

  @Test
  public void postProcessWithTasks() throws Exception {
    final Process process = new Process("Process with tasks");
    int tasksCount = 10;
    createTasksForProcess(process, tasksCount);
    final ProcessDTO processDTO =
        client
            .target("http://localhost:" + RULE.getLocalPort() + "/processes")
            .request()
            .post(Entity.entity(process, APPLICATION_JSON_TYPE))
            .readEntity(ProcessDTO.class);
    // TODO check Response
    assertThat(processDTO.getId()).isNotNull();
    assertThat(processDTO.getName()).isEqualTo(process.getName());
    assertThat(processDTO.getTasks()).isNotEmpty();
    assertThat(processDTO.getTasks().size()).isEqualTo(tasksCount);
  }

  private void createTasksForProcess(Process process, int count) {
    for (long i = 1; i <= count; i++) {
      Task task = new Task("Task " + i);
      final TaskDTO taskDTO =
          client
              .target("http://localhost:" + RULE.getLocalPort() + "/tasks")
              .request()
              .post(Entity.entity(task, APPLICATION_JSON_TYPE))
              .readEntity(TaskDTO.class);

      ProcessTask assoc = new ProcessTask();
      task.setId(taskDTO.getId());
      assoc.setTask(task);
      assoc.setPosition(i);
      process.getTaskAssoc().add(assoc);
    }
  }
}
示例#4
0
public class TranxTest {

  @ClassRule
  public static final DropwizardAppRule<TranxConfiguration> RULE =
      new DropwizardAppRule<TranxConfiguration>(
          TranxServer.class, ResourceHelpers.resourceFilePath("tranx.yaml"));

  private WebTarget client;

  @Before
  public void setup() {
    client =
        new JerseyClientBuilder(RULE.getEnvironment())
            .build("test client")
            .target(String.format("http://localhost:%d/", RULE.getLocalPort()));
  }

  @Test
  public void createTransactionReturnsOk() {
    // create "cars" transaction
    Response response =
        client
            .path("/transactionservice/transaction/10")
            .request()
            .put(Entity.json(new Transaction("cars", new BigDecimal(5000), 0l)));

    assertEquals(200, response.getStatus());
    assertEquals(
        TransactionService.OperationResult.OK,
        response.readEntity(TransactionService.OperationResult.class));

    // create "shopping" transaction
    response =
        client
            .path("/transactionservice/transaction/11")
            .request()
            .put(Entity.json(new Transaction("shopping", new BigDecimal(10000), 10l)));

    assertEquals(200, response.getStatus());
    assertEquals(
        TransactionService.OperationResult.OK,
        response.readEntity(TransactionService.OperationResult.class));

    // get "cars" transactions
    response = client.path("/transactionservice/type/cars").request().get();

    assertEquals(200, response.getStatus());
    @SuppressWarnings("unchecked")
    List<Integer> ids = response.readEntity(List.class);
    assertEquals(1, ids.size());
    assertEquals(10, ids.get(0).intValue());

    // get "sum" for transaction 10
    response = client.path("/transactionservice/sum/10").request().get();

    assertEquals(200, response.getStatus());
    AggregatorService.Sum sum = response.readEntity(AggregatorService.Sum.class);
    assertEquals(15000, sum.getSum().intValue());

    // get "sum" for transaction 11
    response = client.path("/transactionservice/sum/11").request().get();

    assertEquals(200, response.getStatus());
    sum = response.readEntity(AggregatorService.Sum.class);
    assertEquals(10000, sum.getSum().intValue());
  }
}