Как mock.patch зависимость в Python?
Дразнить метод, который является зависимостью, не работает для меня. Когда метод, который мне нужно проверить, вызывает метод зависимости, реальный метод вызывается вместо фиктивной версии.
У меня есть следующие файлы:
myLibrary.py
from src.myOtherLibrary import myOtherLibrary
class myLibrary():
def __init__():
self.myVar = myOtherLibrary() #dependancy
def my_method():
method1 = self.method1()
externalMethod2 self.myVar.method2() #method2 called from the external class myOtherLibrary
return method1 + externalMethod2
def method1():
return "method1 from myLibrary..."
SRC /myOtherLibrary.py
class myOtherLibrary():
def method2():
return "method2 from myOtherLibrary..."
Наконец модульный тест:
TestMyLibrary.py
import unittest
import mock
from myLibrary import myLibrary
from src.myOtherLibrary import myOtherLibrary
class TestMyLibrary(unittest.TestCase):
@mock.patch('myLibrary.myLibrary.method1') #mocking this works because it's a sibling method from my_method() to test
@mock.patch('src.myOtherLibrary.myOtherLibrary.method2') #this does not work because it's an external class from myLibrary
def test_my_method(my_method1_to_mock, my_method2_to_mock):
my_method1_to_mock.return_value = "something_to_return.."
my_method2_to_mock.return_value = "something_else_to_return.."
myLibraryVar = myLibrary()
result = myLibraryVar.my_method()
print result #I would expect to see this: "something_to_return..something_else_to_return.."
#But it actually prints this: "something_to_return..method2 from myOtherLibrary..."
#So mocking is not working for method2
self.assertEqual('something_to_return..something_else_to_return..', result)
if __name__ == '__main__':
unittest.main()
Возможно, важно отметить, что myLibrary.py и TestMyLibrary.py находятся в одной и той же папке, но myOtherLibrary.py находится на другом уровне папок.
Я надеюсь, что вы можете помочь мне найти то, что мне не хватает здесь.
Любое предложение будет очень хорошо оценено. Благодарю.
1 ответ
Ты не можешь patch
так, потому что методы, которые вы пытаетесь смоделировать, являются методами класса, а не функциями. Так что вам нужно использовать patch.object
,
@mock.patch.object(myLibrary, 'method1')
@mock.patch.object(myOtherLibrary, 'method2')
...