Python Mocking function 在嵌套函數中調用

Question

抱歉，這個問題可能很容易解決，我是 python mocking 的新手，很難弄清楚這一點。

 @mock.patch('my_module.PypiImportedPackage.SomeClassAdvancedFunctions.get_attrib')
    def test_mocking(self, get_attrib_mock):
       get_attrib.return_value = '1'
       my_instance = SomeClass(param1)
       mytest_result = my_instance.advanced.get_attrib('doesntmattertheinputthisismocked')
       self.assertEqual(mytest_result, 1)
       # pass 
       calltoafunctionthatoimportesSomeClassfrom__my_module__andfailsif__get_attrib__doesnotreturn1(param2,param3)
      #fail

在 my_module 我有

import PypiImportedPackage 
from PypiImportedPackage import SomeClass

在SomeClass.__init__ （我從 pypi 導入，所以我沒有太多機會改變），有一行self.advanced = SomeClassAdvancedFunctions所以我不能真正修補 SomeClass.advanced.get_attrib 因為它只存在於實例等級，

這就是為什么我必須修補 PypiImportedPackage.SomeClassAdvancedFunctions 並且實際上這會傳入我的代碼的AssetEqual操作。

知道我做錯了什么嗎？

謝謝

Answer 1

您的get_attrib.return_value = '1'應該是get_attrib_mock.return_value = '1'因為 get_attrib_mock 現在可以在修補后作為 get_attrib 工作。