使用 Mockito 模拟 Joda DateTime 方法

2022-01-14 00:00:00 mockito java powermock

我希望 millis 返回指定的值.

I want millis to return specified value.

public long myMethod(){
    DateTime nowDateTime = new DateTime(DateTimeZone.UTC);
    long millis = nowDateTime.getMillis();
    System.out.println(millis);
}

我试过了,但没有运气.

I tried this with no luck.

@RunWith(PowerMockRunner.class)
@PrepareForTest({ DateTime.class })
@PowerMockIgnore({ "javax.crypto.*", "javax.management*" })
...
...
public void testMyMethod(){
    DateTime nowDateTime = PowerMockito.mock(DateTime.class);
    Mockito.when(nowDateTime.getMillis()).thenReturn(10L);
}

我该如何解决这个问题?

How can I fix this?

推荐答案

只需使用 JodaTime 的 org.joda.time.DateTimeUtils#setCurrentMillisFixed 方法,该方法旨在修复 new DateTime() 到与当前时间不同的时间.要返回正常时间,请稍后使用 org.joda.time.DateTimeUtils#setCurrentMillisSystem.无需嘲笑.

Just use the org.joda.time.DateTimeUtils#setCurrentMillisFixed method of JodaTime which was designed to fix new DateTime() to a different time than the current time. To return to the normal time use org.joda.time.DateTimeUtils#setCurrentMillisSystem afterwards. No mocking needed.


@Test
public void test() {
DateTimeUtils.setCurrentMillisFixed(10L);
// .. your code
}

@After
public void cleanup() {
// Make sure to cleanup afterwards
DateTimeUtils.setCurrentMillisSystem()
}

相关文章