Mockito，void方法如何返回HTTP状态？

import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

import javax.servlet.http.HttpServletResponse;

public class MyClass {
    public void myVoidMethod(HttpServletResponse response) {
        // 一些逻辑代码
        response.setStatus(HttpServletResponse.SC_OK);
    }
}

public class MyClassTest {
    @Test
    public void testMyVoidMethod() {
        HttpServletResponse mockResponse = Mockito.mock(HttpServletResponse.class);
        MyClass myClass = new MyClass();

        Mockito.doAnswer((Answer<Void>) invocation -> {
            Object[] args = invocation.getArguments();
            ((HttpServletResponse) args[0]).setStatus(HttpServletResponse.SC_OK);
            return null;
        }).when(mockResponse).setStatus(Mockito.anyInt());

        myClass.myVoidMethod(mockResponse);

        Mockito.verify(mockResponse).setStatus(HttpServletResponse.SC_OK);
    }
}