81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @Time : 2023/11/8 16:53
|
||
# @Author : floraachy
|
||
# @File : api_workflow.py
|
||
# @Software: PyCharm
|
||
# @Desc:
|
||
|
||
|
||
# 标准库导入
|
||
import os
|
||
# 第三方库导入
|
||
from loguru import logger
|
||
# 本地应用/模块导入
|
||
from utils.files_utils.yaml_handle import YamlHandle
|
||
from utils.requests_utils.request_control import RequestPreDataHandle, RequestHandle, after_request_extract
|
||
from utils.files_utils.files_handle import get_files
|
||
from utils.data_utils.eval_data_handle import eval_data
|
||
from utils.assertion_utils.assert_control import AssertHandle
|
||
from config.global_vars import GLOBAL_VARS
|
||
|
||
|
||
def get_api_data(api_file_path: str, key: str = None):
|
||
"""
|
||
根据指定的yaml文件路径，以及key值，获取对应的接口
|
||
:param:api_file_path 接口yaml文件路径
|
||
:param:key 对应接口的id
|
||
"""
|
||
api_data = []
|
||
if os.path.isdir(api_file_path):
|
||
logger.debug(f"目标路径是一个目录：{api_file_path}")
|
||
api_files = get_files(target=api_file_path, end=".yaml") + get_files(target=api_file_path, end=".yml")
|
||
for api_file in api_files:
|
||
api_data.append(YamlHandle(filename=api_file).read_yaml)
|
||
|
||
elif os.path.isfile(api_file_path):
|
||
logger.debug(f"目标路径是一个文件：{api_file_path}")
|
||
api_data.append(YamlHandle(filename=api_file_path).read_yaml)
|
||
|
||
else:
|
||
logger.error(f"目标路径错误，请检查！api_file_path={api_file_path}")
|
||
return None
|
||
|
||
for api in api_data:
|
||
matching_api = next((item for item in api["case_info"] if item["id"] == key), None)
|
||
if matching_api:
|
||
logger.info("\n----------匹配到的api----------\n"
|
||
f"类型：{type(matching_api)}"
|
||
f"值：{matching_api}\n")
|
||
return matching_api
|
||
|
||
# 在找不到匹配的情况下，返回一个默认值且记录一条错误日志
|
||
logger.warning(f"未找到id为{key}的接口， 返回值是None")
|
||
raise Exception(f"未找到id为{key}的接口， 返回值是None")
|
||
|
||
|
||
def api_work_flow(req_data: dict, source: dict) -> dict:
|
||
"""
|
||
请求过程：请求前用例数据处理，发送请求，断言，参数提取
|
||
:param:req_data 接口请求数据
|
||
:param:source 全局变量，保存接口相关变量的实际值， 例如接口中的${login}，会从source中找到key=login的元素进行替换
|
||
"""
|
||
logger.debug(f"\n----------------api_work_flow-----------------\n"
|
||
f"接口请求数据：{req_data}\n"
|
||
f"全局变量：{source}\n")
|
||
if req_data:
|
||
extract_result = {}
|
||
api_data = RequestPreDataHandle(request_data=req_data, global_var=source).request_data_handle()
|
||
# 发送请求
|
||
response = RequestHandle(case_data=api_data, global_var=source).http_request()
|
||
# 进行响应断言
|
||
AssertHandle(assert_data=api_data["assert_response"], response=response).assert_handle()
|
||
# 断言成功后进行参数提取
|
||
res = after_request_extract(response, api_data["extract"])
|
||
for k, v in res.items():
|
||
extract_result[k] = eval_data(v)
|
||
return extract_result
|
||
else:
|
||
logger.error(f"接口请求数据不能为空！\n"
|
||
f"req_data = {req_data}")
|
||
raise f"接口请求数据不能为空！\nreq_data = {req_data}"
|