def f1():
return 10, True
def f2():
num, stat = f1()
return 2*num, stat
Comment utiliser la bibliothèque fictive de python pour patcher f1()
et retourner un résultat personnalisé afin que je puisse tester f2()
?
Modifié: Y a-t-il un problème 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))
Le premier exemple suggère que f1 () et f2 () sont définis dans le même module. Par conséquent, les éléments suivants devraient 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))
Le patch est identique à l'importation: @patch('foo.bar.f1')
Voici 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 ce mock libary:
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 d'accès à votre module/fonction.