Python单元测试中如何Mock嵌套的数据库调用？

我来帮你解决这个嵌套数据库调用的Mock问题，其实用Python的unittest.mock模块就能搞定，核心是找对Patch的路径，然后一步步模拟嵌套的数据库结构就行，具体步骤和示例如下：

核心思路 #

不管数据库调用的嵌套有多深，关键要记住：你需要Patch的是被测试代码（也就是some_func.py）中实际引用数据库类/方法的路径，而不是db_module.py里的原始定义路径。这是因为Python的导入机制是基于模块的命名空间的，只有Patch被测试模块里的那个引用，才能让Mock生效。

具体步骤与示例 #

1. 确定要Patch的目标路径 #

假设你的some_func.py里是这样导入数据库类的：

from db_module import MySQLConn

def some_func():
    conn = MySQLConn()
    # 后续用conn._conn执行数据库操作...

那你需要Patch的路径就是some_func.MySQLConn，而不是db_module.MySQLConn。

2. 模拟嵌套的数据库结构 #

接下来要逐步Mock嵌套的属性和方法，比如MySQLConn实例的_conn属性、cursor()方法、execute()方法等。我们可以用MagicMock来链式设置这些嵌套对象的返回值：

3. 完整的单元测试示例 #

这里用unittest框架结合patch装饰器来写测试：

from unittest.mock import patch, MagicMock
import unittest
from some_func import some_func

class TestSomeFunc(unittest.TestCase):
    @patch('some_func.MySQLConn')  # 重点：Patch被测试模块中的MySQLConn引用
    def test_some_func_mocked_db(self, mock_mysql_cls):
        # 1. 模拟MySQLConn实例的_conn属性
        mock_db_conn = mock_mysql_cls.return_value._conn
        
        # 2. 模拟游标对象及其方法
        mock_cursor = mock_db_conn.cursor.return_value
        # 设置游标fetchall的返回结果（根据你的业务逻辑调整）
        mock_cursor.fetchall.return_value = [('user1', 'active'), ('user2', 'inactive')]
        # 如果需要模拟execute的异常，也可以这样设置：
        # mock_cursor.execute.side_effect = Exception("DB Error")

        # 3. 调用被测试函数
        result = some_func()

        # 4. 断言验证：确保数据库交互符合预期
        # 验证游标是否被创建
        mock_db_conn.cursor.assert_called_once()
        # 验证execute是否执行了正确的SQL（替换成你实际的SQL语句）
        mock_cursor.execute.assert_called_with("SELECT username, status FROM users")
        # 验证函数返回值是否符合预期
        self.assertEqual(result, ['user1', 'user2'])

4. 处理更复杂的嵌套场景 #

如果你的代码还用到了db_module里的config或get_creds，同样可以用相同的思路Patch：

• 比如some_func间接调用了get_creds，就Patchsome_func.get_creds，返回模拟的凭证：

  @patch('some_func.get_creds')
  @patch('some_func.MySQLConn')
  def test_some_func_with_creds(self, mock_mysql_cls, mock_get_creds):
      mock_get_creds.return_value = ('test_user', 'test_pass')
      # 后续逻辑和之前一样...
  

常见坑点提醒 #

• 路径错误：很多人会错误地Patchdb_module.MySQLConn，但这样对some_func.py里的引用不会生效，必须Patch被测试模块里的那个导入路径。
• 嵌套属性顺序：Mock的时候要从外层到内层逐步设置，比如先Mock类的返回值，再Mock实例的_conn，再Mock_conn的游标，不能跳步。
• 测试隔离：如果MySQLConn是单例模式（看你提供的代码里有_conn = None的单例迹象），要确保每个测试用例的Mock都是独立的，避免测试之间的状态污染。

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