使用Python模擬框架嘲笑時TypeError

Question

我越來越TypeError需要正好2個參數（0給出），而試圖嘲笑類功能使用python模擬框架。

>>> class ExampleClass(): 
...  @staticmethod 
...  def _process_updates(arg1, arg2): 
...   pass 
... 
>>> 
>>> @patch("ExampleClass._process_updates") 
... def process_updates(arg1, arg2): 
... return "test" 
... 
>>> ExampleClass._process_updates() 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
TypeError: _process_updates() takes exactly 2 arguments (0 given) 
>>> 

Answer 1

正如@DanielRoseman說，@patch替代的功能（方法）由函數將其修補使用。即在下面的例子中，函數calling_func_with_mock是calling_func的修補版本，因爲我們用process_updates代替_process_updates（在本地ExampleClass對象內）。

我認爲他們對你的關鍵是使用ExampleClass修補更高級別的功能，即。 HTH

from unittest.mock import patch 

class ExampleClass(): 
    @staticmethod 
    def _process_updates(arg1, arg2): 
     return arg1 

def patched_process_updates(arg1, arg2): 
    return arg2 

def calling_func(): 
    """Uses normal ExampleClass._process_updates""" 
    print(ExampleClass._process_updates('Hello','world')) 

@patch('__main__.ExampleClass._process_updates', new=patched_process_updates) 
def calling_func_with_mock(): 
    """Uses ExampleClass._process_updates patch""" 
    print(ExampleClass._process_updates('Hello','world')) 

if __name__=='__main__': 
    calling_func() 
    # 'Hello' 
    calling_func_with_mock() 
    # 'world'