Python Chế nhạo một hàm từ một mô-đun đã nhập

Tôi muốn hiểu cách sử dụng @patchmột hàm từ một mô-đun đã nhập.

Đây là nơi tôi đang ở cho đến nay.

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

Điều này không hoạt động như tôi mong đợi. Mô-đun "được vá" chỉ trả về giá trị chưa được sửa của get_user_name. Làm cách nào để mô phỏng các phương thức từ các gói khác mà tôi đang nhập vào một không gian tên đang được kiểm tra?

Khi bạn đang sử dụng trình patchtrang trí từ unittest.mockgói, bạn không vá không gian tên mà mô-đun được nhập từ (trong trường hợp này app.my_module.get_user_name), bạn đang vá nó trong không gian tên đang được kiểm tra app.mocking.get_user_name.

Để thực hiện điều trên, Mockhãy thử một số thứ như sau:

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

Tài liệu thư viện tiêu chuẩn bao gồm một phần hữu ích mô tả điều này.

Mặc dù câu trả lời của Matti John giải quyết được vấn đề của bạn (và cũng giúp tôi, cảm ơn!), Tuy nhiên, tôi sẽ đề xuất bản địa hóa việc thay thế hàm 'get_user_name' ban đầu bằng hàm bị chế nhạo. Điều này sẽ cho phép bạn kiểm soát khi nào chức năng được thay thế và khi nào thì không. Ngoài ra, điều này sẽ cho phép bạn thực hiện nhiều thay thế trong cùng một thử nghiệm. Để làm như vậy, hãy sử dụng trạng thái 'with' theo cách khá đơn giản:

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