Django+vue自动化测试平台（24）-- 接口自动化之处理变量

众所周知，接口自动化用例最重要的是测试数据，而测试用例本质上依赖各种数据的组合。
做过接口自动化的朋友可以想一下，我们在使用postman或JMeter编写用例脚本的时候，比较费时间的就是接口间参数的传递了。
而通过现有工具实现接口间参数传递，只需要在下游接口中的"变量"设置{{}｝或者${}即可，而如何替换的我们似乎并未考虑过。
在我们设计自己的框架时就不得不面对这个问题了。大家都知道http协议接口的请求参数格式大多是json格式，但本文介绍的service服务的参数格式是对象。

参数化如何实现？

python 复制代码

{
  "env_id": 1,
  "test_obj": "{{test_pre}}",
  "test_list": [
    {"test_obj1": "{{code}}"},
    {"test_obj2": "{{message}}"}
    ]
}
这样一段引用{{xxx}}变量的json对象怎么办，如何转化成一下
{
  "env_id": 1,
  "test_obj": "123",
  "test_list": [
    {"test_obj1": "100"},
    {"test_obj2": "登录成功"}
  ]
}


这个是做参数化的关键步骤，废话少说直接看实现：
# def resolve_var(data, env_id)：定义了一个名为 resolve_var 的函数，接收要处理的数据 data 和环境标识 env_id。
# 首先获取与环境变量相关的模型对象 env_var 和全局变量相关的模型对象 global_var。
# 定义了空的字典 global_vars 和 local_vars。
# 然后检查数据类型，如果是字典：
# 遍历字典的键值对，对于值是字符串且包含特定模板标记（{{ 和 }}）的情况，分别在环境变量和全局变量中根据变量名和环境标识进行查找，如果找到对应的值，就用找到的值替换原字符串；如果都没找到，就用原字符串加上"变量未找到"来替换。对于值不是字符串的情况，递归调用自身进行处理。
# 如果是列表：则对列表中的每个元素递归调用进行处理。
# 如果是字符串且包含特定模板标记：
# 通过正则表达式提取出变量名，然后同样在环境变量和全局变量中查找并进行相应替换操作。
def resolve_var(data, env_id):
    env_var = Env_var.objects  # 环境变量表
    global_var = Global_var.objects  # 全局变量表
    global_vars = {}
    local_vars = {}
    if isinstance(data, dict):
        for key, value in data.items():
            if isinstance(value, str):
                if '{{' in value and '}}' in value:
                    env_var_value = env_var.filter(var_name=value, env_id=env_id)
                    global_var_value = global_var.filter(var_name=value)
                    if env_var_value.count() > 0:
                        data[key] = value.replace(value, env_var_value[0].var_value)
                    elif global_var_value.count() > 0:
                        data[key] = value.replace(value, global_var_value[0].var_value)
                    else:
                        data[key] = value.replace(value, value + "变量未找到")

            else:
                resolve_var(value, env_id)
    elif isinstance(data, list):
        for item in data:
            resolve_var(item, env_id)
    elif isinstance(data, str):
        if "{{" in data and "}}" in data:
            key = re.findall(r"\{{(.+?)}}", data)
            for a in range(len(key)):
                v_key = "{{" + key[a] + "}}"
                env_var_value = env_var.filter(var_name=v_key, env_id=env_id)
                global_var_value = global_var.filter(var_name=v_key)
                if env_var_value.count() > 0:
                    data = data.replace(v_key, env_var_value[0].var_value)
                elif global_var_value.count() > 0:
                    data = data.replace(v_key, global_var_value[0].var_value)
                else:
                    data = data.replace(v_key, v_key + "变量未找到")
    return data

通过以上的代码，实现了从变量{{xxx}}到实际值的替换，传入dict，list，str类型的数据都可以，解决了变量的问题。（个人短见，仅供参考）