Python Mockowanie funkcji z zaimportowanego modułu

Chcę zrozumieć, jak wykonać @patchfunkcję z zaimportowanego modułu.

To jest, gdzie jestem do tej pory.

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()

To nie działa tak, jak bym się spodziewał. „Połatany” moduł po prostu zwraca niezakodowaną wartość get_user_name. Jak pozorować metody z innych pakietów, które importuję do testowanej przestrzeni nazw?

Kiedy używasz patchdekoratora z unittest.mockpakietu, nie łatasz przestrzeni nazw, z której moduł jest importowany (w tym przypadku app.my_module.get_user_name), łatasz go w testowanej przestrzeni nazw app.mocking.get_user_name.

Aby zrobić powyższe, Mockspróbuj czegoś takiego jak poniżej:

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')

Dokumentacja biblioteki standardowej zawiera przydatną sekcję opisującą to.

Chociaż odpowiedź Matti Johna rozwiązuje Twój problem (i mi też pomogła, dzięki!), Sugerowałbym jednak zlokalizowanie zamiany oryginalnej funkcji „get_user_name” na fałszywą. Umożliwi to kontrolowanie, kiedy funkcja zostanie zastąpiona, a kiedy nie. Pozwoli to również na dokonanie kilku wymian w tym samym teście. Aby to zrobić, użyj wyrażenia „with” w całkiem podobny sposób:

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')