如何解决Flask-SQLAlchemy中mock<ModelClass>.query.filter单测未覆盖问题

• 被测代码硬编码了分支判断的判断值：你提供的代码片段里，函数内直接写死industry_sele = "All"，整个执行流没有任何可以改变这个变量值的逻辑，导致if industry_sele != "All" 这个判断永远为False，分支本身是不可达的死代码，无论怎么执行测试都不会进入这个判断块。
• 参考方案的适配偏差：常见的Flask-SQLAlchemy查询mock实现多是针对filter_by()方法设计的，而你实际分支内触发的查询方法是filter()，就算解决了硬编码问题，如果你仅按照原方案的逻辑mockfilter_by的调用，也无法正确捕获分支触发的执行痕迹。

第一步：修正被测代码的可控性问题 #

把硬编码的industry_sele改为从路由可接收的动态输入读取（符合Flask路由的常规业务逻辑），让分支判断的条件可以被测试用例控制：

from flask import request

@process_bp.route("/getListCSV")
def get_list_csv():
    # 从请求参数读取行业筛选值，默认值为All
    industry_sele = request.args.get("industry", "All")
    # 业务剩余逻辑
    if industry_sele != "All":
        companies_flask_list = Companies.query.filter(Companies.industry_desc == industry_sele)

第二步：编写分支对应的测试用例 #

测试时不需要连接真实数据库，直接替换Companies.query为Mock对象即可，重点要匹配代码中实际调用的filter方法：

from unittest.mock import patch

def test_industry_filter_branch(test_client):
    # 指向被测代码实际引用的Companies模型的query属性，做mock替换
    with patch("your_app.models.Companies.query") as mock_query:
        # 配置mock的filter方法返回值，模拟查询结果
        mock_query.filter.return_value.all.return_value = []
        
        # 构造传入指定行业值的请求，触发if分支的判断条件
        resp = test_client.get("/getListCSV?industry=Tech")
        
        # 断言校验：确认分支内的filter查询被正确触发
        mock_query.filter.assert_called_once()
        # 校验传入的过滤条件和预期的行业值一致
        filter_condition = mock_query.filter.call_args[0][0]
        assert filter_condition.right.value == "Tech"
        
        # 校验接口响应正常
        assert resp.status_code == 200

常见实现注意点 #

• Mock的路径要准确：要patch被测代码导入路径下的Companies.query对象，而非SQLAlchemy框架内部的源码路径，否则mock不会生效。
• 方法匹配要对齐：代码中用filter()就监听filter方法的调用记录，用filter_by()就监听filter_by，二者的参数传参逻辑完全不同，混用会导致断言失效。
• 不要硬编码分支判断的输入值：所有影响分支判断的变量，都要从测试可注入的入口（请求参数、上下文、依赖注入）传入，避免写出永远无法触发的死分支。

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