test_framework
/
apitest_unittest
mirror of https://gitee.com/chenyongzhiaaron_admin/apitest.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

增加统一自定义异常类

This commit is contained in:
chenyongzhiaaron 2023-08-01 09:40:40 +08:00
parent 94d1694b06
commit 0596a58892
5 changed files with 275 additions and 202 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/http_client/http_client.py
View File

@ -22,6 +22,7 @@ class Pyt(LoadModulesFromFolder):
super().__init__()
self.request = None
self.response = None
self.response_json = None
@request_retry_on_exception()
def http_client(self, host, url, method, **kwargs):
@ -71,6 +72,10 @@ class Pyt(LoadModulesFromFolder):
self.request = requests.Request(method, __url, **kwargs)
self.response = self.session.send(self.request.prepare(), timeout=30, verify=True)
[i.close() for i in fs if len(fs) > 0]
try:
self.response_json = self.response.json()
except Exception as e:
self.response_json = None
return self.response

34
common/utils/action.py
View File

@ -14,6 +14,7 @@ from common.crypto.encrypt_data import EncryptData
from common.database.mysql_client import MysqlClient
from common.log_utils.mylogger import MyLogger
from common.utils.decorators import singleton
from common.utils.exceptions import *
from common.validation.extractor import Extractor
from common.validation.load_and_execute_script import LoadScript
from common.validation.validator import Validator
@ -38,15 +39,11 @@ class Action(Extractor, LoadScript, Validator, MysqlClient):
compiled = compile(ast_obj, '<string>', 'exec')
exec(compiled, {"pm": self})
except SyntaxError as e:
error_message = f'动态代码语法异常: {e}'
self._handle_error(error_message)
except Exception as e:
error_message = f"动态代码执行异常: {e}"
self._handle_error(error_message)
return self.variables
ExecuteDynamiCodeError(code, e)
except ExecuteDynamiCodeError as e:
ExecuteDynamiCodeError(code, e)
def _handle_error(self, error_message):
print(f'发现异常: {error_message}')
return self.variables
@property
def variables(self, key=None):
@ -68,10 +65,9 @@ class Action(Extractor, LoadScript, Validator, MysqlClient):
return headers, request_data
def send_request(self, host, url, method, teardown_script, **kwargs):
response = self.http_client(host, url, method, **kwargs)
self.update_environments("responseTime", response.elapsed.total_seconds() * 1000) # 存响应时间
self.execute_dynamic_code(response, teardown_script)
return response
self.http_client(host, url, method, **kwargs)
self.update_environments("responseTime", self.response.elapsed.total_seconds() * 1000) # 存响应时间
self.execute_dynamic_code(self.response, teardown_script)
@staticmethod
def base_info(item):
@ -142,9 +138,8 @@ class Action(Extractor, LoadScript, Validator, MysqlClient):
if sleep_time:
try:
time.sleep(sleep_time)
except Exception as e:
self.logger.error("暂时时间必须是数字")
raise e
except MyBaseException as e:
raise MyBaseException(f"暂停时间必须是数字!")
def exc_sql(self, item):
sql, sql_params_dict = self.sql_info(item)
@ -152,11 +147,14 @@ class Action(Extractor, LoadScript, Validator, MysqlClient):
if sql:
try:
execute_sql_results = self.execute_sql(sql)
except DatabaseExceptionError as e:
raise DatabaseExceptionError(sql, str(e))
else:
if execute_sql_results and sql_params_dict:
try:
self.substitute_data(execute_sql_results, jp_dict=sql_params_dict)
except Exception as e:
self.logger.error(f'执行 sql 失败:{sql},异常信息:{e}')
raise e
except ParameterExtractionError as e:
ParameterExtractionError(sql_params_dict, str(e))
if __name__ == '__main__':

2
common/utils/decorators.py
View File

@ -37,7 +37,7 @@ def request_retry_on_exception(retries=2, delay=1.5):
nonlocal e
for i in range(retries):
try:
print(f"第{i}次发送请求的参数： {kwargs}")
print(f"第{i + 1}次发送请求的参数： {kwargs}")
response = func(*args, **kwargs)
print(f"请求地址 --> {response.request.url}")
print(f"请求头 --> {response.request.headers}")

71
common/utils/exceptions.py Normal file
View File

@ -0,0 +1,71 @@
#!/usr/bin/env python
# encoding: utf-8
"""
@author: kira
@contact: 262667641@qq.com
@file: exceptions.py
@time: 2023/8/1 9:12
@desc:
"""
class MyBaseException(Exception):
def __init__(self, msg):
self.msg = msg
def __str__(self):
return self.msg
class RequestSendingError(MyBaseException):
"""请求异常"""
ERROR_CODE = 1001
def __init__(self, url, reason):
msg = f"请求异常：URL={url}, 原因={reason}"
super().__init__(msg)
class DatabaseExceptionError(MyBaseException):
"""数据库异常"""
ERROR_CODE = 1002
def __init__(self, operation, reason):
msg = f"数据库异常：操作={operation}, 原因={reason}"
super().__init__(msg)
class ParameterExtractionError(MyBaseException):
"""参数提取异常"""
ERROR_CODE = 1003
def __init__(self, parameter_path, reason):
msg = f"参数提取异常：参数路径={parameter_path}, 原因={reason}"
super().__init__(msg)
class ParameterReplacementError(MyBaseException):
"""参数替换异常"""
ERROR_CODE = 1004
def __init__(self, parameter_name, reason):
msg = f"参数替换异常：参数名称={parameter_name}, 原因={reason}"
super().__init__(msg)
class AssertionFailedError(MyBaseException):
"""断言异常"""
ERROR_CODE = 1005
def __init__(self, assertion_name, actual_value, expected_value):
msg = f"断言失败：断言名称={assertion_name}, 实际值={actual_value}, 期望值={expected_value}"
super().__init__(msg)
class ExecuteDynamiCodeError(MyBaseException):
"""执行动态代码异常"""
ERROR_CODE = 1006
def __init__(self, code, reason):
msg = f"执行动态代码异常：动态代码={code}, 原因={reason}"
super().__init__(msg)

11
test_script/test_executor.py
View File

@ -12,9 +12,9 @@ import unittest
from ddt import ddt, data
from common import bif_functions
from common.file_handling.do_excel import DoExcel
from common.utils.action import Action
from config import Config
from common.file_handling.do_excel import DoExcel
from extensions import dynamic_scaling_methods
test_file = Config.test_case # 获取 excel 文件路径
@ -61,16 +61,15 @@ class TestProjectApi(unittest.TestCase):
# 分析请求参数信息
headers, request_data = self.action.analysis_request(request_data, h_crypto, headers, r_crypto, ex_request_data)
result_tp = None
result = "PASS"
# 执行请求操作
kwargs = {request_data_type: request_data, 'headers': headers, "params": query_str}
response = self.action.send_request(host, url, method, teardown_script, **kwargs)
self.action.send_request(host, url, method, teardown_script, **kwargs)
try:
# 提取响应
self.action.substitute_data(response.json(), regex=regex, keys=keys, deps=deps, jp_dict=jp_dict)
self.action.substitute_data(self.action.response_json, regex=regex, keys=keys, deps=deps, jp_dict=jp_dict)
except Exception as err:
self.action.logger.error(f"提取响应失败：{sheet}_{iid}_{name}_{desc}"
f"\nregex={regex};"
@ -84,14 +83,14 @@ class TestProjectApi(unittest.TestCase):
try:
# print(f"期望结果--> {expected}")
# 执行断言 返回结果元组
self.action.run_validate(expected, response.json())
self.action.run_validate(expected, self.action.response_json)
except Exception as e:
result = "FAIL"
self.action.logger.error(f'异常用例: **{sheet}_{iid}_{name}_{desc}**\n{e}')
raise e
finally:
print(f"断言结果-->", self.action.assertions)
response = response.text if response is not None else str(response)
response = self.action.response.text if self.action.response is not None else str(self.action.response)
# 响应结果及测试结果回写 excel
excel.write_back(sheet_name=sheet, i=iid, response=response, test_result=result,
assert_log=str(self.action.assertions))