package io.dropwizard.jersey.guava;
import com.codahale.metrics.MetricRegistry;
import com.google.common.base.Optional;
import io.dropwizard.jersey.DropwizardResourceConfig;
import io.dropwizard.jersey.params.UUIDParam;
import io.dropwizard.logging.LoggingFactory;
import org.glassfish.jersey.test.JerseyTest;
import org.junit.Test;
import javax.ws.rs.BadRequestException;
import javax.ws.rs.GET;
import javax.ws.rs.HeaderParam;
import javax.ws.rs.Path;
import javax.ws.rs.core.Application;
import static org.assertj.core.api.Assertions.assertThat;
public class OptionalHeaderParamResourceTest extends JerseyTest {
static {
LoggingFactory.bootstrap();
}
@Override
protected Application configure() {
return DropwizardResourceConfig.forTesting(new MetricRegistry())
.register(OptionalHeaderParamResource.class)
.register(MyMessageParamConverterProvider.class);
}
@Test
public void shouldReturnDefaultMessageWhenMessageIsNotPresent() {
String defaultMessage = "Default Message";
String response = target("/optional/message").request().get(String.class);
assertThat(response).isEqualTo(defaultMessage);
}
@Test
public void shouldReturnMessageWhenMessageIsPresent() {
String customMessage = "Custom Message";
String response = target("/optional/message").request().header("message", customMessage).get(String.class);
assertThat(response).isEqualTo(customMessage);
}
@Test
public void shouldReturnDefaultMessageWhenMyMessageIsNotPresent() {
String defaultMessage = "My Default Message";
String response = target("/optional/my-message").request().get(String.class);
assertThat(response).isEqualTo(defaultMessage);
}
@Test
public void shouldReturnMyMessageWhenMyMessageIsPresent() {
String myMessage = "My Message";
String response = target("/optional/my-message").request().header("mymessage", myMessage).get(String.class);
assertThat(response).isEqualTo(myMessage);
}
@Test(expected = BadRequestException.class)
public void shouldThrowBadRequestExceptionWhenInvalidUUIDIsPresent() {
String invalidUUID = "invalid-uuid";
target("/optional/uuid").request().header("uuid", invalidUUID).get(String.class);
}
@Test
public void shouldReturnDefaultUUIDWhenUUIDIsNotPresent() {
String defaultUUID = "d5672fa8-326b-40f6-bf71-d9dacf44bcdc";
String response = target("/optional/uuid").request().get(String.class);
assertThat(response).isEqualTo(defaultUUID);
}
@Test
public void shouldReturnUUIDWhenValidUUIDIsPresent() {
String uuid = "fd94b00d-bd50-46b3-b42f-905a9c9e7d78";
String response = target("/optional/uuid").request().header("uuid", uuid).get(String.class);
assertThat(response).isEqualTo(uuid);
}
@Path("/optional")
public static class OptionalHeaderParamResource {
@GET
@Path("/message")
public String getMessage(@HeaderParam("message") Optional<String> message) {
return message.or("Default Message");
}
@GET
@Path("/my-message")
public String getMyMessage(@HeaderParam("mymessage") Optional<MyMessage> myMessage) {
return myMessage.or(new MyMessage("My Default Message")).getMessage();
}
@GET
@Path("/uuid")
public String getUUID(@HeaderParam("uuid") Optional<UUIDParam> uuid) {
return uuid.or(new UUIDParam("d5672fa8-326b-40f6-bf71-d9dacf44bcdc")).get().toString();
}
}
}