Mockito测试同一方法重复调用：覆盖Student构造器分支

看起来你遇到了Mockito中针对同一方法不同参数调用做分支覆盖的典型问题——因为两次调用的是同一个方法，稍不注意参数匹配就会搞混，导致第二个分支始终触发不了。别担心，咱们一步步来解决这个问题。

首先先明确你的核心场景：你有两个无法修改的类Person和Student，Student构造器里两次调用person.getInfo()（分别传"firstName"和"bio"），并针对返回值做非空校验抛出SomethingError。你需要编写两个测试用例分别覆盖这两个if分支，但目前第二个测试总是触发第一个分支，或者自定义的Answer逻辑没执行。

核心类代码 #

Person类 #

public class Person {
    public String getInfo(String key) {
        return key; // 简化实现
    }
}

Student类（无法修改） #

public class Student {
    private String copyFirstName;
    private String otherInfo;
    private Person person;

    Student(Person p) {
        person = p;
        copyFirstName = person.getInfo("firstName");
        if (copyFirstName == null || copyFirstName.equals("")) { // entry point A
            throw new SomethingError("message here");
        }
        otherInfo = person.getInfo("bio");
        if (otherInfo == null || otherInfo.equals("")) { // entry point B
            throw new SomethingError("message here");
        }
    }
}

// 假设SomethingError是自定义异常
class SomethingError extends RuntimeException {
    public SomethingError(String message) {
        super(message);
    }
}

问题原因分析 #

你的测试没达到预期，大概率是以下两个原因：

• Mock参数匹配不精准：如果用了模糊的参数匹配（比如anyString()），或者没明确区分"firstName"和"bio"的返回值，会导致构造器里的第一个调用就触发异常，根本走不到第二个分支。
• Mock实例未隔离：如果多个测试共用同一个Person Mock实例，前一个测试的设置会污染后一个测试的执行逻辑。

正确的测试实现方案 #

我们需要为每个测试用例单独设置Mock行为，明确区分不同参数的返回值，同时确保每个测试用例使用独立的Mock实例。下面是JUnit 5 + Mockito的实现示例：

测试类代码 #

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
class StudentTest {

    // 每个测试方法会自动获得全新的Mock实例，避免测试间状态污染
    @Mock
    private Person personMock;

    // 测试覆盖entry point A：firstName为空/Null的情况
    @Test
    void shouldThrowErrorWhenFirstNameIsEmpty() {
        // 设置：调用getInfo("firstName")时返回空字符串，触发第一个分支
        when(personMock.getInfo("firstName")).thenReturn("");
        // 验证是否抛出预期异常
        assertThrows(SomethingError.class, () -> new Student(personMock));
    }

    // 测试覆盖entry point B：bio为空/Null的情况
    @Test
    void shouldThrowErrorWhenBioIsEmpty() {
        // 第一步：让firstName返回非空，跳过第一个分支的校验
        when(personMock.getInfo("firstName")).thenReturn("Alice");
        // 第二步：让bio返回null，触发第二个分支
        when(personMock.getInfo("bio")).thenReturn(null);
        // 验证是否抛出预期异常
        assertThrows(SomethingError.class, () -> new Student(personMock));
    }
}

关键细节说明 #

1. 独立Mock实例：使用@ExtendWith(MockitoExtension.class)和@Mock注解，Mockito会为每个测试方法创建全新的Person Mock实例，彻底避免测试间的状态干扰。
2. 精准参数匹配：明确指定不同参数对应的返回值，确保构造器中两次getInfo调用的行为完全符合测试预期。
3. 异常验证：用assertThrows来验证构造器是否抛出了预期的SomethingError，这是JUnit 5中验证异常的标准方式。

如果你更倾向于用Answer实现复杂的参数判断逻辑，也可以这样写：

// 用Answer定制返回逻辑的版本
when(personMock.getInfo(anyString())).thenAnswer(invocation -> {
    String key = invocation.getArgument(0);
    if ("firstName".equals(key)) {
        return "Bob"; // 非空，跳过第一个分支
    } else if ("bio".equals(key)) {
        return ""; // 空，触发第二个分支
    }
    return null;
});

这样就能保证两个测试用例分别覆盖对应的if分支，实现代码行全覆盖了。

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