Python Mocking fungsi dari modul yang diimpor

Saya ingin memahami bagaimana @patchsuatu fungsi dari modul yang diimpor.

Di sinilah saya sejauh ini.

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

Ini tidak bekerja seperti yang saya harapkan. Modul "patched" hanya mengembalikan nilai unmocked dari get_user_name. Bagaimana cara meniru metode dari paket lain yang saya impor ke namespace yang sedang diuji?

Saat Anda menggunakan patchdekorator dari unittest.mockpaket, Anda tidak menambal namespace tempat modul diimpor (dalam hal ini app.my_module.get_user_name) Anda menambalnya di namespace yang diuji app.mocking.get_user_name.

Untuk melakukan hal di atas dengan Mockmencoba sesuatu seperti di bawah ini:

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

Dokumentasi pustaka standar menyertakan bagian berguna yang menjelaskan hal ini.

Sementara jawaban Matti John memecahkan masalah Anda (dan membantu saya juga, terima kasih!), Namun saya akan menyarankan untuk melokalkan penggantian fungsi 'get_user_name' yang asli dengan fungsi tiruan. Ini akan memungkinkan Anda untuk mengontrol kapan fungsi diganti dan kapan tidak. Selain itu, ini memungkinkan Anda membuat beberapa penggantian dalam pengujian yang sama. Untuk melakukannya, gunakan pernyataan 'with' dengan cara yang cukup mirip:

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