在执行单个用例之前我们就可以带上 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']