java - 使用Mockito模拟Joda DateTime方法

我希望millis返回指定值。

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

我没有运气尝试过。

@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);
}

我怎样才能解决这个问题？

最佳答案

只需使用JodaTime的org.joda.time.DateTimeUtils#setCurrentMillisFixed方法即可，该方法旨在将new DateTime()固定为与当前时间不同的时间。要恢复正常时间，请稍后使用org.joda.time.DateTimeUtils#setCurrentMillisSystem。无需 mock 。

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

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