Jak wyśmiać nową datę() w java przy użyciu Mockito

Question

Mam funkcję, która wykorzystuje bieżący czas do wykonywania niektórych obliczeń. Chciałbym wyśmiać go za pomocą mockito.

Przykładem klasy Chciałbym przetestować:

public class ClassToTest { 
    public long getDoubleTime(){ 
     return new Date().getTime()*2; 
    } 
} 

Chciałbym coś takiego:

@Test 
public void testDoubleTime(){ 
    mockDateSomeHow(Date.class).when(getTime()).return(30); 
    assertEquals(60,new ClassToTest().getDoubleTime()); 
} 

to możliwe, aby drwić, że? Nie chciałbym zmieniać "przetestowanego" kodu w celu przetestowania.

Answer 1

Prawidłową rzeczą jest zrestrukturyzowanie kodu tak, aby był bardziej testowalny, jak pokazano poniżej. Restrukturyzacja swój kod, aby usunąć bezpośrednią zależność Data pozwoli Ci wstrzyknąć różne implementacje dla prawidłowego wykonywania i testowego środowiska wykonawczego:

interface DateTime { 
    Date getDate(); 
} 

class DateTimeImpl implements DateTime { 
    @Override 
    public Date getDate() { 
     return new Date(); 
    } 
} 

class MyClass { 

    private final DateTime dateTime; 
    // inject your Mock DateTime when testing other wise inject DateTimeImpl 

    public MyClass(final DateTime dateTime) { 
     this.dateTime = dateTime; 
    } 

    public long getDoubleTime(){ 
     return dateTime.getDate().getTime()*2; 
    } 
} 

public class MyClassTest { 
    private MyClass myClassTest; 

    @Before 
    public void setUp() { 
     final Date date = Mockito.mock(Date.class); 
     Mockito.when(date.getTime()).thenReturn(30L); 

     final DateTime dt = Mockito.mock(DateTime.class); 
     Mockito.when(dt.getDate()).thenReturn(date); 

     myClassTest = new MyClass(dt); 
    } 

    @Test 
    public void someTest() { 
     final long doubleTime = myClassTest.getDoubleTime(); 
     assertEquals(60, doubleTime); 
    } 
} 

Answer 2

Ty mógłby zrobić to za pomocą PowerMock, która zwiększa Mockito móc szydzić metody statyczne. Możesz wtedy mock System.currentTimeMillis(), od której ostatecznie otrzymasz czas.

Możesz może. Nie zamierzam wysuwać opinii, czy należy powinno być.

Answer 3

Jeśli masz starszego kodu, który nie może byłaby i nie chcesz, aby wpłynąć System.currentTimeMillis(), spróbuj to za pomocą Powermock i PowerMockito

//note the static import 
import static org.powermock.api.mockito.PowerMockito.whenNew; 

@PrepareForTest({ LegacyClassA.class, LegacyClassB.class }) 

@Before 
public void setUp() throws Exception { 

    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); 
    sdf.setTimeZone(TimeZone.getTimeZone("PST")); 

    Date NOW = sdf.parse("2015-05-23 00:00:00"); 

    // everytime we call new Date() inside a method of any class 
    // declared in @PrepareForTest we will get the NOW instance 
    whenNew(Date.class).withNoArguments().thenReturn(NOW); 

} 

public class LegacyClassA { 
    public Date getSomeDate() { 
    return new Date(); //returns NOW 
    } 
} 

Answer 4

Jedno podejście, które nie bezpośrednio odpowiedzieć na pytanie, ale może rozwiązać bazowego problem (posiadający powtarzalne testy), pozwala na Date jako parametr do testów i dodaje delegata do domyślnej daty.

Podobnie jak

public class ClassToTest { 

    public long getDoubleTime() { 
     return getDoubleTime(new Date()); 
    } 

    long getDoubleTime(Date date) { // package visibility for tests 
     return date.getTime() * 2; 
    } 
} 

W produkcji kodu, należy użyć getDoubleTime() i próby przed getDoubleTime(Date date).