优化请求头及拼接URL方式

This commit is contained in:
wangjie 2023-08-24 17:56:30 +08:00
parent 890fd803ab
commit cda0cc2b45
2 changed files with 14 additions and 8 deletions

View File

@ -23,20 +23,26 @@ from utils.time_utils import TimeUtil
class BaseApi:
"""基础类,对请求方法进行二次封装"""
host = EntryPoint.URL()
default_headers = EntryPoint.DEFAULT_HEADERS()
@staticmethod
def _make_url(address: str) -> str:
"""整理拼接URL"""
return BaseApi.host + address
# 如果address是以http开头的则直接使用该address不与host进行拼接
if address.lower().startswith("http"):
return address
# 确保host不以/结尾
host = EntryPoint.URL().rstrip("/")
# 确保address是以/开头
address = "/" + address.lstrip("/")
return f"{host}{address}"
@staticmethod
def _make_headers(headers) -> dict[Any, Any]:
"""对请求头进行预处理"""
default_headers = EntryPoint.DEFAULT_HEADERS()
headers = headers or {}
headers = {**BaseApi.default_headers, **headers}
return headers
merged_headers = {**default_headers, **headers}
return merged_headers
@staticmethod
def _make_method(method) -> str:

View File

@ -12,7 +12,7 @@ from utils.yaml_handle import YamlHandle
@allure.feature("登录模块")
class TestLogin:
data_smsCode = YamlHandle(DATAS_DIR+os.sep+'smsCode.yaml').read_yaml()
data_smsCode = YamlHandle(DATAS_DIR + os.sep + 'smsCode.yaml').read_yaml()
params = [(item['case_title'], item['expected']) for item in data_smsCode]
@allure.story("测试获取验证码")
@ -27,7 +27,7 @@ class TestLogin:
r = "获取验证码成功"
assert r == message
data_login = YamlHandle(DATAS_DIR+os.sep+'login.yaml').read_yaml()
data_login = YamlHandle(DATAS_DIR + os.sep + 'login.yaml').read_yaml()
params = [(item['case_title'], item['username'], item['password'], item['expected']) for item in data_login]
@allure.story("测试登录")