使用Mockito测试Servlet时测试类无法获取响应问题求助

我帮你分析下问题所在，然后给出两种可行的解决方案：

首先，你遇到的核心问题是：Mockito默认生成的ServletResponse mock对象，它的ServletOutputStream是一个空实现——写入的内容不会被保存，只是“路过”一下，所以测试代码里自然拿不到输出的响应内容，哪怕控制台能看到打印的日志。

下面是两种解决方法，推荐第一种更简便的方案：

方案1：使用官方提供的MockHttpServletResponse #

Servlet API（或Spring Test等测试库）提供了专门用于测试的MockHttpServletResponse类，它已经帮你实现了响应内容的存储逻辑，能直接获取输出的内容。

代码示例： #

import org.junit.Test;
import javax.servlet.http.MockHttpServletResponse;
import static org.junit.Assert.*;

public class YourServletTest {

    @Test
    public void testCommitResponseBuffer() throws IOException {
        // 1. 创建MockHttpServletResponse实例（而非Mockito mock）
        MockHttpServletResponse response = new MockHttpServletResponse();
        String testBuffer = "测试响应内容";

        // 2. 调用Servlet的目标方法
        YourServlet servlet = new YourServlet();
        servlet.commitResponseBuffer(testBuffer, response);

        // 3. 验证响应参数和内容
        assertEquals("text/plain", response.getContentType());
        assertEquals(testBuffer, response.getContentAsString());
    }
}

方案2：用Mockito自定义ServletOutputStream捕获输出 #

如果你的场景必须使用Mockito的ServletResponse mock，可以自定义一个ServletOutputStream来捕获写入的内容，再验证结果：

代码示例： #

import org.junit.Test;
import javax.servlet.ServletOutputStream;
import javax.servlet.WriteListener;
import javax.servlet.ServletResponse;
import java.io.ByteArrayOutputStream;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

public class YourServletTest {

    @Test
    public void testCommitResponseBufferWithMockito() throws IOException {
        // 1. 创建ByteArrayOutputStream用来保存输出内容
        ByteArrayOutputStream captureStream = new ByteArrayOutputStream();
        
        // 2. 自定义ServletOutputStream，把写入的内容转存到captureStream
        ServletOutputStream mockStream = new ServletOutputStream() {
            @Override
            public void write(int b) throws IOException {
                captureStream.write(b);
            }

            @Override
            public boolean isReady() {
                return true;
            }

            @Override
            public void setWriteListener(WriteListener writeListener) {
                // 测试场景下无需实现，留空即可
            }
        };

        // 3. Mock ServletResponse，让它返回我们自定义的输出流
        ServletResponse response = mock(ServletResponse.class);
        when(response.getOutputStream()).thenReturn(mockStream);

        // 4. 调用Servlet方法
        String testBuffer = "测试响应内容";
        YourServlet servlet = new YourServlet();
        servlet.commitResponseBuffer(testBuffer, response);

        // 5. 验证结果
        verify(response).setContentType("text/plain");
        assertEquals(testBuffer, captureStream.toString());
    }
}

总结一下：第一种方案更简洁，适合大多数Servlet测试场景；第二种方案更灵活，适合必须使用Mockito mock对象的特殊场景。

内容的提问来源于stack exchange，提问作者Jaydeep Bobade