Mockito, how can I call verify on an object created in a method?

Admin 1
Viewed 43

Simple example:

public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
    HttpServletRequest req = (HttpServletRequest) request;
    HttpServletResponse res = (HttpServletResponse) response;
    res.addCookie(cookie);
    chain.doFilter(request, response);
}

How can I verify that addCoockie method was called on a HttpServletResponse object, if that object is created from ServletResponse.

mockito junit java
1 Answers

Pass mock of HttpServletResponse to the filter and then verify it.

Custom filter that we need to test

import javax.servlet.*;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

public class CustomFilter implements Filter {
    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) request;
        HttpServletResponse res = (HttpServletResponse) response;
        res.addCookie(new Cookie("name", "value")); // filter call addCookie method which need to be verified
        filterChain.doFilter(request, response);
    }

    @Override
    public void destroy() {

    }
}

Test implementation

import org.junit.jupiter.api.Test;

import javax.servlet.FilterChain;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import static org.mockito.Mockito.*;

public class FilterTest {
    CustomFilter customFilter = new CustomFilter();

    @Test
    public void testDoFilter() throws Exception {
        HttpServletRequest request = mock(HttpServletRequest.class);
        //create mock of HttpServletResponse object
        HttpServletResponse response = mock(HttpServletResponse.class);
        FilterChain filterChain = mock(FilterChain.class);

        //pass HttpServletResponse object to the filter
        customFilter.doFilter(request, response, filterChain);

        //verify that addCookie method was executed with correct value
        verify(response, times(1)).addCookie(argThat(cookie -> "name".equals(cookie.getName()) && "value".equals(cookie.getValue())));
    }
}
user_38344130
Related