Add HttpStatusAccessDeniedHandler

Signed-off-by: earlgrey02 <san06036@naver.com>
This commit is contained in:
earlgrey02 2025-01-30 13:25:15 +09:00 committed by Steve Riesenberg
parent 22605be60e
commit 1fa1848f9f
2 changed files with 80 additions and 0 deletions

View File

@ -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");
}
}

View File

@ -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());
}
}