0

0

Python单元测试:正确模拟json模块以避免TypeError

霞舞

霞舞

发布时间:2025-09-13 11:26:01

|

481人浏览过

|

来源于php中文网

原创

python单元测试:正确模拟json模块以避免typeerror

本文深入探讨了在Python单元测试中模拟json.dumps()时可能遇到的TypeError: Object of type MagicMock is not JSON serializable问题。核心解决方案在于理解Python的导入机制和unittest.mock的工作原理,即应模拟被测试模块中导入并使用的json模块引用,而非全局的json.dumps()函数,从而有效控制json序列化行为,确保测试的隔离性和正确性。

在Python单元测试中,我们经常需要模拟外部依赖,例如网络请求、数据库操作或文件系统交互。unittest.mock库是Python标准库中用于实现这一目标强大工具。然而,当尝试模拟像json.dumps()这样的标准库函数时,开发者有时会遇到意料之外的问题,特别是TypeError: Object of type MagicMock is not JSON serializable。本文将详细解释这一问题的原因,并提供一个稳健的解决方案。

问题根源:Python的导入机制与mock.patch

当一个模块(例如my_module.py)导入另一个模块(例如json)时,它会在自己的命名空间中创建一个对该模块的引用。例如:

# my_module.py
import json

def serialize_data(data):
    return json.dumps(data)

在这个例子中,my_module内部使用的json对象是my_module命名空间中的一个引用。如果我们尝试使用@mock.patch("json.dumps")来模拟json.dumps(),我们实际上是在修改全局json模块中的dumps函数。这通常不会影响到my_module中已经导入并使用的json模块引用,因为my_module在导入时已经将json模块加载到了自己的命名空间,并且后续对json.dumps的调用会通过my_module自己的json引用进行。因此,my_module仍然会调用原始的json.dumps(),导致模拟失败。

至于TypeError: Object of type MagicMock is not JSON serializable,这通常是由于以下情况之一造成的:

立即学习Python免费学习笔记(深入)”;

  1. 错误的模拟目标: mock.patch("json.dumps")未能成功替换my_module中使用的json.dumps。因此,my_module中的代码仍然调用原始的json.dumps。如果测试代码随后将一个MagicMock对象作为数据传给了这个未被模拟的json.dumps(例如,某个依赖于json.dumps的函数返回了一个MagicMock对象,然后这个MagicMock对象又被传给了另一个json.dumps),就会触发此TypeError。
  2. 模拟配置不当: 即使json.dumps被成功模拟,如果MagicMock对象没有被正确配置以返回一个字符串(json.dumps的预期输出),并且其返回值被后续代码尝试再次序列化,也可能导致此错误。

核心问题在于,mock.patch需要知道“在哪里查找”被模拟的对象。对于一个模块内部使用的导入对象,正确的模拟目标应该是"."。

倍塔塞司
倍塔塞司

AI职业规划、AI职业测评、定制测评、AI工具等多样化职业类AI服务。

下载

正确的模拟策略:模拟模块引用

要正确模拟my_module中使用的json模块,我们应该模拟my_module命名空间中的json引用,而不是全局的json模块或其dumps函数。这意味着正确的模拟路径应该是"my_module.json"。通过这种方式,当my_module尝试访问json模块时,它会得到一个MagicMock对象,我们可以配置这个MagicMock对象来控制json.dumps等方法的行为。

示例代码

为了更好地理解,我们通过一个具体的例子来演示。

my_module.py (被测试的模块):

# my_module.py
import json

def serialize_user_profile(user_data: dict) -> str:
    """
    序列化用户数据为JSON字符串,并添加一些处理信息。
    """
    processed_data = {
        "user_id": user_data.get("id"),
        "username": user_data.get("name"),
        "status": "processed",
        "original_input_keys": list(user_data.keys())
    }
    # my_module 内部调用 json.dumps
    return json.dumps(processed_data, indent=2, ensure_ascii=False)

def get_user_data_as_json(user_id: str) -> str:
    """
    模拟从数据库获取用户数据并序列化。
    (这里简化为直接构造数据)
    """
    user_info = {"id": user_id, "name": f"User_{user_id}", "email": f"user{user_id}@example.com"}
    return serialize_user_profile(user_info)

test_my_module.py (单元测试):

# test_my_module.py
import unittest
from unittest import mock
import json # 引入 json 模块仅用于演示,不用于模拟

# 导入被测试的模块
import my_module

class TestMyModuleSerialization(unittest.TestCase):

    def test_serialize_user_profile_without_mock(self):
        """
        测试不使用mock时 serialize_user_profile 的正常行为。
        """
        user_data = {"id": "123", "name": "张三", "age": 30}
        expected_output = '{\n  "user_id": "123",\n  "username": "张三",\n  "status": "processed",\n  "original_input_keys": [\n    "id",\n    "name",\n    "age"\n  ]\n}'
        self.assertEqual(my_module.serialize_user_profile(user_data), expected_output)

    @mock.patch("json.dumps")
    def test_get_user_data_as_json_incorrect_patch(self, mock_global_dumps):
        """
        尝试模拟全局的 json.dumps,但对 my_module 无效。
        """
        mock_global_dumps.return_value = "MOCKED GLOBAL DUMPS OUTPUT"
        test_user_id = "456"

        # 此时 my_module.json.dumps 仍然是原始的 json.dumps
        # 因为 my_module 拥有自己的 json 模块引用
        result = my_module.get_user_data_as_json(test_user_id)

        # 验证原始函数被调用,而不是mock
        mock_global_dumps.assert_not_called()
        self.assertNotEqual(result, "MOCKED GLOBAL DUMPS OUTPUT")

        # 实际结果会是原始 json.dumps 的输出,包含处理后的数据
        self.assertIn("processed", result) 
        self.assertIn(f"User_{test_user_id}", result)

        # 注意:在此场景下,通常不会直接出现 TypeError: Object of type MagicMock is not JSON serializable,
        # 因为原始的 json.dumps 正常运行。但如果后续代码期望 mock 的返回值,
        # 而此处得到的是原始 json.dumps 的结果,可能导致其他逻辑错误,
        # 甚至在更复杂的交互中间接触发 TypeError。
        # 关键在于,这种方式未能成功模拟 my_module 内部的 json 行为。

    @mock.patch("my_module.json")
    def test_get_

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

420

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

536

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

313

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

77

2025.09.10

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

340

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

212

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1503

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

625

2023.11.24

2026赚钱平台入口大全
2026赚钱平台入口大全

2026年最新赚钱平台入口汇总,涵盖任务众包、内容创作、电商运营、技能变现等多类正规渠道,助你轻松开启副业增收之路。阅读专题下面的文章了解更多详细内容。

54

2026.01.31

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.7万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号