阶段三03-SessionCookie机制和优化

在执行单个用例之前我们就可以带上 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 vallocal_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_):  # 也不允许直接importraise Exception(f'非法导入{input_}')# 约定可提供的数据 resultlocal_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']