如何在Django单元测试中遍历所有路由对应的View对象？

要确保项目里所有View（包括以后新增的）都符合合规要求，你可以按下面的步骤实现自动化测试：

1. 批量获取所有View类 #

首先得把项目里所有的View类都找出来，以Django为例，用importlib和inspect模块遍历项目模块：

import importlib
import inspect
from django.views.generic import View

def get_all_view_classes():
    # 把"your_project"替换成你的项目根模块名称
    project_root = importlib.import_module("your_project")
    all_views = []

    def scan_module(module):
        # 遍历模块里的成员
        for name, obj in inspect.getmembers(module):
            # 递归遍历子模块
            if inspect.ismodule(obj) and obj.__name__.startswith("your_project"):
                scan_module(obj)
            # 收集所有继承自View的自定义类
            elif inspect.isclass(obj) and issubclass(obj, View) and obj != View:
                all_views.append(obj)

    scan_module(project_root)
    return all_views

如果是Flask的类视图，逻辑类似，把继承的基类换成flask.views.View即可；要是函数视图，就改成遍历注册的路由获取视图函数。

2. 编写合规性测试用例 #

拿到所有View后，循环检查每个View是否具备合规属性/行为：

def test_view_compliance():
    for view_cls in get_all_view_classes():
        # 检查是否存在指定合规属性
        assert hasattr(view_cls, 'some_compliance_thing'), \
            f"View类 {view_cls.__name__} 未满足合规要求：缺少属性 'some_compliance_thing'"
        
        # 如果需要检查方法可调用性，改成下面的断言
        # assert hasattr(view_cls, 'compliance_method') and callable(getattr(view_cls, 'compliance_method')), \
        #     f"View类 {view_cls.__name__} 未满足合规要求：缺少可调用的方法 'compliance_method'"

3. 扩展优化（可选） #

• 排除特殊View：如果某些View不需要遵守规则，加个黑名单过滤：

  excluded_views = [SkipComplianceView, AdminDebugView]
  for view_cls in get_all_view_classes():
      if view_cls in excluded_views:
          continue
      # 执行断言检查
  

• 自动发现新增View：只要你的项目模块结构不变，这个扫描逻辑会自动覆盖新增的View类，不用手动更新测试用例。

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