Python имитирует функцию из импортированного модуля

Я хочу понять, как выполнить @patchфункцию из импортированного модуля.

Вот где я пока нахожусь.

Приложение / 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()

Приложение / my_module / __ init__.py:

def get_user_name():
  return "Unmocked User"

тест / 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()

Это не работает так, как я ожидал. «Патченный» модуль просто возвращает разблокированное значение get_user_name. Как имитировать методы из других пакетов, которые я импортирую в тестируемое пространство имен?

Когда вы используете patchдекоратор из unittest.mockпакета, вы не исправляете пространство имен, из которого импортируется модуль (в данном случае app.my_module.get_user_name), вы исправляете его в тестируемом пространстве имен app.mocking.get_user_name.

Чтобы сделать это, Mockпопробуйте что-то вроде следующего:

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

В документации стандартной библиотеки есть полезный раздел, описывающий это.

Хотя ответ Матти Джона решает вашу проблему (и помог мне тоже, спасибо!), Я бы, однако, предложил локализовать замену исходной функции get_user_name на поддельную. Это позволит вам контролировать, когда функция заменяется, а когда нет. Кроме того, это позволит вам выполнить несколько замен в одном тесте. Для этого используйте условие 'with' довольно похожим образом:

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