21.11.10 TIL

오늘의 삽질

Junit으로 Filter 테스트 하기

response, request, filterChain을 mock 객체로 생성하여
given()으로 filterChain.doFileter() 메서드를 mock 으로 응답받을 수 있다.

@ExtendWith(MockitoExtension.class)
class JwtExceptionHandlerFilterTest {

    HttpServletResponse response = null;
    HttpServletRequest request = null;
    FilterChain filterChain = null;
    PrintWriter mockWriter = null;

    @BeforeEach
    private void setUp() throws IOException {
        //given
        response = mock(HttpServletResponse.class);
        request = mock(HttpServletRequest.class);
        filterChain = mock(FilterChain.class);
        mockWriter = mock(PrintWriter.class);

        String clientRequestUri = "/test";
        String message = "test message";
        String body = null;
        String method = "GET";
        int code = 400;

        given(request.getAttribute("clientRequestUri")).willReturn(clientRequestUri);
        given(request.getAttribute("msg")).willReturn(message);
        given(request.getAttribute("messageBody")).willReturn(body);
        given(request.getAttribute("method")).willReturn(method);
        given(request.getAttribute("code")).willReturn(400);
        given(response.getWriter()).willReturn(mockWriter);
    }

    @DisplayName("Jwt ExpiredJwtException 처리")
    @Test
    void filterExpiredJwtException() throws ServletException, IOException {
        //given
        willThrow(ExpiredJwtException.class).given(filterChain).doFilter(request, response);

        //when
        JwtExceptionHandlerFilter jwtExceptionHandlerFilter = new JwtExceptionHandlerFilter();
        jwtExceptionHandlerFilter.doFilterInternal(request,response,filterChain);

        //then
        verify(filterChain).doFilter(request, response);
        verify(response.getWriter()).write(anyString());
    }

    @DisplayName("Jwt SignatureException 처리")
    @Test
    void filterSignatureException() throws ServletException, IOException {
        //given
        willThrow(SignatureException.class).given(filterChain).doFilter(request, response);

        //when
        JwtExceptionHandlerFilter jwtExceptionHandlerFilter = new JwtExceptionHandlerFilter();
        jwtExceptionHandlerFilter.doFilterInternal(request,response,filterChain);

        //then
        verify(filterChain).doFilter(request, response);
        verify(response.getWriter()).write(anyString());
    }

    @DisplayName("Jwt MalformedJwtException 처리")
    @Test
    void filterMalformedJwtException() throws ServletException, IOException {
        //given
        willThrow(MalformedJwtException.class).given(filterChain).doFilter(request, response);

        //when
        JwtExceptionHandlerFilter jwtExceptionHandlerFilter = new JwtExceptionHandlerFilter();
        jwtExceptionHandlerFilter.doFilterInternal(request,response,filterChain);

        //then
        verify(filterChain).doFilter(request, response);
        verify(response.getWriter()).write(anyString());
    }
}