Python单元测试中模拟json.dumps()的正确姿势

from unittest import mock
import json

# 这是一个常见的尝试，但可能导致问题
@mock.patch("json.dumps")
def test_serialize_data_incorrect_mock(mock_dumps):
    # 假设这里是被测代码调用json.dumps()
    # mock_dumps.return_value = "mocked_json_string"
    # ...
    pass

# my_app.py
import json

def serialize_user_data(user_info: dict) -> str:
    """
    将用户信息字典序列化为格式化的JSON字符串。
    """
    if not isinstance(user_info, dict):
        raise TypeError("User info must be a dictionary.")
    return json.dumps(user_info, indent=4, sort_keys=True)

def process_and_serialize_config(config_data: dict) -> str:
    """
    处理配置数据并将其序列化。
    """
    # 假设这里有一些处理逻辑
    processed_data = {k.upper(): v for k, v in config_data.items()}
    return serialize_user_data(processed_data) # 内部调用了json.dumps

# test_my_app.py
from unittest import mock
from my_app import serialize_user_data, process_and_serialize_config

class TestJsonDumpsMocking(mock.TestCase):

    @mock.patch("my_app.json") # 模拟my_app模块中导入的json对象
    def test_serialize_user_data_mocked(self, mock_json):
        """
        测试serialize_user_data函数，模拟json.dumps()的行为。
        """
        # 配置mock_json对象的dumps方法，使其返回预设值
        mock_json.dumps.return_value = "mocked_user_json_string"

        test_data = {"name": "Alice", "age": 30}
        result = serialize_user_data(test_data)

        # 验证mock_json.dumps()是否被正确调用，以及调用参数
        mock_json.dumps.assert_called_once_with(test_data, indent=4, sort_keys=True)
        self.assertEqual(result, "mocked_user_json_string")

    @mock.patch("my_app.json") # 再次模拟my_app模块中导入的json对象
    def test_process_and_serialize_config_mocked(self, mock_json):
        """
        测试process_and_serialize_config函数，确保内部json.dumps()被模拟。
        """
        mock_json.dumps.return_value = "mocked_config_json_output"

        initial_config = {"app_name": "My App", "version": "1.0"}
        expected_processed_data = {"APP_NAME": "My App", "VERSION": "1.0"}

        result = process_and_serialize_config(initial_config)

        # 验证内部调用json.dumps()的参数
        mock_json.dumps.assert_called_once_with(expected_processed_data, indent=4, sort_keys=True)
        self.assertEqual(result, "mocked_config_json_output")

    def test_serialize_user_data_type_error(self):
        """
        测试serialize_user_data函数在输入类型错误时的行为（不涉及mock）。
        """
        with self.assertRaises(TypeError):
            serialize_user_data("not_a_dict")