如何模拟函数中的函数.是否可以使用两个或多个补丁来模拟函数中的函数调用

Question

我需要模拟一个函数(funk(调用，该调用依赖于在该函数(fun克(中调用的另外两个函数的返回值。是否可以使用多个Patch来模拟两个被调用函数的返回值。

待测试/模拟的API：

from subprocess import check_output

def get_value_a(value_a: int):  
return value_a  
def get_value_b(value_b: int):  
return value_b  
def print_contents_of_cwd(val_a, val_b):  
if val_a > 0:  
return check_output(['ls']).split()  
else:  
return val_b  

我试图使用两个补丁来模拟两个函数(get_value_a和get_value_b(的返回值，如下所示：

from Mack_Test_Practise import list_dir
from unittest import mock, TestCase

class TestExamples(TestCase):
# Using Decorator Method
@mock.patch('Mack_Test_Practise.list_dir.check_output',
return_value=b'list_dir.py')
@mock.patch('Mack_Test_Practise.list_dir.get_value_a',
return_value=5)
@mock.patch('Mack_Test_Practise.list_dir.get_value_b',
return_value=3)
def test_print_contents_of_file(self, mock_check_output,val_a, val_b):
actual_result = list_dir.print_contents_of_cwd(val_a, val_b)
print("actual_result", actual_result)
expected_result = b'list_dir.py'
self.assertIn(expected_result, actual_result)  

当我运行测试时，会抛出以下错误。

Error  
Traceback (most recent call last):  
File "/usr/lib/python3.6/unittest/mock.py", line 1183, in patched  
return func(*args, **keywargs)
File  
"/home/shyam/PycharmProjects/pythonProject/Mock_Test/test_list_dir.py", line 15, in test_print_contents_of_file
actual_result = list_dir.print_contents_of_cwd(val_a, val_b)
File "/home/shyam/PycharmProjects/pythonProject/Mack_Test_Practise/list_dir.py", line 13, in print_contents_of_cwd  
if val_a > 0:  
**TypeError: '>' not supported between instances of 'MagicMock' and 'int'**  

我的第一个问题是，是否允许像我一样使用多个补丁
我的第二个问题是，既然Type Error是Python中的合法运算符，为什么我们会得到它
请提出解决方案。

万分感谢并致以最良好的问候！！

Answer 1

使用这样的多个补丁是合法的。

我认为你的问题是因为你已经修补了一个函数，并试图将修补后的函数与整数进行比较。您在mock中设置了返回值，但仍然需要调用它来获得该返回值。

actual_result = list_dir.print_contents_of_cwd(val_a(), val_b())

注意val_a()和val_b()上包含的括号

现在，由于这些不再是值，而是函数(从技术上讲，它们是MagicMock对象，但它们正在替换函数(，我建议将它们重命名为get_val_a()和get_val_b()，这样您的意图就更清楚了。