
在Python单元测试中,我们经常需要模拟外部依赖,例如网络请求、数据库操作或文件系统交互。unittest.mock库是Python标准库中用于实现这一目标强大工具。然而,当尝试模拟像json.dumps()这样的标准库函数时,开发者有时会遇到意料之外的问题,特别是TypeError: Object of type MagicMock is not JSON serializable。本文将详细解释这一问题的原因,并提供一个稳健的解决方案。
当一个模块(例如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免费学习笔记(深入)”;
核心问题在于,mock.patch需要知道“在哪里查找”被模拟的对象。对于一个模块内部使用的导入对象,正确的模拟目标应该是"<被测试模块名>.<被导入模块名>"。
Easily find JSON paths within JSON objects using our intuitive Json Path Finder
30
要正确模拟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_以上就是Python单元测试:正确模拟json模块以避免TypeError的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号