JUnit测试中重定向System.in进行输入测试

// InputProcessor.java
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class InputProcessor {

    /**
     * 从System.in读取输入，并将每一行作为“客户端请求”打印到System.out。
     * 异常处理仅为示例，实际应用中应更健壮。
     */
    public static void processSystemIn() {
        // 使用try-with-resources确保BufferedReader自动关闭
        try (BufferedReader input = new new InputStreamReader(System.in))) {
            String line;
            while ((line = input.readLine()) != null) {
                System.out.println("Client request: " + line);
            }
        } catch (IOException e) {
            System.err.println("Error reading from System.in: " + e.getMessage());
            e.printStackTrace();
        }
    }
}

// InputProcessorTest.java
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.PrintStream;
import java.io.UnsupportedEncodingException;
import java.nio.charset.StandardCharsets;

public class InputProcessorTest {

    // 用于保存原始的System.in和System.out
    private static InputStream originalSystemIn;
    private static PrintStream originalSystemOut;

    // 用于捕获System.out的输出
    private static ByteArrayOutputStream outputStreamCaptor;

    /**
     * 在所有测试方法执行前执行一次。
     * 用于保存原始的System.in和System.out，并设置输出捕获器。
     */
    @BeforeAll
    static void setupAll() {
        originalSystemIn = System.in;
        originalSystemOut = System.out;
        outputStreamCaptor = new ByteArrayOutputStream();
        System.setOut(new PrintStream(outputStreamCaptor, true, StandardCharsets.UTF_8));
    }

    /**
     * 在所有测试方法执行后执行一次。
     * 用于恢复原始的System.in和System.out。
     */
    @AfterAll
    static void tearDownAll() {
        System.setIn(originalSystemIn);
        System.setOut(originalSystemOut);
    }

    /**
     * 测试processSystemIn方法，模拟多行输入。
     *
     * @throws UnsupportedEncodingException 如果UTF-8编码不支持
     */
    @Test
    void testProcessSystemInWithMultipleLines() throws UnsupportedEncodingException {
        // 1. 准备测试输入数据
        String testInput = "Line 1\nLine 2\nLine 3";
        ByteArrayInputStream testInputStream = new ByteArrayInputStream(testInput.getBytes(StandardCharsets.UTF_8));

        // 2. 重定向System.in到模拟输入流
        System.setIn(testInputStream);

        // 3. 调用被测方法
        InputProcessor.processSystemIn();

        // 4. 验证输出
        String expectedOutput = "Client request: Line 1" + System.lineSeparator() +
                                "Client request: Line 2" + System.lineSeparator() +
                                "Client request: Line 3" + System.lineSeparator();

        // 获取并清空捕获的输出流，以便下一个测试方法使用
        String actualOutput = outputStreamCaptor.toString(StandardCharsets.UTF_8.name());
        outputStreamCaptor.reset(); // 清空，确保每个测试都是独立的

        assertEquals(expectedOutput, actualOutput, "System.out的输出应与预期匹配");
    }

    /**
     * 测试processSystemIn方法，模拟空输入。
     *
     * @throws UnsupportedEncodingException 如果UTF-8编码不支持
     */
    @Test
    void testProcessSystemInWithEmptyInput() throws UnsupportedEncodingException {
        // 1. 准备测试输入数据 (空字符串)
        String testInput = "";
        ByteArrayInputStream testInputStream = new ByteArrayInputStream(testInput.getBytes(StandardCharsets.UTF_8));

        // 2. 重定向System.in到模拟输入流
        System.setIn(testInputStream);

        // 3. 调用被测方法
        InputProcessor.processSystemIn();

        // 4. 验证输出 (应为空)
        String expectedOutput = "";
        String actualOutput = outputStreamCaptor.toString(StandardCharsets.UTF_8.name());
        outputStreamCaptor.reset();

        assertEquals(expectedOutput, actualOutput, "System.out的输出应为空");
    }
}