Servlet环境下Java代码测试：无需System.setIn()注入System.in测试用例的方法

方法1：修改待测试代码的输入依赖（推荐，简单直接） #

如果允许调整待测试代码的输入逻辑，直接把依赖System.in的部分改成可传入InputStream的形式：

• 原代码示例：

public class TestCode {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String input = sc.nextLine();
        System.out.println("Input: " + input);
    }
}

• 修改后代码：

public class TestCode {
    public static void main(String[] args) {
        // 保留原入口，兼容正常运行
        run(System.in);
    }

    // 新增测试专用入口，接受自定义输入流
    public static void run(InputStream in) {
        Scanner sc = new Scanner(in);
        String input = sc.nextLine();
        System.out.println("Input: " + input);
    }
}

测试时直接调用run(new ByteArrayInputStream("测试用例".getBytes()))，完全不依赖全局的System.in，天然支持多线程并发。

如果无法手动修改用户提交的代码，可以用JavaParser或ASM等工具在内存编译前自动注入上述改造逻辑：解析代码找到读取System.in的节点，替换为从参数或指定变量获取输入流。

方法2：用ThreadLocal隔离线程输入流（无需修改原代码结构） #

通过字节码修改，把待测试代码中所有System.in的引用替换为从ThreadLocal获取的流，实现线程级隔离：

1. 定义一个存储输入流的ThreadLocal类：

public class TestInputHolder {
    private static final ThreadLocal<InputStream> INPUT_STREAM = new ThreadLocal<>();

    public static InputStream get() {
        return INPUT_STREAM.get();
    }

    public static void set(InputStream in) {
        INPUT_STREAM.set(in);
    }

    public static void remove() {
        INPUT_STREAM.remove();
    }
}

2. 在内存编译待测试代码时，用ASM工具遍历字节码，将所有getstatic java/lang/System.in指令替换为invokestatic TestInputHolder.get()Ljava/io/InputStream;
3. 测试流程：
   • 调用TestInputHolder.set(new ByteArrayInputStream("测试用例".getBytes()))
   • 执行待测试代码逻辑
   • 测试完成后调用TestInputHolder.remove()清理线程变量

方法3：启动独立进程运行测试（完全隔离环境） #

把内存编译好的类写入临时文件，通过ProcessBuilder启动独立Java进程，利用进程的标准输入传递测试用例：

// 假设已将编译后的类写入tempDir目录
ProcessBuilder pb = new ProcessBuilder("java", "-cp", tempDir.getAbsolutePath(), "TestCode");
Process process = pb.start();

// 向进程输入测试用例
try (OutputStream out = process.getOutputStream()) {
    out.write("测试用例".getBytes());
    out.flush();
}

// 读取进程输出
String output = new String(process.getInputStream().readAllBytes());
int exitCode = process.waitFor();

每个测试请求对应独立进程，完全避免多线程共享资源的问题，缺点是进程启动会带来一定性能开销，适合对隔离性要求极高的场景。

内容的提问来源于stack exchange，提问作者Vimal Arjun