阶段三03-Session&Cookie机制和优化

在执行单个用例之前我们就可以带上 suite_ctx。在perform_case里处理

 采用方法从下游到上游，还是用之前的方法改一下就能实现套件的参数传递，先给参数传递的方法加两个参数，默认为空。然后通过local_params暴露给用户

def proc_param_expression(val, case_ctx=None, suite_ctx=None, proj_ctx=None):
    """
    处理用例接口参数中的python表达式，将表达式替换为结果
    :param case_ctx:
    :param suite_ctx:
    :param proj_ctx:
    :param val:
    :return:
    """
    ms = re.findall(r"#\{.+\}", val)
    if len(ms) == 0:
        return val
    local_params = {
        'case_ctx': case_ctx,
        'suite_ctx': suite_ctx,
        'proj_ctx': proj_ctx,
        're': re,
        'parse': common.parse_json_like
    }

    def new_content(matched):
        s = matched.group()
        if re.search(r"__.+__", s):  # __import__.os 等被过滤掉
            raise Exception(f'python表达式包含非法字符或操作[{s}]')
        return str(eval(s[2:-1], {}, local_params))

    return re.sub(r"#\{.+\}", new_content, val)

相应的调用 proc_param_expression函数时也要加传参 get_request_bodies 等3个

接下来把所有用到case_ctx都检查一遍，加上suite_ctx和proj_ctx。

再回到perform_case方法查找用法。

异常处理

图中情况是上一个用例没有给suite_ctx token的值，后一个用例直接去拿 suite_ctx['token']导致的

前置处理和参数预处理没有获取异常，所以或导致系统崩溃。

但是目前以我们的水平可能无法写出可读性很好的异常处理，所有干脆在之前的异常处理前面再try一次。 perform_case 方法里：

 如果 exec_py_script 、 proc_apidef_params 、item.get_query_params等抛出异常的话就会记录堆栈信息到info日志里，用户勾选了失败后终止就break

下面代基本不变。

 只是捕获异常后判断，如果是RespCheckException异常直接返回改接口的校验是吧；如果是其他异常则记录堆栈信息到日志。并且最后通过判断接口是否执行成功（89行）

参数的重复使用

完成异常处理后我们完善好用例，我们想到的是在登陆用例里把token放到case_ctx 和 suite_ctx.

那么就出现了另一个问题了，exec_py_script现在只支持单条  #｛｝，我们要改成支持多条

def exec_py_script(input_, result, case_ctx=None, suite_ctx=None, proj_ctx=None):

    if not input_:
        return

    # m = re.match(r"#\{.+\}", input_)
    # if not m:
    #     raise Exception(f'内容格式不支持，请使用#{{}}包含[{input_}]')
    # exp = m.group()

    ms = re.findall(r"#\{.+}", input_)
    if len(ms) == 0:
        raise Exception(f'内容格式不支持，请使用#{{}}包含[{input_}]')

    if re.search(r"__.+__", input_):  # __import__.os 等被过滤掉
        raise Exception(f'输入了python表达式不允许操作{input_}')
    if re.search(r"import[ \s]\w]", input_):  # 也不允许直接import
        raise Exception(f'非法导入{input_}')
    # 约定可提供的数据 result
    local_params = {
        'result': result,
        're': re,
        'parse': common.parse_json_like,
        'case_ctx': case_ctx,
        'suite_ctx': suite_ctx,
        'proj_ctx': proj_ctx,
    }
    for exp in ms:
        exec(exp[2:-1], {}, local_params)  # eval执行后会返回一个布尔值,[2:-1]做个切片去掉用户输入的#｛｝

之后在去执行多用例，OK了，然而事情没有那么顺利，re会这样匹配.导致 前后两个都用不了

我们希望不用户不换行写也行，就需要再改下re的匹配，让它不那么贪婪。改成

ms = re.findall(r"#\{.+?\}", input_)  就行了

Session

目前项目所有的接口、用例都混在一起，没有通过项目隔离。我们想要实现只展示此账号所在项目的接口、用例。就要用到  django的Session机制

 Session放在我们创建的mysql表里

用测试平台举例:在登录时创建一个Session id，退出时销毁

 一般框架都是一个账户只能有一个Session id。新id产生时删掉旧id

按缺省项目过滤数据

我们之前没有使用过滤，接口里面展示的是所有的。现在我们希望按照所选的项目来展示接口

设计实现方法，先在项目里设置默认项目

    def select_pj(self, request, queryset):
        proj = queryset.first()  # queryset.first取到的就是 Model的对象
        logger = logging.getLogger('test_plt')
        # 缺省项目存到django的 request.session里,用字典格式维护
        request.session['default_project_id'] = proj.id
        # pj默认调用__str__
        self.message_user(request, f'默认项目为【{proj}】,后续操作基于此项目。您可以随时重新切换', level=messages.WARNING)

    select_pj.short_description = '设为默认项目'

在接口定义的model里

    def get_queryset(self, request):
        """
        重写获取queryset方法，调用session里的default_project_id；实现选择默认项目后【接口定义】中展示该项目的所有接口
        :param request:
        :return:
        """
        qs: QuerySet = super().get_queryset(request)
        proj_id = request.session.get('default_project_id', default=None)
        return qs.filter(pj__id=proj_id) if proj_id else qs

其他有需要过滤的模块，上面代码可以复用。唯一需要确定的是 使用的model中有无pj__id的外键

完成其他模块优化

首先是我们选择了默认项目以后，希望进入新建接口时只能选择我们之前给定的默认

在ApiDefAdmin的get_form方法里引入sesstion，默认为空。如果用户之前选了默认项目，则测试项目里只能选择该项目。环境只能选择该项目的环境

 192行，通过get_form方法的obj。如果obj为空则填入用户选择的默认项目

接下来是用例套件中的用例只展示默认项目

在CaseSuiteAdmin里新建一个方法

def formfield_for_manytomany(self, db_field, request, **kwargs):
    proj_id = request.session.get('default_project_id', default=None)
    if proj_id and db_field.name == 'cases':
        kwargs['queryset'] = Case.objects.filter(pj_id=proj_id)
    return super().formfield_for_manytomany(db_field, request, **kwargs)

重写formfield_for_manytomany方法，需要注意的是这个方法是匹配该modles所有多对多方法。所以我们加个db_field == 'cases'的限制。

方法地址：Django 管理站点 | Django 文档 | Django

用例接口的过滤

此时新建用例时选择的接口还没过滤，显示的是所有项目的接口

我们期望仅展示选择的默认项目里的接口，又因为这个是内联的表格。所以我们找到admin中的CaseApiDefInline，新建一个过滤方法使用formfield_for_foreignkey，和上面代码基本一样。

只需要

if proj_id and db_field.name == 'api':
kwargs['queryset'] = ApiDef.objects.filter(pj_id=proj_id)

最后完成 部署环境、测试成员的新建时的默认值就行了。使用上面form.base_fields['字段名'].initial

总结，缺省值过滤实践中

对于列表页：get_queryset

对于inline（在内联表里）：formfield_for_foreignkey 之 kwargs['queryset']

对于普通表单：get_form 之 form.base_fields['pj'].queryset

对于manytomany（多对多表）：formfield_for_manytomany 之 kwargs['queryset']