Mockito Spy实战：高效测试调用内部方法的Java函数

import java.time.LocalDate;

public class CalculateDateService {

    // 假设这里可能有一个 DateService 的依赖，但在本示例中我们简化它
    // private DateService dateService;

    LocalDate rollDays(int days, LocalDate date, String mar) {
        int dir = days > 0 ? 1 : -1;
        // 循环条件中调用了 closed 方法
        for(int numOfDays = Math.abs(days); closed(date, mar) || numOfDays-- > 0; date = date.plusDays((long) dir)) {
            // 循环体为空，仅根据条件推进日期
        }
        return date;
    }

    Boolean closed(LocalDate date, String mar){
        // closed 方法内部调用了 nonSettle 方法
        // 简化：这里不再依赖外部dateService，仅依赖nonSettle
        return this.nonSettle(date);
    }

    Boolean nonSettle(LocalDate date) {
        int year = LocalDate.now().getYear(); // 假设这里获取当前年份
        LocalDate dayBeforeChristmas = LocalDate.of(year, 12, 24);
        LocalDate dayBeforenewYear = LocalDate.of(year, 12, 31);
        return (date.isEqual(dayBeforeChristmas) || date.isEqual(dayBeforenewYear));
    }
}

@Mock
private DateService dateService; // 假设存在外部依赖
@InjectMocks
private CalculateDateService tested;

@Test
public void testRollDaysInitialAttempt() {
    LocalDate initDate = LocalDate.of(2022, 12, 22);

    // 尝试直接在 @InjectMocks 对象上 stub 内部方法
    // when(tested.Closed(initDate, dateService.marID)).thenReturn(Boolean.FALSE); // 这将导致问题
    // when(tested.NonSettle(initDate)).thenReturn(Boolean.FALSE); // 这也将导致问题

    // ... 其他测试逻辑
}

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>4.x.x</version> <!-- 使用最新稳定版本 -->
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.junit.jupiter</groupId>
    <artifactId>junit-jupiter-api</artifactId>
    <version>5.x.x</version> <!-- 使用最新稳定版本 -->
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.junit.jupiter</groupId>
    <artifactId>junit-jupiter-engine</artifactId>
    <version>5.x.x</version> <!-- 使用最新稳定版本 -->
    <scope>test</scope>
</dependency>

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import java.time.LocalDate;

public class CalculateDateServiceTest {

    // 示例一：测试 rollDays 在 nonSettle 始终为 false 的情况下
    @Test
    void testSpyFalseRollByDays() {
        var year = LocalDate.now().getYear();
        // 创建 CalculateDateService 的 Spy 对象
        var spy = Mockito.spy(new CalculateDateService());

        // 存根 nonSettle 方法，使其对任何 LocalDate 参数都返回 false
        // 使用 doReturn().when() 避免真实方法被调用
        Mockito.doReturn(false).when(spy).nonSettle(Mockito.any(LocalDate.class));

        // 调用 rollDays 方法，起始日期为当年12月1日，滚动24天
        var result = spy.rollDays(24, LocalDate.of(year, 12, 1), "");

        // 预期结果：12月1日 + 24天 = 12月25日 (因为 nonSettle 始终为 false，没有跳过日期)
        Assertions.assertEquals(LocalDate.of(year, 12, 25), result);
    }

    // 示例二：测试 rollDays 在特定日期 nonSettle 为 true 的情况下
    @Test
    void testSpyTrueOnSpecificDateRollByDays() {
        var year = LocalDate.now().getYear();
        var spy = Mockito.spy(new CalculateDateService());

        // 默认情况下，nonSettle 对任何日期都返回 false
        Mockito.doReturn(false).when(spy).nonSettle(Mockito.any(LocalDate.class));
        // 特别地，存根 nonSettle 方法，使其在 12月24日 返回 true (模拟非结算日)
        Mockito.doReturn(true).when(spy).nonSettle(LocalDate.of(year, 12, 24));

        // 调用 rollDays 方法，起始日期为当年12月1日，滚动24天
        var result = spy.rollDays(24, LocalDate.of(year, 12, 1), "");

        // 预期结果：
        // 从 12月1日 开始滚动24天。
        // 正常情况下会到 12月25日。
        // 但由于 12月24日 是 nonSettle 日期，在循环中 nonSettle(12月24日) 会返回 true，
        // 导致 12月24日 这一天被跳过，实际天数计数器 numOfDays 不会减少。
        // 因此，需要额外一天来完成24天的滚动，最终日期是 12月26日。
        Assertions.assertEquals(LocalDate.of(year, 12, 26), result);
    }
}