Pythonインポートされたモジュールの関数を模倣する

Question

インポートされたモジュールの関数@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() 

これはではなく、の仕事を期待しています。 「patched」モジ​​ュールは、単に、ロックされていない値get_user_nameを返します。テスト中の名前空間にインポートする他のパッケージからメソッドをモックするにはどうすればよいですか？

Answer 1

あなたはモジュールが（この場合はapp.my_module.get_user_nameに）あなたは、テストapp.mocking.get_user_name下の名前空間でそれをパッチを適用しているから輸入された名前空間をパッチないあるunittest.mockパッケージからpatchデコレータを使用しています。 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') 

標準ライブラリのドキュメントがこれを記述した便利なsectionが含まれています。

Answer 2

Matti Johnの答えがあなたの問題を解決してくれました（そして私も助けてくれました）が、元の 'get_user_name'関数の置き換えを模擬したものにローカライズすることをお勧めします。これにより、機能がいつ置き換えられ、いつ機能しないかを制御することができます。また、同じテストでいくつかの置き換えを行うことができます。そうするためには、かなりsimillarようにしなステートメント「で」を使用：

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