java - JUnit Java 即时依赖测试

Question

我使用 Instant.now() 获取当前 UTC 毫秒，然后将其截断为最接近的小时。我的所有 JUnit 都失败了，因为它们占用了当前的系统时间。如何使 Instant.now() 返回一个可以在 JUnit 测试中提供的固定值。

public static Long getCurrentHour() {
    Instant now = Instant.now();
    Instant cH = now.truncatedTo(ChronoUnit.HOURS);
    return cH.toEpochMilli();
}

Answer 1

您应该模拟静态方法 Instant.now() 来为您提供静态即时值。
您可以使用 PowerMockito 来实现此目的。

@RunWith(PowerMockRunner.class)
@PrepareForTest(Instant.class)
public class TestClass {
    @Mock private Instant mockInstant;

    @Test
    public void getCurrentHour() throws Exception {
        PowerMockito.mockStatic(Instant.class);
        when(Instant.now()).thenReturn(mockInstant);
        when(mockInstant.truncatedTo(ChronoUnit.HOURS)).thenReturn(mockInstant);
        long expectedMillis = 999l;
        when(mockInstant.toEpochMilli()).thenReturn(expectedMillis);

        assertEquals(expectedMillis, YourClass.getCurrentHour());
    }
}