floraachy/apiautotest
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1. 优化日志信息 2. 每一个参数替换增加异常捕获，方便定位问题 3. cookies进行参数替换时，增加特殊处理

Browse Source
This commit is contained in:
floraachy
2023-05-27 14:58:34 +08:00
parent cf71d3fdac
commit e8e27e8a53
3 changed files with 123 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files
case_utils/request_data_handle.py
+73 -38
View File
@@ -5,7 +5,9 @@
# @File : request_data_handle.py # @File : request_data_handle.py
# @Software: PyCharm # @Software: PyCharm
# @Desc: 处理request请求前后的用例数据 # @Desc: 处理request请求前后的用例数据
import json
import requests
from common_utils.data_handle import eval_data_process, data_replace from common_utils.data_handle import eval_data_process, data_replace
from config.global_vars import GLOBAL_VARS from config.global_vars import GLOBAL_VARS
@@ -26,17 +28,17 @@ class RequestPreDataHandle:
def __init__(self, request_data): def __init__(self, request_data):
logger.debug(f"\n======================================================\n" \ logger.debug(f"\n======================================================\n" \
"-------------Start：处理用例数据前--------------------\n" "-------------Start：处理用例数据前--------------------\n"
f"用例标题: {request_data.get('title', None)}\n" \ f"用例标题: {type(request_data.get('title', None))} || {request_data.get('title', None)}\n" \
f"请求路径: {request_data.get('url', None)}\n" \ f"请求路径: {type(request_data.get('url', None))} || {request_data.get('url', None)}\n" \
f"请求方式: {request_data.get('method', None)}\n" \ f"请求方式: {type(request_data.get('method', None))} || {request_data.get('method', None)}\n" \
f"请求头: {request_data.get('headers', None)}\n" \ f"请求头: {type(request_data.get('headers', None))} || {request_data.get('headers', None)}\n" \
f"请求cookies: {request_data.get('cookies', None)}\n" \ f"请求cookies: {type(request_data.get('cookies', None))} || {request_data.get('cookies', None)}\n" \
f"请求关键字: {request_data.get('pk', None)}\n" \ f"请求类型: {type(request_data.get('pk', None))} || {request_data.get('pk', None)}\n" \
f"请求内容: {request_data.get('payload', None)}\n" \ f"请求内容: {type(request_data.get('payload', None))} || {request_data.get('payload', None)}\n" \
f"请求文件: {request_data.get('files', None)}\n" \ f"请求文件: {type(request_data.get('files', None))} || {request_data.get('files', None)}\n" \
f"后置提取参数: {request_data.get('extract', None)}\n" \ f"后置提取参数: {type(request_data.get('extract', None))} || {request_data.get('extract', None)}\n" \
f"响应断言: {request_data.get('assert_response', None)}\n" \ f"响应断言: {type(request_data.get('assert_response', None))} || {request_data.get('assert_response', None)}\n" \
f"数据库断言: {request_data.get('assert_sql', None)}\n" \ f"数据库断言: {type(request_data.get('assert_sql', None))} || {request_data.get('assert_sql', None)}\n" \
"=====================================================") "=====================================================")
self.request_data = request_data self.request_data = request_data
@@ -54,21 +56,22 @@ class RequestPreDataHandle:
self.assert_handle() self.assert_handle()
logger.debug(f"\n======================================================\n" \ logger.debug(f"\n======================================================\n" \
"-------------End：处理用例数据后--------------------\n" "-------------End：处理用例数据后--------------------\n"
f"用例标题: {self.request_data.get('title', None)}\n" \ f"用例标题: {type(self.request_data.get('title', None))} || {self.request_data.get('title', None)}\n" \
f"请求路径: {self.request_data.get('url', None)}\n" \ f"请求路径: {type(self.request_data.get('url', None))} || {self.request_data.get('url', None)}\n" \
f"请求方式: {self.request_data.get('method', None)}\n" \ f"请求方式: {type(self.request_data.get('method', None))} || {self.request_data.get('method', None)}\n" \
f"请求头: {self.request_data.get('headers', None)}\n" \ f"请求头: {type(self.request_data.get('headers', None))} || {self.request_data.get('headers', None)}\n" \
f"请求cookies: {self.request_data.get('cookies', None)}\n" \ f"请求cookies: {type(self.request_data.get('cookies', None))} || {self.request_data.get('cookies', None)}\n" \
f"请求关键字: {self.request_data.get('pk', None)}\n" \ f"请求类型: {type(self.request_data.get('pk', None))} || {self.request_data.get('pk', None)}\n" \
f"请求内容: {self.request_data.get('payload', None)}\n" \ f"请求内容: {type(self.request_data.get('payload', None))} || {self.request_data.get('payload', None)}\n" \
f"请求文件: {self.request_data.get('files', None)}\n" \ f"请求文件: {type(self.request_data.get('files', None))} || {self.request_data.get('files', None)}\n" \
f"后置提取参数: {self.request_data.get('extract', None)}\n" \ f"后置提取参数: {type(self.request_data.get('extract', None))} || {self.request_data.get('extract', None)}\n" \
f"响应断言: {self.request_data.get('assert_response', None)}\n" \ f"响应断言: {type(self.request_data.get('assert_response', None))} || {self.request_data.get('assert_response', None)}\n" \
f"数据库断言: {self.request_data.get('assert_sql', None)}\n" \ f"数据库断言: {type(self.request_data.get('assert_sql', None))} || {self.request_data.get('assert_sql', None)}\n" \
"=====================================================") "=====================================================")
return self.request_data return self.request_data
def url_handle(self): def url_handle(self):
try:
""" """
用例数据中获取到的url(一般是不带host的，个别特殊的带有host，则不进行处理) 用例数据中获取到的url(一般是不带host的，个别特殊的带有host，则不进行处理)
""" """
@@ -95,28 +98,48 @@ class RequestPreDataHandle:
# 如果host不以/结尾 且 url不以/开头，则将host和url拼接起来的时候增加/，组成新的url # 如果host不以/结尾 且 url不以/开头，则将host和url拼接起来的时候增加/，组成新的url
full_url = host + "/" + url full_url = host + "/" + url
self.request_data["url"] = full_url self.request_data["url"] = full_url
except Exception as e:
logger.error(f"处理url报错了：{e}")
print(f"处理url报错了：{e}")
def method_handle(self): def method_handle(self):
# TODO 暂时不需要处理，后续有需要在处理 # TODO 暂时不需要处理，后续有需要在处理
pass pass
def cookies_handle(self): def cookies_handle(self):
try:
# 从用例数据中获取cookies， 处理cookies # 从用例数据中获取cookies， 处理cookies
if self.request_data.get("cookies", None): if self.request_data.get("cookies", None):
self.request_data["cookies"] = eval_data_process( # 检测替换后的cookies是否为字典类型，如果不是，则使用json.loads转换
data_replace(content=self.request_data.get("cookies", None), source=GLOBAL_VARS)) cookies = data_replace(content=self.request_data.get("cookies"), source=GLOBAL_VARS)
if isinstance(cookies, str):
# 使用json的loads函数，把str转化为字典，再将字典恢复成原来的cookies
self.request_data["cookies"] = requests.utils.cookiejar_from_dict(json.loads(cookies))
else:
self.request_data["cookies"] = cookies
except Exception as e:
logger.error(f"处理cookies报错了：{e}")
print(f"处理cookies报错了：{e}")
def headers_handle(self): def headers_handle(self):
try:
# 从用例数据中获取header， 处理header # 从用例数据中获取header， 处理header
if self.request_data.get("headers", None): if self.request_data.get("headers", None):
self.request_data["headers"] = eval_data_process( self.request_data["headers"] = eval_data_process(
data_replace(content=self.request_data.get("headers", None), source=GLOBAL_VARS)) data_replace(content=self.request_data.get("headers"), source=GLOBAL_VARS))
except Exception as e:
logger.error(f"处理header报错了：{e}")
print(f"处理header报错了：{e}")
def payload_handle(self): def payload_handle(self):
try:
# 处理请求参数payload # 处理请求参数payload
if self.request_data.get("payload", None): if self.request_data.get("payload", None):
self.request_data["payload"] = eval_data_process( self.request_data["payload"] = eval_data_process(
data_replace(content=self.request_data.get("payload", None), source=GLOBAL_VARS)) data_replace(content=self.request_data.get("payload"), source=GLOBAL_VARS))
except Exception as e:
logger.error(f"处理payload报错了：{e}")
print(f"处理payload报错了：{e}")
def files_handle(self): def files_handle(self):
# 处理文件 # 处理文件
@@ -124,17 +147,25 @@ class RequestPreDataHandle:
pass pass
def extract_handle(self): def extract_handle(self):
try:
# 处理后置提取参数 # 处理后置提取参数
if self.request_data.get("extract", None): if self.request_data.get("extract", None):
# 仅提取参数中的python表达式，不需要进行数据替换 # 仅提取参数中的python表达式，不需要进行数据替换
self.request_data["extract"] = eval_data_process(self.request_data.get("extract", None)) self.request_data["extract"] = eval_data_process(self.request_data.get("extract"))
except Exception as e:
logger.error(f"处理extract报错了：{e}")
print(f"处理extract报错了：{e}")
def assert_handle(self): def assert_handle(self):
try:
# 处理响应断言参数 # 处理响应断言参数
if self.request_data.get("assert_response", None): if self.request_data.get("assert_response", None):
self.request_data["assert_response"] = eval_data_process( self.request_data["assert_response"] = eval_data_process(
data_replace(content=self.request_data.get("assert_response", None), source=GLOBAL_VARS)) data_replace(content=self.request_data.get("assert_response"), source=GLOBAL_VARS))
# 由于数据库断言里面的变量需要请求响应后进行提取，因此目前不进行处理 # 由于数据库断言里面的变量需要请求响应后进行提取，因此目前不进行处理
except Exception as e:
logger.error(f"处理assert报错了：{e}")
print(f"处理assert报错了：{e}")
# ---------------------------------------- 进行请求，请求后的参数提取处理----------------------------------------# # ---------------------------------------- 进行请求，请求后的参数提取处理----------------------------------------#
@@ -157,14 +188,14 @@ class RequestHandle:
data_replace(content=self.case_data["assert_sql"], source=GLOBAL_VARS)) data_replace(content=self.case_data["assert_sql"], source=GLOBAL_VARS))
logger.info(f"\n======================================================\n" \ logger.info(f"\n======================================================\n" \
"-------------请求数据--------------------\n" "-------------请求数据--------------------\n"
f"用例标题: {self.case_data.get('title', None)}\n" \ f"用例标题: {type(self.case_data.get('title', None))} || {self.case_data.get('title', None)}\n" \
f"请求路径: {self.case_data.get('url', None)}\n" \ f"请求路径: {type(self.case_data.get('url', None))} || {self.case_data.get('url', None)}\n" \
f"请求方式: {self.case_data.get('method', None)}\n" \ f"请求方式: {type(self.case_data.get('method', None))} || {self.case_data.get('method', None)}\n" \
f"请求头: {self.case_data.get('headers', None)}\n" \ f"请求头: {type(self.case_data.get('headers', None))} || {self.case_data.get('headers', None)}\n" \
f"请求cookies: {self.case_data.get('cookies', None)}\n" \ f"请求cookies: {type(self.case_data.get('cookies', None))} || {self.case_data.get('cookies', None)}\n" \
f"请求关键字: {self.case_data.get('pk', None)}\n" \ f"请求类型: {type(self.case_data.get('pk', None))} || {self.case_data.get('pk', None)}\n" \
f"请求内容: {self.case_data.get('payload', None)}\n" \ f"请求内容: {type(self.case_data.get('payload', None))} || {self.case_data.get('payload', None)}\n" \
f"请求文件: {self.case_data.get('files', None)}\n" \ f"请求文件: {type(self.case_data.get('files', None))} || {self.case_data.get('files', None)}\n" \
f"请求响应数据: {response.text}\n" \ f"请求响应数据: {response.text}\n" \
f"请求响应码: {response.status_code}\n" \ f"请求响应码: {response.status_code}\n" \
f"响应耗时: {round(response.elapsed.total_seconds(), 2)} s || {round(response.elapsed.total_seconds() * 1000, 2)} ms\n" \ f"响应耗时: {round(response.elapsed.total_seconds(), 2)} s || {round(response.elapsed.total_seconds() * 1000, 2)} ms\n" \
@@ -172,12 +203,13 @@ class RequestHandle:
allure_step(step_title=f"请求地址:{self.case_data['url']}") allure_step(step_title=f"请求地址:{self.case_data['url']}")
allure_step(step_title=f"请求方式：{self.case_data['method']}") allure_step(step_title=f"请求方式：{self.case_data['method']}")
allure_step(step_title="请求头", content=self.case_data['headers']) allure_step(step_title="请求头", content=self.case_data['headers'])
allure_step(step_title="请求Cookies", content=self.case_data['cookies']) allure_step(step_title="请求Cookies", content=str(self.case_data['cookies']))
allure_step(step_title="请求参数", content=self.case_data['payload']) allure_step(step_title="请求参数", content=self.case_data['payload'])
allure_step(step_title="请求文件", content=self.case_data['files']) allure_step(step_title="请求文件", content=self.case_data['files'])
allure_step(step_title="请求响应数据", content=response.text) allure_step(step_title="请求响应数据", content=response.text)
allure_step(step_title=f"请求响应码:{response.status_code}") allure_step(step_title=f"请求响应码:{response.status_code}")
allure_step(step_title=f"响应耗时:{round(response.elapsed.total_seconds(), 2)} s || {round(response.elapsed.total_seconds() * 1000, 2)} ms") allure_step(
step_title=f"响应耗时:{round(response.elapsed.total_seconds(), 2)} s || {round(response.elapsed.total_seconds() * 1000, 2)} ms")
return response return response
@@ -195,6 +227,7 @@ def after_request_extract(response: Response, extract):
"=====================================================") "=====================================================")
result = {} result = {}
if extract: if extract:
try:
if response_type(response) == "json": if response_type(response) == "json":
# 如果响应数据是json格式，则将按照json方式对后置提取参数进行处理 # 如果响应数据是json格式，则将按照json方式对后置提取参数进行处理
res = response.json() res = response.json()
@@ -205,6 +238,8 @@ def after_request_extract(response: Response, extract):
res = response.text res = response.text
for k, v in extract.items(): for k, v in extract.items():
result[k] = re_extract(res, v) result[k] = re_extract(res, v)
except Exception as e:
logger.error(f"提取后置参数报错：{e}")
logger.debug(f"\n======================================================\n" \ logger.debug(f"\n======================================================\n" \
"-------------End：从响应数据中提取后置参数保存到全局变量--------------------\n" "-------------End：从响应数据中提取后置参数保存到全局变量--------------------\n"
f"后置提取参数（新）: {result}\n" \ f"后置提取参数（新）: {result}\n" \
data/1test_login_demo.yaml → data/test_login_demo.yaml
View File
test_case/test_manual_case/1test_demo.py → test_case/test_manual_case/test_demo.py
View File