Simular la primera llamada falla, la segunda llamada tiene éxito

Question

Quiero usar Mockito para probar el código (simplificado) a continuación. No sé cómo decirle a Mockito que falle la primera vez, y luego triunfar la segunda vez.

for(int i = 1; i < 3; i++) { 
    String ret = myMock.doTheCall(); 

    if("Success".equals(ret)) { 
    log.write("success"); 
    } else if (i < 3) { 
    log.write("failed, but I'll try again. attempt: " + i); 
    } else { 
    throw new FailedThreeTimesException(); 
    } 
} 

puedo configurar la prueba de éxito con:

Mockito.when(myMock).doTheCall().thenReturn("Success"); 

Y la prueba de fallos con:

Mockito.when(myMock).doTheCall().thenReturn("you failed"); 

Pero, ¿cómo puedo probar que si falla una vez (o dos veces) y luego tiene éxito, está bien?

Answer 1

De the docs:

veces tenemos que stub con diferente valor de retorno/excepción para la misma llamada al método. El caso de uso típico podría ser iteradores burlones. La versión original de Mockito no tenía esta función para promover la burla simple. Por ejemplo, en lugar de iteradores, uno podría usar Iterable o simplemente colecciones. Esos ofrecen formas naturales de stubbing (por ejemplo, utilizando colecciones reales). En escenarios poco comunes stubbing llamadas consecutivas podría ser útil, sin embargo:

when(mock.someMethod("some arg")) 
    .thenThrow(new RuntimeException()) 
    .thenReturn("foo"); 

//First call: throws runtime exception: 
mock.someMethod("some arg"); 

//Second call: prints "foo" 
System.out.println(mock.someMethod("some arg")); 

Así, en su caso, que querría:

when(myMock.doTheCall()) 
    .thenReturn("You failed") 
    .thenReturn("Success"); 

Answer 2

El camino más corto para escribir lo que desea es

when(myMock.doTheCall()).thenReturn("Success", "you failed"); 

Cuando proporciona argumentos mutiples a thenReturn de esta manera, cada argumento se utilizará a lo sumo una vez, excepto el último argumento, w Que se usa tantas veces como sea necesario. Por ejemplo, en este caso, si realiza la llamada 4 veces, obtendrá "Éxito", "falló", "falló", "falló".

Answer 3

Dado que el comentario que se refiere a esto es difícil de leer. Añadiré una respuesta formateada.

Si está intentando hacer esto con una función anulada. Donde podría querer una excepción que ningún comportamiento. De allí tendría que hacer algo como esto:

Mockito.doThrow(new Exception("MESSAGE")) 
      .doNothing() 
      .when(mockService).method(eq()); 

Answer 4

para añadir a this y this respuesta, también se puede utilizar un bucle de cadena de las llamadas burlado. Esto es útil si necesita burlarse de lo mismo varias veces, o se burla de algún patrón.

ejemplo (aunque sea inverosímil):

import org.mockito.stubbing.Stubber; 

Stubber stubber = doThrow(new Exception("Exception!")); 
for (int i=0; i<10; i++) { 
    if (i%2 == 0) { 
     stubber.doNothing(); 
    } else { 
     stubber.doThrow(new Exception("Exception")); 
    } 
} 
stubber.when(myMockObject).someMethod(anyString());