python使用不同的参数模拟函数，并使用返回值进行断言

Question

这可能是一个简单的问题，但我对模拟测试是个新手，不知道如何让它工作。

我有一个这样的代码

def check_item(item):
    <do some processing>
    return processed_item

基本上，我试图用多个项和各自的返回值来模拟这个函数。例如，如果item是apple，返回值是processed_apple，如果是橙子，返回值是processed_orange，依此类推。

到目前为止，我正在尝试编写一个测试用例，并坚持使用以下内容

import my_module
from unittest.mock import patch

@patch("my_module.check_item")
def test_check_item(self, check_item):
    check_item.assert_called_once_with("apple")
    check_item.return_value = 'processed_apple'

不确定我是否做对了(或者是正确的方式)。如何检查每个参数具有不同返回值的多个参数？

Answer 1

测试方法是否被调用：

check_item.assert_called_once_with("apple")

您应该首先调用函数：

@patch("my_module.check_item")
def test_check_item(self, check_item):
   check_item("apple")
   check_item.assert_called_once_with("apple")
   check_item.return_value = 'processed_apple'

但在代码中，我看不到有任何理由这样测试它，因为它只是在测试模拟模块。

如果你想测试函数返回值，你不应该模拟函数，而应该让它保持原样，只针对不同的场景进行测试。例如：

def test_check_item(self):
   result = check_item("apple")
   self.assertEqual(result, "some predefined result")

单元测试的目的是测试方法/类的正确行为