如何在Python的unittest中向side_effect传递额外参数？

下面针对你的场景，提供三种实用的实现方式：

方法1：使用闭包（最直观的方式） #

在测试函数内部定义一个闭包函数，让它直接访问测试函数里的额外参数，再调用你的get_dynamic_users_mock并传入所需参数：

import unittest
from unittest.mock import patch
from api import get_users_from_api

def get_users(user_ids):
    for user_id in user_ids:
        res = get_users_from_api(user_id)

def get_dynamic_users_mock(user_id, extra_param1, extra_param2):
    # 这里可以使用传入的额外参数动态生成mock数据
    mock_by_user_id = {
        1: {"id": 1, "first_name": "John", "last_name": "Doe"},
        2: {"id": 2, "first_name": "Jane", "last_name": "Smith"},
    }
    # 示例：用额外参数修改返回值
    if extra_param1 == "add_age":
        mock_by_user_id[user_id]["age"] = 30
    return mock_by_user_id[user_id]

class TestGetUsers(unittest.TestCase):
    @patch("api.get_users_from_api")
    def test_get_users(self, mock_get_users):
        user_ids = [1, 2]
        # 定义测试用的额外参数
        extra_arg1 = "add_age"
        extra_arg2 = {"custom_tag": "test_user"}
        
        # 闭包捕获测试函数内的额外参数
        def mock_wrapper(user_id):
            return get_dynamic_users_mock(user_id, extra_arg1, extra_arg2)
        
        mock_get_users.side_effect = mock_wrapper
        get_users(user_ids)
        # 后续可添加断言，比如验证mock调用次数、参数等

方法2：使用functools.partial绑定参数 #

用functools.partial把额外参数预先绑定到mock函数上，让side_effect调用时自动将原函数参数（这里是user_id）和绑定参数一起传入：

import unittest
from unittest.mock import patch
from functools import partial
from api import get_users_from_api

# 原get_users和get_dynamic_users_mock定义同上

class TestGetUsers(unittest.TestCase):
    @patch("api.get_users_from_api")
    def test_get_users(self, mock_get_users):
        user_ids = [1, 2]
        extra_arg1 = "add_age"
        extra_arg2 = {"custom_tag": "test_user"}
        
        # 绑定额外参数到mock函数
        bound_mock = partial(get_dynamic_users_mock, extra_param1=extra_arg1, extra_param2=extra_arg2)
        mock_get_users.side_effect = bound_mock
        
        get_users(user_ids)
        # 断言逻辑

注意：如果mock函数用位置参数接收额外值，partial会按顺序绑定；用关键字参数则直接指定，避免参数顺序混乱。

方法3：使用lambda表达式（适合简单场景） #

如果额外参数不多且逻辑简单，直接用lambda表达式包装mock函数传递参数：

import unittest
from unittest.mock import patch
from api import get_users_from_api

# 原get_users和get_dynamic_users_mock定义同上

class TestGetUsers(unittest.TestCase):
    @patch("api.get_users_from_api")
    def test_get_users(self, mock_get_users):
        user_ids = [1, 2]
        extra_arg = "add_age"
        
        mock_get_users.side_effect = lambda user_id: get_dynamic_users_mock(user_id, extra_arg)
        
        get_users(user_ids)
        # 断言逻辑

关键说明 #

side_effect要求传入的可调用对象参数列表和被Mock的原函数（get_users_from_api）一致。我们通过闭包、partial或lambda包装原mock函数，既适配原函数的参数结构，又能传递额外参数。其中闭包灵活性最高，partial则更简洁。

内容的提问来源于stack exchange，提问作者user21656699