Python Mocking функція із імпортного модуля

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

Ось де я поки що.

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"

тест / 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" на знущену функцію. Це дозволить вам контролювати, коли функція замінена та коли її немає. Також це дозволить вам зробити кілька замін в одному тесті. Для цього використовуйте статтю "з" досить подібним чином:

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