Как 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') 
...
Другие вопросы по тегам