diff options
Diffstat (limited to 'jdisc-security-filters/src/test/java/com/yahoo/jdisc')
-rw-r--r-- | jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/base/JsonSecurityRequestFilterBaseTest.java | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/base/JsonSecurityRequestFilterBaseTest.java b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/base/JsonSecurityRequestFilterBaseTest.java new file mode 100644 index 00000000000..9b0a875a73d --- /dev/null +++ b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/base/JsonSecurityRequestFilterBaseTest.java @@ -0,0 +1,57 @@ +package com.yahoo.jdisc.http.filter.security.base; + +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import com.yahoo.container.jdisc.RequestHandlerTestDriver; +import com.yahoo.jdisc.Response; +import com.yahoo.jdisc.http.filter.DiscFilterRequest; +import org.junit.Test; + +import java.io.IOException; +import java.util.Optional; + +import static org.hamcrest.CoreMatchers.equalTo; +import static org.hamcrest.CoreMatchers.notNullValue; +import static org.junit.Assert.assertThat; +import static org.mockito.Mockito.mock; + +/** + * @author bjorncs + */ +public class JsonSecurityRequestFilterBaseTest { + + private final ObjectMapper mapper = new ObjectMapper(); + + @Test + public void filter_renders_errors_as_json() throws IOException { + int statusCode = 403; + String message = "Forbidden"; + DiscFilterRequest request = mock(DiscFilterRequest.class); + SimpleSecurityRequestFilter filter = + new SimpleSecurityRequestFilter(new JsonSecurityRequestFilterBase.ErrorResponse(statusCode, message)); + RequestHandlerTestDriver.MockResponseHandler responseHandler = new RequestHandlerTestDriver.MockResponseHandler(); + filter.filter(request, responseHandler); + + Response response = responseHandler.getResponse(); + assertThat(response, notNullValue()); + assertThat(response.getStatus(), equalTo(statusCode)); + + JsonNode jsonNode = mapper.readTree(responseHandler.readAll()); + assertThat(jsonNode.get("message").asText(), equalTo(message)); + assertThat(jsonNode.get("code").asInt(), equalTo(statusCode)); + } + + private static class SimpleSecurityRequestFilter extends JsonSecurityRequestFilterBase { + private final ErrorResponse errorResponse; + + SimpleSecurityRequestFilter(ErrorResponse errorResponse) { + this.errorResponse = errorResponse; + } + + @Override + protected Optional<ErrorResponse> filter(DiscFilterRequest request) { + return Optional.ofNullable(this.errorResponse); + } + } + +}
\ No newline at end of file |