पायथन एक आयातित मॉड्यूल से एक फ़ंक्शन मॉकिंग

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

यह काम करता है जैसा कि मैं अपेक्षा करता हूं। "पैच" मॉड्यूल बस get_user_name का अनमॉक मान देता है। मैं अन्य संकुलों से विधियों का नकल कैसे करूं जिन्हें मैं परीक्षण के तहत नामस्थान में आयात कर रहा हूं?

Answer 1

आप unittest.mock पैकेज से patch डेकोरेटर का उपयोग कर रहे जब आप नहीं नाम स्थान मॉड्यूल आप परीक्षण app.mocking.get_user_name के तहत नाम स्थान में यह पैचिंग कर रहे हैं (इस मामले app.my_module.get_user_name में) से आयात किया जाता पैचिंग कर रहे हैं।

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 इस का वर्णन शामिल हैं: के साथ Mock नीचे की तरह कुछ कोशिश

ऊपर ऐसा करने के लिए।

Answer 2

जबकि मैटी जॉन का जवाब आपकी समस्या हल करता है (और मुझे भी मदद मिली, धन्यवाद!), हालांकि, मैं मजाक के साथ मूल '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')