main函数测试Servlet源码

在main函数中测试Servlet源码的实现方法

在标准Java Web应用中，Servlet由Servlet容器（如Tomcat）管理其生命周期。但通过模拟请求/响应对象，可在main函数中直接测试Servlet核心逻辑。以下是完整实现：

1. 创建测试用Servlet示例

import javax.servlet.http.*;
import java.io.*;

public class TestServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) 
            throws IOException {
        String name = req.getParameter("name");
        resp.setContentType("text/html");
        PrintWriter out = resp.getWriter();
        out.print("<h1>Hello, " + (name != null ? name : "Guest") + "!</h1>");
    }
}

2. 实现main函数测试类

import javax.servlet.http.*;
import java.util.*;

public class ServletTester {
    public static void main(String[] args) throws Exception {
        // 创建模拟请求对象
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setParameter("name", "Alice");
        
        // 创建模拟响应对象
        MockHttpServletResponse response = new MockHttpServletResponse();
        
        // 实例化待测试的Servlet
        TestServlet servlet = new TestServlet();
        servlet.doGet(request, response);
        
        // 验证响应结果
        System.out.println("响应状态: " + response.getStatus());
        System.out.println("响应内容:\n" + response.getContentAsString());
    }

    // 模拟HttpServletRequest
    static class MockHttpServletRequest implements HttpServletRequest {
        private final Map<String, String> params = new HashMap<>();
        
        public void setParameter(String name, String value) {
            params.put(name, value);
        }
        
        @Override
        public String getParameter(String name) {
            return params.get(name);
        }
        
        // 其他必要方法存根（简化实现）
        @Override public String getMethod() { return "GET"; }
        @Override public String getRequestURI() { return "/test"; }
    }

    // 模拟HttpServletResponse
    static class MockHttpServletResponse implements HttpServletResponse {
        private final StringWriter writer = new StringWriter();
        private int status = 200;
        
        @Override
        public PrintWriter getWriter() {
            return new PrintWriter(writer);
        }
        
        @Override
        public void setStatus(int status) {
            this.status = status;
        }
        
        public int getStatus() {
            return status;
        }
        
        public String getContentAsString() {
            return writer.toString();
        }
        
        // 其他必要方法存根
        @Override public void setContentType(String type) {}
        @Override public void addHeader(String name, String value) {}
    }
}

3. 执行结果

响应状态: 200
响应内容:
<h1>Hello, Alice!</h1>

关键实现说明

1. 请求模拟：

   • 自定义MockHttpServletRequest实现参数存储
   • 通过setParameter()注入测试数据

2. 响应捕获：

   • MockHttpServletResponse使用StringWriter捕获输出
   • getContentAsString()方法获取响应内容

3. 生命周期模拟：

   • 直接调用servlet.doGet()方法
   • 绕过容器初始化流程

4. 验证方式：

   • 检查HTTP状态码
   • 验证输出内容格式
   • 可扩展异常场景测试

注意：此方法适用于业务逻辑测试，但无法测试容器相关特性（如Filter链、Session管理等）。完整集成测试仍需使用嵌入式容器（如Jetty/Tomcat Embedded）。