【接口自动化框架】测试数据部分增加替换变量功能
我们的接口自动化框架的测试数据放到json文件中，
但是有些参数是动态生成的(例如随机字符串)；有些参数可以进行再次提取，放在config文件里（例如host，URL，accountid等等）；
解决这个问题我想到两个解决方法：
1，在测试case的python文件中生成变量后再赋值给json文件中的变量（缺点：重复代码多）
2，在json文件中直接引用函数 or 变量名称，在case读取json文件时，替换变量（此文重点介绍这种方法）
进入正题，第一步：替换变量。
首先介绍一下我的测试数据是放在json文件里的，形式如下；比如像host，uri，account_id 是需要替换的
"test": [
{
"tag":"test render string",
"case": "测试如何替换测试参数",
"http": {
"host": "1",
"uri": "1",
"method": "POST",
"headers": {
"Content-Type": "application/x-www-form-urlencoded"
},
"params": {
"region": "cn",
"lang": "zh-cn",
"app_id": "10001",
"nonce": "${generate_random_digits()}"
},
"data": {
"account_id": "{user.owner.account_id}"
"mobile":"{user.owner.mobile}"
}
},
"expected": {
"response": {
"result_code": "success",
},
"database": {},
"kafka": {}
}}
]
}第二步：
我们把例如account_id这种数据放在config文件夹下，可以存放为yaml活着json都可以；只要读取数据出来的形式是字典就行
cache = {
"VARIABLES":{
"user":{
"owner":{
"account_id":"test123",
"mobile": "13677777777"
}
}}
}至此，需要替换的数据是第一步读取的；要替换的target数据是第二步读取的
第三步：编写render函数进行数据替换，这里替换的是变量
import logging
import re
p = '''{
"test": [
{
"tag":"test render string",
"case": "测试如何替换测试参数",
"http": {
"host": "1",
"uri": "1",
"method": "POST",
"headers": {
"Content-Type": "application/x-www-form-urlencoded"
},
"params": {
"region": "cn",
"lang": "zh-cn",
"app_id": "10001",
"nonce": "${generate_random_digits()}"
},
"data": {
"account_id": "{user.owner.account_id}"
"mobile":"{user.owner.mobile}"
}
},
"expected": {
"response": {
"result_code": "success",
},
"database": {},
"kafka": {}
}}
]
}'''
cache = {
"VARIABLES":{
"user":{
"owner":{
"account_id":"1559759948",
"mobile": "98762424749"
}
}}
}
varial_matcher = re.compile(
r"\{([\w.]+)\}"
r"|\$\{([\w.]+)\}" # 匹配$开头并{}包括的变量
)
def render(p, cache):
# 替换变量
if isinstance(p, str) and varial_matcher.findall(p):
variables_list = []
try:
for var_tuple in varial_matcher.findall(p):
variables_list.append(
var_tuple[0] or var_tuple[1]
)
for items in variables_list:
items_list = items.split('.')
target_data = cache.get("VARIABLES")
for i in items_list:
target_data = target_data.get(i, {})
if target_data != {}:
p = p.replace('${', '{').replace("{" + items + "}", str(target_data))
return p
except Exception as e:
logging.error(e)
logging.error(p)
print(render(p, cache))第四步：在测试数据的json文件中还有这种
"nonce": "${generate_random_digits(2)}"引用函数的，需要替换成引用函数后的结果；怎么做呢？和步骤三大同小异
func_matcher = re.compile(
r"[\$]{1}[\{]{1}" # 可以匹配${}的函数
r"([0-9a-zA-Z_.=]+)" # 匹配函数名称可以包含.
r"[ ]*" # 函数方法名称后0或多个空格
r"\((.*)\)" # 匹配函数参数
)
def render(p, cache):
# 替换函数
if isinstance(p, str) and func_matcher.findall(p):
for fn_tuple in func_matcher.findall(p):
fun, varials = fn_tuple[0], fn_tuple[1]
fnc_map = cache.get("FUNCS")
try:
fn_raw = "{fun}({varials})".format(fun=fun, varials=varials)
fn = "{fun}({varials})".format(fun=fun, varials=varials)
target_data = eval(fn, fnc_map)
p = p.replace('${', '{').replace("{" + fn_raw + "}", str(target_data))
except (ModuleNotFoundError, ImportError) as e:
logging.warning("函数没找到或者导入失败", p, cache)
except NameError as e:
logging.warning("err", e)
return p
print(render(p, cache))但是在这步前要再自行加一步，就是在cache的数据中，加入我们写好的FUNC参数，以及我们需要替换的函数文件
比如 我的替换函数文件：
#我的func_tools文件
def generate_random_digits(num):
return str(round(random.uniform(-90, 90), num))就是在cache的数据中，加入我们写好的FUNC参数
cache["FUNCS"] = load_func_tools("func_tools")def load_func_tools(module):
try:
module_functions = {}
module = import_module(module)
if module:
for name, item in vars(module).items():
if isinstance(item, types.FunctionType):
module_functions[name] = item
return module_functions
except Exception:
return {}