Assert call to method using Mock python

Question

Estoy tratando de hacer algunas pruebas unitarias utilizando la biblioteca de simulacro en Python. Tengo el siguiente código:

def a(): 
    print 'a' 

def b(): 
    print 'b' 
    if some condition 
     a() 

¿Cómo afirmar que una llamada de b se ha hecho cuando se ha hecho una llamada fingida a b? He probado el siguiente código, pero ha fallado:

mymock=Mock() 
mymock.b() 
assertTrue(a.__call__ in mymock.mock_calls) 

Por alguna razón, creo que la mymock.b() no tiene nada que ver con el método b(). ¿Qué se puede hacer por esto?

Answer 1

Si un parche a, puede asegurarse de que se llamaba así:

with mock.patch('__main__.a') as fake_a: 
    b() 
    fake_a.assert_called_with() 

Si el método está en un módulo diferente:

import mymodule 

with mock.patch('mymodule.a') as fake_a: 
    mymodule.b() 
    fake_a.assert_called_with() 

Answer 2

Somehow, I think that the mymock.b() has nothing to do with the method b() What can be done for this?

Tienes razón. Cuando te burlas de un objeto, estás dando a entender que no te importa lo que hace tu burla detrás de escena. Si quería asegurarse de que se llamara a desde b, querría patcha en b.

>>> from mock import patch 
>>> with patch('__main__.a') as patch_a: 
...  b() 
...  patch_a.assert_called_with() 

Así, la moraleja de la historia es, simulada o parche sobre el objeto que se desea medir sin tener que llamar, no el objeto cuya aplicación le interesan. En este caso, usted se preocupa por b y desea saber cómo se usa a. Como no nos importa lo que haga a, solo que se está llamando, podemos parcharlo.

Además, si desea más detalles sobre las llamadas a a, en lugar de assert_called_with puede analizar todas las llamadas realizadas accediendo al atributo parches mock_calls. En este caso sería patch_a.mock_calls.