mirror of
https://github.com/spring-projects/spring-security.git
synced 2025-06-23 20:42:14 +00:00
Add HttpStatusAccessDeniedHandler
Signed-off-by: earlgrey02 <san06036@naver.com>
This commit is contained in:
parent
22605be60e
commit
1fa1848f9f
@ -0,0 +1,34 @@
|
||||
package org.springframework.security.web.access;
|
||||
|
||||
import jakarta.servlet.ServletException;
|
||||
import jakarta.servlet.http.HttpServletRequest;
|
||||
import jakarta.servlet.http.HttpServletResponse;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.springframework.core.log.LogMessage;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.security.access.AccessDeniedException;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
public class HttpStatusAccessDeniedHandler implements AccessDeniedHandler {
|
||||
|
||||
protected static final Log logger = LogFactory.getLog(HttpStatusAccessDeniedHandler.class);
|
||||
|
||||
private final HttpStatus httpStatus;
|
||||
|
||||
public HttpStatusAccessDeniedHandler(HttpStatus httpStatus) {
|
||||
Assert.notNull(httpStatus, "httpStatus cannot be null");
|
||||
this.httpStatus = httpStatus;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void handle(HttpServletRequest request, HttpServletResponse response,
|
||||
AccessDeniedException accessDeniedException) throws IOException, ServletException {
|
||||
logger.debug(LogMessage.format("Access denied with status code %d", this.httpStatus.value()));
|
||||
|
||||
response.sendError(this.httpStatus.value(), "Access Denied");
|
||||
}
|
||||
|
||||
}
|
@ -0,0 +1,46 @@
|
||||
package org.springframework.security.web.access;
|
||||
|
||||
import jakarta.servlet.ServletException;
|
||||
import jakarta.servlet.http.HttpServletRequest;
|
||||
import jakarta.servlet.http.HttpServletResponse;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.ExtendWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.jupiter.MockitoExtension;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.mock.web.MockHttpServletResponse;
|
||||
import org.springframework.security.access.AccessDeniedException;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
|
||||
|
||||
@ExtendWith(MockitoExtension.class)
|
||||
public class HttpStatusAccessDeniedHandlerTests {
|
||||
|
||||
@Mock
|
||||
private HttpServletRequest request;
|
||||
|
||||
@Mock
|
||||
private HttpServletResponse response;
|
||||
|
||||
private HttpStatus httpStatus = HttpStatus.FORBIDDEN;
|
||||
|
||||
private HttpStatusAccessDeniedHandler handler = new HttpStatusAccessDeniedHandler(this.httpStatus);
|
||||
|
||||
private AccessDeniedException exception = new AccessDeniedException("Forbidden");
|
||||
|
||||
@Test
|
||||
public void constructorHttpStatusWhenNullThenException() {
|
||||
assertThatIllegalArgumentException().isThrownBy(() -> new HttpStatusAccessDeniedHandler(null));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void commenceThenStatusSet() throws IOException, ServletException {
|
||||
this.response = new MockHttpServletResponse();
|
||||
this.handler.handle(this.request, this.response, this.exception);
|
||||
assertThat(this.response.getStatus()).isEqualTo(this.httpStatus.value());
|
||||
}
|
||||
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user