使用Mockito模拟Joda DateTime方法

Question

我想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);
}

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

Answer 1

只需使用org.joda.time.DateTimeUtils#setCurrentMillisFixedJodaTime 的方法,该方法旨在修复new DateTime()与当前时间不同的时间.恢复到正常时间org.joda.time.DateTimeUtils#setCurrentMillisSystem后再使用.不需要嘲笑.

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

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