单元测试Flask蓝图时如何用Mock.patch模拟auth()函数？

要在单元测试中模拟蓝图里的auth()函数，核心是通过mock工具替换模块级别的函数引用，让index_view()调用自定义的mock函数而非原函数。以下是两种常用实现方式：

方式一：使用unittest.mock（原生unittest框架） #

创建测试文件test_index.py，代码如下：

import unittest
from unittest.mock import patch
from main import app

class TestIndexBlueprint(unittest.TestCase):
    def setUp(self):
        # 初始化Flask测试客户端
        self.client = app.test_client()

    @patch('views.index.auth')
    def test_index_view_mocked_auth(self, mock_auth):
        # 设置mock函数的返回值为"cat"
        mock_auth.return_value = "cat"
        # 发起对首页的请求
        response = self.client.get('/')
        # 断言响应内容中包含mock返回的"cat"（需匹配模板渲染逻辑）
        self.assertIn(b'cat', response.data)

if __name__ == '__main__':
    unittest.main()

方式二：使用pytest的monkeypatch（pytest框架） #

如果用pytest作为测试框架，可通过monkeypatch工具替换目标函数：

from main import app

def test_index_view_with_mocked_auth(monkeypatch):
    # 定义自定义的mock auth函数
    def mock_auth():
        return "cat"
    # 替换views.index模块中的auth函数
    monkeypatch.setattr('views.index.auth', mock_auth)
    # 创建测试客户端并发起请求
    client = app.test_client()
    response = client.get('/')
    # 断言响应包含预期值
    assert b'cat' in response.data

关键注意点 #

• 替换路径必须准确：因为index_view()是在views.index模块内调用auth()，所以要替换的是views.index.auth这个模块级别的函数引用，不能写错路径。
• 断言逻辑需匹配模板：如果模板中直接渲染user变量（比如<p>{{ user }}</p>），则响应数据中会包含对应的字符串，直接断言即可。

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