Python Mock patch une fonction dans une autre fonction
def f1():
return 10, True
def f2():
num, stat = f1()
return 2*num, stat
comment utiliser la bibliothèque simulée de python pour patch f1() et retourner un résultat personnalisé afin que je puisse tester f2()?
Édition: Quelque chose ne va pas avec mon test? Cela ne semble pas fonctionner, tous les tests ont échoué avec AssertionError
from foo.bar import f2
from mock import patch
class MyTest(TestCase):
def test_f2_1(self):
with patch('project.module.f1') as some_func:
some_func.return_value = (20, False)
num, stat = f2()
self.assertEqual((num, stat), (40, False))
@patch('project.module.f1')
def test_f2_2(self, some_func):
some_func.return_value = (20, False)
num, stat = f2()
self.assertEqual((num, stat), (40, False))
2 réponses
le premier exemple suggère que f1() et f2 () sont définis dans le même module. Par conséquent, ce qui suit devrait fonctionner:
from foo.bar import f2
from mock import patch
class MyTest(TestCase):
@patch('foo.bar.f1')
def test_f2_2(self, some_func):
some_func.return_value = (20, False)
num, stat = f2()
self.assertEqual((num, stat), (40, False))
Patch est la même que l'importation: @patch('foo.bar.f1')
Ici est une bonne réponse sur la question:
http://bhfsteve.blogspot.nl/2012/06/patching-tip-using-mocks-in-python-unit.html
en supposant que vous utilisez ceci faux bibliothèque:
def f1():
return 10, True
def f2():
num, stat = f1()
return 2*num, stat
import mock
print f2() # Unchanged f1 -> prints (20, True)
with mock.patch('__main__.f1') as MockClass: # replace f1 with MockClass
MockClass.return_value = (30, True) # Change the return value
print f2() # f2 with changed f1 -> prints (60, True)
si votre code est divisé en modules, vous devrez probablement remplacer __main__.f1 avec le chemin vers votre module / fonction.