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
+123 -88
View File
@@ -5,7 +5,9 @@
# @File : request_data_handle.py
# @Software: PyCharm
# @Desc: 处理request请求前后的用例数据
import json
import requests
from common_utils.data_handle import eval_data_process, data_replace
from config.global_vars import GLOBAL_VARS
@@ -26,17 +28,17 @@ class RequestPreDataHandle:
def __init__(self, request_data):
logger.debug(f"\n======================================================\n" \
"-------------Start：处理用例数据前--------------------\n"
f"用例标题: {request_data.get('title', None)}\n" \
f"请求路径: {request_data.get('url', None)}\n" \
f"请求方式: {request_data.get('method', None)}\n" \
f"请求头: {request_data.get('headers', None)}\n" \
f"请求cookies: {request_data.get('cookies', None)}\n" \
f"请求关键字: {request_data.get('pk', None)}\n" \
f"请求内容: {request_data.get('payload', None)}\n" \
f"请求文件: {request_data.get('files', None)}\n" \
f"后置提取参数: {request_data.get('extract', None)}\n" \
f"响应断言: {request_data.get('assert_response', None)}\n" \
f"数据库断言: {request_data.get('assert_sql', None)}\n" \
f"用例标题: {type(request_data.get('title', None))} || {request_data.get('title', None)}\n" \
f"请求路径: {type(request_data.get('url', None))} || {request_data.get('url', None)}\n" \
f"请求方式: {type(request_data.get('method', None))} || {request_data.get('method', None)}\n" \
f"请求头: {type(request_data.get('headers', None))} || {request_data.get('headers', None)}\n" \
f"请求cookies: {type(request_data.get('cookies', None))} || {request_data.get('cookies', None)}\n" \
f"请求类型: {type(request_data.get('pk', None))} || {request_data.get('pk', None)}\n" \
f"请求内容: {type(request_data.get('payload', None))} || {request_data.get('payload', None)}\n" \
f"请求文件: {type(request_data.get('files', None))} || {request_data.get('files', None)}\n" \
f"后置提取参数: {type(request_data.get('extract', None))} || {request_data.get('extract', None)}\n" \
f"响应断言: {type(request_data.get('assert_response', None))} || {request_data.get('assert_response', None)}\n" \
f"数据库断言: {type(request_data.get('assert_sql', None))} || {request_data.get('assert_sql', None)}\n" \
"=====================================================")
self.request_data = request_data
@@ -54,69 +56,90 @@ class RequestPreDataHandle:
self.assert_handle()
logger.debug(f"\n======================================================\n" \
"-------------End：处理用例数据后--------------------\n"
f"用例标题: {self.request_data.get('title', None)}\n" \
f"请求路径: {self.request_data.get('url', None)}\n" \
f"请求方式: {self.request_data.get('method', None)}\n" \
f"请求头: {self.request_data.get('headers', None)}\n" \
f"请求cookies: {self.request_data.get('cookies', None)}\n" \
f"请求关键字: {self.request_data.get('pk', None)}\n" \
f"请求内容: {self.request_data.get('payload', None)}\n" \
f"请求文件: {self.request_data.get('files', None)}\n" \
f"后置提取参数: {self.request_data.get('extract', None)}\n" \
f"响应断言: {self.request_data.get('assert_response', None)}\n" \
f"数据库断言: {self.request_data.get('assert_sql', None)}\n" \
f"用例标题: {type(self.request_data.get('title', None))} || {self.request_data.get('title', None)}\n" \
f"请求路径: {type(self.request_data.get('url', None))} || {self.request_data.get('url', None)}\n" \
f"请求方式: {type(self.request_data.get('method', None))} || {self.request_data.get('method', None)}\n" \
f"请求头: {type(self.request_data.get('headers', None))} || {self.request_data.get('headers', None)}\n" \
f"请求cookies: {type(self.request_data.get('cookies', None))} || {self.request_data.get('cookies', None)}\n" \
f"请求类型: {type(self.request_data.get('pk', None))} || {self.request_data.get('pk', None)}\n" \
f"请求内容: {type(self.request_data.get('payload', None))} || {self.request_data.get('payload', None)}\n" \
f"请求文件: {type(self.request_data.get('files', None))} || {self.request_data.get('files', None)}\n" \
f"后置提取参数: {type(self.request_data.get('extract', None))} || {self.request_data.get('extract', None)}\n" \
f"响应断言: {type(self.request_data.get('assert_response', None))} || {self.request_data.get('assert_response', None)}\n" \
f"数据库断言: {type(self.request_data.get('assert_sql', None))} || {self.request_data.get('assert_sql', None)}\n" \
"=====================================================")
return self.request_data
def url_handle(self):
"""
用例数据中获取到的url(一般是不带host的，个别特殊的带有host，则不进行处理)
"""
# 检测url中是否存在需要替换的参数，如果存在则进行替换
data_replace(content=self.request_data.get("url", None), source=GLOBAL_VARS)
# 进行url处理，最终得到full_url
host = GLOBAL_VARS.get("host", "")
url = self.request_data.get("url", "")
# 从用例数据中获取url，如果键url不存在，则返回空字符串
# 如果url是以http开头的，则直接使用该url，不与host进行拼接
if url.lower().startswith("http"):
full_url = url
else:
# 如果host以/结尾 并且 url以/开头
if host.endswith("/") and url.startswith("/"):
full_url = host[0:len(host) - 1] + url
# 如果host以/结尾 并且 url不以/开头
elif host.endswith("/") and (not url.startswith("/")):
full_url = host + url
elif (not host.endswith("/")) and url.startswith("/"):
# 如果host不以/结尾 且 url以/开头，则将host和url拼接起来，组成新的url
full_url = host + url
try:
"""
用例数据中获取到的url(一般是不带host的，个别特殊的带有host，则不进行处理)
"""
# 检测url中是否存在需要替换的参数，如果存在则进行替换
data_replace(content=self.request_data.get("url", None), source=GLOBAL_VARS)
# 进行url处理，最终得到full_url
host = GLOBAL_VARS.get("host", "")
url = self.request_data.get("url", "")
# 从用例数据中获取url，如果键url不存在，则返回空字符串
# 如果url是以http开头的，则直接使用该url，不与host进行拼接
if url.lower().startswith("http"):
full_url = url
else:
# 如果host不以/结尾 且 url不以/开头，则将host和url拼接起来的时候增加/，组成新的url
full_url = host + "/" + url
self.request_data["url"] = full_url
# 如果host以/结尾 并且 url以/开头
if host.endswith("/") and url.startswith("/"):
full_url = host[0:len(host) - 1] + url
# 如果host以/结尾 并且 url不以/开头
elif host.endswith("/") and (not url.startswith("/")):
full_url = host + url
elif (not host.endswith("/")) and url.startswith("/"):
# 如果host不以/结尾 且 url以/开头，则将host和url拼接起来，组成新的url
full_url = host + url
else:
# 如果host不以/结尾 且 url不以/开头，则将host和url拼接起来的时候增加/，组成新的url
full_url = host + "/" + url
self.request_data["url"] = full_url
except Exception as e:
logger.error(f"处理url报错了：{e}")
print(f"处理url报错了：{e}")
def method_handle(self):
# TODO 暂时不需要处理，后续有需要在处理
pass
def cookies_handle(self):
# 从用例数据中获取cookies， 处理cookies
if self.request_data.get("cookies", None):
self.request_data["cookies"] = eval_data_process(
data_replace(content=self.request_data.get("cookies", None), source=GLOBAL_VARS))
try:
# 从用例数据中获取cookies， 处理cookies
if self.request_data.get("cookies", None):
# 检测替换后的cookies是否为字典类型，如果不是，则使用json.loads转换
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):
# 从用例数据中获取header， 处理header
if self.request_data.get("headers", None):
self.request_data["headers"] = eval_data_process(
data_replace(content=self.request_data.get("headers", None), source=GLOBAL_VARS))
try:
# 从用例数据中获取header， 处理header
if self.request_data.get("headers", None):
self.request_data["headers"] = eval_data_process(
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):
# 处理请求参数payload
if self.request_data.get("payload", None):
self.request_data["payload"] = eval_data_process(
data_replace(content=self.request_data.get("payload", None), source=GLOBAL_VARS))
try:
# 处理请求参数payload
if self.request_data.get("payload", None):
self.request_data["payload"] = eval_data_process(
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):
# 处理文件
@@ -124,17 +147,25 @@ class RequestPreDataHandle:
pass
def extract_handle(self):
# 处理后置提取参数
if self.request_data.get("extract", None):
# 仅提取参数中的python表达式，不需要进行数据替换
self.request_data["extract"] = eval_data_process(self.request_data.get("extract", None))
try:
# 处理后置提取参数
if self.request_data.get("extract", None):
# 仅提取参数中的python表达式，不需要进行数据替换
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):
# 处理响应断言参数
if self.request_data.get("assert_response", None):
self.request_data["assert_response"] = eval_data_process(
data_replace(content=self.request_data.get("assert_response", None), source=GLOBAL_VARS))
# 由于数据库断言里面的变量需要请求响应后进行提取，因此目前不进行处理
try:
# 处理响应断言参数
if self.request_data.get("assert_response", None):
self.request_data["assert_response"] = eval_data_process(
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))
logger.info(f"\n======================================================\n" \
"-------------请求数据--------------------\n"
f"用例标题: {self.case_data.get('title', None)}\n" \
f"请求路径: {self.case_data.get('url', None)}\n" \
f"请求方式: {self.case_data.get('method', None)}\n" \
f"请求头: {self.case_data.get('headers', None)}\n" \
f"请求cookies: {self.case_data.get('cookies', None)}\n" \
f"请求关键字: {self.case_data.get('pk', None)}\n" \
f"请求内容: {self.case_data.get('payload', None)}\n" \
f"请求文件: {self.case_data.get('files', None)}\n" \
f"用例标题: {type(self.case_data.get('title', None))} || {self.case_data.get('title', None)}\n" \
f"请求路径: {type(self.case_data.get('url', None))} || {self.case_data.get('url', None)}\n" \
f"请求方式: {type(self.case_data.get('method', None))} || {self.case_data.get('method', None)}\n" \
f"请求头: {type(self.case_data.get('headers', None))} || {self.case_data.get('headers', None)}\n" \
f"请求cookies: {type(self.case_data.get('cookies', None))} || {self.case_data.get('cookies', None)}\n" \
f"请求类型: {type(self.case_data.get('pk', None))} || {self.case_data.get('pk', None)}\n" \
f"请求内容: {type(self.case_data.get('payload', None))} || {self.case_data.get('payload', None)}\n" \
f"请求文件: {type(self.case_data.get('files', None))} || {self.case_data.get('files', None)}\n" \
f"请求响应数据: {response.text}\n" \
f"请求响应码: {response.status_code}\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['method']}")
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['files'])
allure_step(step_title="请求响应数据", content=response.text)
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
@@ -195,16 +227,19 @@ def after_request_extract(response: Response, extract):
"=====================================================")
result = {}
if extract:
if response_type(response) == "json":
# 如果响应数据是json格式，则将按照json方式对后置提取参数进行处理
res = response.json()
for k, v in extract.items():
result[k] = json_extractor(res, v)
else:
# 如果响应数据是str格式，则将按照str方式对后置提取参数进行处理
res = response.text
for k, v in extract.items():
result[k] = re_extract(res, v)
try:
if response_type(response) == "json":
# 如果响应数据是json格式，则将按照json方式对后置提取参数进行处理
res = response.json()
for k, v in extract.items():
result[k] = json_extractor(res, v)
else:
# 如果响应数据是str格式，则将按照str方式对后置提取参数进行处理
res = response.text
for k, v in extract.items():
result[k] = re_extract(res, v)
except Exception as e:
logger.error(f"提取后置参数报错：{e}")
logger.debug(f"\n======================================================\n" \
"-------------End：从响应数据中提取后置参数保存到全局变量--------------------\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