首先打开项目,找到我们的并发文件:wqrf_run_case.py
然后我们先回忆下,这个文件我们目前很多的初始化步骤,其实都是把请求数据从数据库step表中拿出来,拿出来的时候基本都是字符串,我们又合并了一些全局变量替换。
接下来继续:
1. 请求体类型初始化:
2. 是否mock判断
这个判断很重要,如果确定mock_res是有内容的状态,那我们其实什么都可以不做了,直接返回数据库提取的mock_res内容即可。
看上图,注意,这里我们新建了一个叫res的变量,它将作为我们最终的返回值,哪怕是非mock的正常请求,最终返回值也依然是res。
还有,这里我们用not in 来对mock_res进行判断,好处是写的简单,方便之后对列表内容进行增减。
3. 插播个请求头为空处理
看上图,我们对空字符串进行赋值为{}字符串,目的是后面对api_header进行转字典时安全不报错。
还有,像这种字段的异常处理,其实也可以写很多,也可以单独用个函数处理,只要保证一点即可,就是在使用它们之前处理。
4. 占位符处理
也就是##变量名## 这种,是在当本step并非第一个step,而是在之前step中有返回值需要用在这里当做请求数据的情况才会出现。简单的用正则找出并替换。后面还有一些针对请求数据的处理步骤,但是之所以放在占位符之后,就是防止影响。
别忘了在文件开头引入re:
首先,是检测url中的占位符并处理:
第一行是找出所有被## ## 包含的中间变量名字,成为一个列表rlist_url
第二行是遍历它的循环
第三行是进行替换,用eval求值的结果替换,如果前面步骤生成了该变量,此处则可eval成功,否则会报错导致用例失败。
然后是header:
然后是请求体:
请求体处理这步骤,比较多,其实逻辑简单,只是针对不同的请求体类型做判断。这个具体优化有待商榷,不过目前我们还是直接从run_case.py中直接复制过来使用。
为了方便大家抄写,占位符替换步骤,提供复制文案:
####占位符变量替换 # url 处理 rlist_url = re.findall(r"##(.*?)##", api_url) for i in rlist_url: api_url = api_url.replace("##" + i + "##", str(eval(i))) # header 处理 rlist_header = re.findall(r"##(.*?)##", api_header) for i in rlist_header: api_header = api_header.replace("##" + i + "##", repr(str(eval(i)))) # 请求体 处理 if api_body_method == 'none': pass elif api_body_method == 'form-data' or api_body_method == 'x-www-form-urlencoded': rlist_body = re.findall(r"##(.*?)##", api_body) for i in rlist_body: api_body = api_body.replace("##" + i + "##", str(eval(i))) elif api_body_method == 'Json': rlist_body = re.findall(r"##(.*?)##", api_body) for i in rlist_body: api_body = api_body.replace("##" + i + "##", repr(eval(i))) else: rlist_body = re.findall(r"##(.*?)##", api_body) for i in rlist_body: api_body = api_body.replace("##" + i + "##", str(eval(i)))