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

สิ่งนี้ไม่ได้ผลตามที่ฉันคาดหวัง ว่า "ปะ" โมดูลเพียงแค่ส่งกลับค่า unmocked get_user_nameของ ฉันจะจำลองวิธีการจากแพ็คเกจอื่น ๆ ที่ฉันนำเข้ามาในเนมสเปซที่อยู่ระหว่างการทดสอบได้อย่างไร

เมื่อคุณกำลังใช้patchมัณฑนากรจากunittest.mockแพคเกจที่คุณไม่ได้ปะ namespace โมดูลนำเข้าจาก (ในกรณีนี้app.my_module.get_user_name) คุณกำลัง patching ใน namespace 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')

เอกสารคู่มือไลบรารีมาตรฐานประกอบด้วยส่วนที่เป็นประโยชน์ซึ่งอธิบายถึงสิ่งนี้

แม้ว่าคำตอบของ Matti John จะช่วยแก้ปัญหาของคุณได้ (และช่วยฉันด้วยขอบคุณ!) แต่ฉันขอแนะนำให้แปลการแทนที่ฟังก์ชัน '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')