Mockito单元测试：如何正确Mock内部创建的依赖

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

public class View {
    private final BufferedReader reader;

    public View() {
        // 问题所在：BufferedReader在此处被内部实例化
        this.reader = new BufferedReader(new InputStreamReader(System.in));
    }

    public void inputCommand() throws IOException {
        String line;
        System.out.println("Waiting for command..."); // 添加提示，更清晰
        while ((line = reader.readLine()) != null) {
            System.out.println("Received command: " + line);
            if ("Q".equals(line.trim())) { // trim() 避免空白字符影响判断
                System.out.println("Exiting command input.");
                break;
            }
            // 可以在此处添加处理命令的逻辑
        }
    }
}

import org.junit.jupiter.api.Test;
import java.io.BufferedReader;
import java.io.IOException;
import static org.mockito.Mockito.*;

public class ViewTestFails {

    @Test
    void whenCommandIsR_CallBuildRectangle_Fails() throws IOException {
        BufferedReader bufferedReader = mock(BufferedReader.class);
        // 尝试为mock对象设置行为
        when(bufferedReader.readLine()).thenReturn("C 10 10").thenReturn("Q");

        // 此时View内部创建了它自己的BufferedReader实例，而非我们mock的实例
        View view = new View(); 
        view.inputCommand();

        // 这里的verify将失败，因为View使用的是内部实例，而非我们mock的实例
        // 并且测试会阻塞，等待System.in的真实输入
        verify(bufferedReader, times(2)).readLine();
    }
}

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

public class View {
    private final BufferedReader reader;

    // 修改为构造函数注入：View类不再创建BufferedReader，而是接收它
    public View(BufferedReader reader) {
        this.reader = reader;
    }

    // 可选：提供一个无参构造函数，用于生产环境，以保持与旧代码的兼容性
    // 或者当View总是需要从System.in读取时
    public View() {
        this(new BufferedReader(new InputStreamReader(System.in)));
    }

    public void inputCommand() throws IOException {
        String line;
        System.out.println("Waiting for command...");
        while ((line = reader.readLine()) != null) {
            System.out.println("Received command: " + line);
            if ("Q".equals(line.trim())) { 
                System.out.println("Exiting command input.");
                break;
            }
            // 实际应用中，这里会包含处理用户输入的业务逻辑
        }
    }
}

import org.junit.jupiter.api.Test;
import java.io.BufferedReader;
import java.io.IOException;

import static org.mockito.Mockito.*;
import static org.junit.jupiter.api.Assertions.*; // 引入断言，用于更全面的测试

public class ViewTest {

    @Test
    void whenCommandIsC_ThenProcessCommandAndExit() throws IOException {
        // 1. 创建Mock对象
        BufferedReader mockBufferedReader = mock(BufferedReader.class);

        // 2. 定义Mock对象的行为
        // 第一次调用readLine()返回"C 10 10"，第二次返回"Q"
        when(mockBufferedReader.readLine())
                .thenReturn("C 10 10")
                .thenReturn("Q"); // 模拟用户输入"Q"退出

        // 3. 将Mock对象注入到被测试类中
        View view = new View(mockBufferedReader);

        // 4. 执行被测试方法
        view.inputCommand();

        // 5. 验证Mock对象的行为
        // 验证readLine()方法被调用了两次
        verify(mockBufferedReader, times(2)).readLine();

        // 可以在此处添加更多断言，例如验证View类内部的业务逻辑是否正确执行
        // 假设View有一个方法可以获取处理过的命令列表，可以这样验证：
        // assertEquals(Arrays.asList("C 10 10"), view.getProcessedCommands());
    }

    @Test
    void whenCommandIsQImmediately_ThenExit() throws IOException {
        BufferedReader mockBufferedReader = mock(BufferedReader.class);
        // 模拟用户直接输入"Q"
        when(mockBufferedReader.readLine()).thenReturn("Q");

        View view = new View(mockBufferedReader);
        view.inputCommand();

        // 验证readLine()方法只被调用了一次
        verify(mockBufferedReader, times(1)).readLine();
    }
}