Python burlarse de una función de un módulo importado

title in python plot

Question

Quiero entender cómo poner en @patch una función de un módulo importado.

Aquí es donde estoy hasta ahora.

app / mocking.py:

from app.my_module import get_user_name
def test_method():
return get_user_name()
if __name__ == "__main__":
print "Starting Program..."
test_method()

app / my_module / __ init__.py:

def get_user_name():
return "Unmocked User"

test / mock-test.py:

import unittest
from app.mocking import test_method
def mock_get_user():
return "Mocked This Silly"
@patch(''app.my_module.get_user_name'')
class MockingTestTestCase(unittest.TestCase):
def test_mock_stubs(self, mock_method):
mock_method.return_value = ''Mocked This Silly'')
ret = test_method()
self.assertEqual(ret, ''Mocked This Silly'')
if __name__ == ''__main__'':
unittest.main()

Esto no funciona como esperaría. El módulo "parcheado" simplemente devuelve el valor no get_user_name de get_user_name . ¿Cómo me burlo de los métodos de otros paquetes que estoy importando en un espacio de nombres bajo prueba?

Answer 1

Cuando está utilizando el decorador de patch del paquete unittest.mock no está aplicando parches al espacio de nombres desde el que se importa el módulo (en este caso, app.my_module.get_user_name ), lo está app.my_module.get_user_name en el espacio de nombres bajo la prueba app.mocking.get_user_name .

Para hacer lo anterior con Mock prueba algo como lo siguiente:

from mock import patch
from app.mocking import test_method
class MockingTestTestCase(unittest.TestCase):
@patch(''app.mocking.get_user_name'')
def test_mock_stubs(self, test_patch):
test_patch.return_value = ''Mocked This Silly''
ret = test_method()
self.assertEqual(ret, ''Mocked This Silly'')

La documentación de la biblioteca estándar incluye una section útil section describe esto.

Answer 2

Mientras que la respuesta de Matti John resuelve su problema (y me ayudó también, ¡gracias!), Yo sugeriría que se localice el reemplazo de la función original ''get_user_name'' por la que se burló. Esto le permitirá controlar cuándo se reemplaza la función y cuándo no. Además, esto le permitirá realizar varios reemplazos en la misma prueba. Para hacerlo, use la declaración ''con'' de una manera bastante similar:

from mock import patch
class MockingTestTestCase(unittest.TestCase):
def test_mock_stubs(self):
with patch(''app.mocking.get_user_name'', return_value = ''Mocked This Silly''):
ret = test_method()
self.assertEqual(ret, ''Mocked This Silly'')