如何用JUnit测试实现文件输入重定向的BufferedReader代码？

你的代码直接依赖System.in，耦合度太高，没法在测试中灵活替换输入源，这是测试难的核心问题。最佳方案是先解耦输入依赖，再针对性编写JUnit测试。

第一步：改造原代码，解耦输入流 #

把readFile方法改成接受InputStream参数，生产环境传入System.in，测试时传入模拟的输入流。同时用try-with-resources自动关闭流，避免手动关闭的疏漏，还要尽量避免静态变量带来的测试状态污染：

public class YourClassName {
    // 去掉静态request，改用局部变量，避免测试间状态残留
    public static void readFile(InputStream inputStream) {
        try (BufferedReader input = new BufferedReader(new InputStreamReader(inputStream))) {
            String request;
            while ((request = input.readLine()) != null) {
                System.out.println("Client request: " + request);
            }
            // try-with-resources会自动关闭input，无需手动调用close()
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        readFile(System.in); // 生产环境传入标准输入流
    }
}

第二步：编写JUnit测试（以JUnit 5为例） #

测试时用ByteArrayInputStream模拟输入内容，用ByteArrayOutputStream捕获System.out的输出，验证打印内容是否符合预期，同时要注意恢复原输出流，避免影响其他测试：

import org.junit.jupiter.api.Test;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

class YourClassNameTest {

    @Test
    void testReadFileWithValidRequests() {
        // 模拟多行输入
        String mockInput = "登录请求\n查询请求\n退出请求";
        ByteArrayInputStream mockInputStream = new ByteArrayInputStream(mockInput.getBytes());

        // 捕获System.out输出
        ByteArrayOutputStream outputCaptor = new ByteArrayOutputStream();
        PrintStream originalOut = System.out;
        System.setOut(new PrintStream(outputCaptor));

        try {
            YourClassName.readFile(mockInputStream);

            // 验证输出是否符合预期
            String expectedOutput = "Client request: 登录请求\nClient request: 查询请求\nClient request: 退出请求\n";
            assertEquals(expectedOutput, outputCaptor.toString());
        } finally {
            // 恢复原输出流，防止影响其他测试
            System.setOut(originalOut);
        }
    }

    @Test
    void testReadFileWithEmptyInput() {
        ByteArrayInputStream mockInputStream = new ByteArrayInputStream("".getBytes());
        ByteArrayOutputStream outputCaptor = new ByteArrayOutputStream();
        PrintStream originalOut = System.out;
        System.setOut(new PrintStream(outputCaptor));

        try {
            YourClassName.readFile(mockInputStream);
            // 空输入无输出
            assertEquals("", outputCaptor.toString());
        } finally {
            System.setOut(originalOut);
        }
    }

    @Test
    void testReadFileWithException() {
        // 传入null模拟异常场景
        ByteArrayOutputStream outputCaptor = new ByteArrayOutputStream();
        PrintStream originalOut = System.out;
        System.setOut(new PrintStream(outputCaptor));

        try {
            YourClassName.readFile(null);
            // 异常会打印堆栈信息，验证输出包含异常关键字
            assertTrue(outputCaptor.toString().contains("NullPointerException"));
        } finally {
            System.setOut(originalOut);
        }
    }
}

核心要点总结 #

• 解耦是基础：不要让代码直接依赖System.in这类全局资源，通过参数注入的方式让依赖可替换，这是可测试性的关键。
• 捕获输出验证行为：因为你的代码逻辑包含打印输出，测试时必须捕获并验证输出内容，确保行为符合预期。
• 清理测试环境：修改System.out这类全局状态后，一定要在测试结束后恢复，避免测试用例之间互相干扰。
• 覆盖多场景：除了正常输入，还要测试空输入、异常输入等边界场景，保证代码的健壮性。

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