def assertJsonpath(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
expect_value = str(apidata['request_data']).strip()
datan = str(apidata['input']).strip()
result_to_json = json.dumps(self.jsonres)
res = jsonpath.jsonpath(result_to_json, f'$..{datan}') # 找不到是结果是 False
with allure.step(
f"[{mTime()}]['assertInRe'][key:{datan},actual_value:{res}][expect_value:{expect_value}]"
):
logger.info(f"key:{datan}")
logger.info(f"ACTUAL_VALUE:[{res}]")
logger.info(f"EXPECT_VALUE:[{expect_value}]")
try:
if isinstance(res, list):
assert expect_value in res
else:
assert expect_value == res
except AssertionError as e:
self.return_value('FAIL')
logger.info('--Fail--用例失败--')
logger.exception(e)
# raise
str_result = 'FAIL'
else:
self.return_value('PASS')
logger.info('--Pass--用例成功--')
str_result = 'PASS'
self.__allurestep(str_result)
write_to_excel(sheet, str_result, row_pos, col_pos_c)
write_to_excel(sheet, str(res), row_pos, col_pos_v)
return str_result
def assertInRe(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
expect_value = str(apidata['request_data']).strip()
d_k = str(apidata['input']).strip()
pattern = f'"{d_k}": "(.+?)"'
result_to_json = json.dumps(self.jsonres)
res = re.findall(pattern, result_to_json) # 正则从json中取值
with allure.step(
f"[{mTime()}]['assertInRe'][key:{d_k},actual_value:{res}][expect_value:{expect_value}]"
):
logger.info(f"key:{d_k}")
logger.info(f"ACTUAL_VALUE:[{res}]")
logger.info(f"EXPECT_VALUE:[{expect_value}]")
try:
assert expect_value in res
except AssertionError as e:
self.return_value('FAIL')
logger.info('--Fail--用例失败--')
logger.exception(e)
# raise
str_result = 'FAIL'
else:
self.return_value('PASS')
logger.info('--Pass--用例成功--')
str_result = 'PASS'
self.__allurestep(str_result)
write_to_excel(sheet, str_result, row_pos, col_pos_c)
write_to_excel(sheet, str(res), row_pos, col_pos_v)
return str_result
def savedata(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
request_data_path = apidata['request_data'].strip()
try:
with allure.step(
f"[{mTime()}]['savedata'][relations_key:{apidata['input']}][relations_value:{request_data_path}]"
):
logger.info(f"relations_key:[{apidata['input']}]")
logger.info(f"relations_value:[{request_data_path}]")
logger.info(
f"self.relations[{apidata['input']}] = {request_data_path}"
)
if str(request_data_path).startswith('${'):
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
write_to_excel(sheet, str(request_data_path), row_pos,
col_pos_v)
else:
self.relations[
apidata['input'].strip()] = request_data_path
self.return_value(self.relations[apidata['input'].strip()])
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(
sheet, str(request_data_path), row_pos,
col_pos_v) # self.jsonres[data['request_data']]
return {f'{apidata["input"].strip()}': f'{request_data_path}'}
except Exception as e:
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
logger.error(e)
def assertequals(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
dictabspath = self.__abs(apidata['input'])
expect_value = str(apidata['request_data']).strip()
actual_value = ''
try:
dict_value = eval(str(self.jsonres) + dictabspath)
actual_value = str(dict_value)
except Exception as e:
logger.error(f"{str(self.jsonres) + dictabspath} is not exist.")
logger.error(e)
pass
with allure.step(
f"[{mTime()}]['assertequals'][key:{apidata['input']},actual_value:{actual_value}][expect_value:{expect_value}]"
):
logger.info(f"key:[{apidata['input']}]")
logger.info(f"ACTUAL_VALUE:[{actual_value}]")
logger.info(f"EXPECT_VALUE:[{expect_value}]")
try:
assert actual_value == expect_value
except AssertionError as e:
self.return_value('FAIL')
logger.info('--Fail--用例失败--')
logger.exception(e)
# raise
str_result = 'FAIL'
else:
self.return_value('PASS')
logger.info('--Pass--用例成功--')
str_result = 'PASS'
self.__allurestep(str_result)
write_to_excel(sheet, str_result, row_pos, col_pos_c)
write_to_excel(sheet, str(actual_value), row_pos, col_pos_v)
return str_result
def savejson(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
try:
request_data_path = self.__abs(apidata['request_data'].strip())
request_data_value = eval(str(self.jsonres) + request_data_path)
with allure.step(
f"[{mTime()}]['savejson'][relations_key:{apidata['input']}][jsonres_key:{request_data_path}]"
):
logger.info(f"relations_key:[{apidata['input']}]")
logger.info(f"jsonres_key:[{request_data_path}]")
logger.info(
f"self.relations[{apidata['input']}] = self.jsonres{request_data_path}"
)
self.relations[apidata['input'].strip()] = str(
request_data_value)
self.return_value(self.relations[apidata['input'].strip()])
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(sheet, str(request_data_value), row_pos,
col_pos_v) # self.jsonres[data['request_data']]
return {f'{apidata["input"].strip()}': f'{request_data_value}'}
except Exception as e:
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
return_str = f"key:'{request_data_path}'is not in {str(self.jsonres)}."
write_to_excel(sheet, str(return_str), row_pos, col_pos_v)
logger.error(return_str)
logger.error(e)
pass
def saveparam(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
try:
input_path = apidata['input'].strip()
_data = apidata['request_data'].strip()
if str(_data).startswith('{') and str(_data).endswith('}'):
request_data_value = self.__get_data(_data)
else:
request_data_value = self.__get_relations(_data) # ???
with allure.step(
f"[{mTime()}]['saveparam'][saveparam_key:{input_path}][saveparam_value:{request_data_value}]"
):
logger.info(f"saveparam_key:[{input_path}]")
logger.info(f"saveparam_value:[{request_data_value}]")
logger.info(f"self.param[{input_path}] = {request_data_value}")
self.param[input_path] = request_data_value
self.return_value(self.param)
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(sheet, str(request_data_value), row_pos,
col_pos_v) # self.jsonres[data['request_data']]
return {f'{input_path}': f'{request_data_value}'}
except Exception as e:
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
logger.error(e)
pass
def addheader(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
try:
with allure.step(
f"[{mTime()}]['addheader'][headers_key:{apidata['input']}][{apidata['request_data']}]"
):
logger.info(f"headers_key:[{apidata['input']}]")
logger.info(
f"headers_value_before:[{apidata['request_data']}]")
rel = self.__get_relations(apidata['request_data'].strip())
logger.info(f"headers_value_after(__get_relations):[{rel}]")
self.session.headers[apidata['input'].strip()] = rel
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(sheet, str(rel), row_pos, col_pos_v)
return self.session.headers
except Exception as e:
logger.error(
f"key:'{self.__get_re(apidata['request_data'].strip())}' is not in [{self.relations}]."
)
logger.error(e)
pass
def post(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:【'post'】")
url_path = str(apidata['input']).strip()
_data = apidata['request_data'].strip()
if url_path.startswith('http'):
new_url = url_path
else:
if url_path == '' or url_path is None:
new_url = self.url
else:
if str(self.url)[-1:] == '/':
new_url = self.url + url_path
else:
new_url = self.url + '/' + url_path
try:
with allure.step(
fr"[{mTime()}]['POST'][post_after:{self.result}]"):
new_url = self.__get_relations(new_url)
self.return_value(f'请求接口:[{new_url}]')
self.return_value(f'请求头:[{self.session.headers}]')
_data = self.__get_data(_data)
self.return_value(f'请求体:[{_data}]')
self.result = self.session.post(new_url,
json=_data,
data=None,
proxies=None)
self.jsonres = json.loads(self.result.text)
self.return_value(f'返回值:[{json.loads(self.result.text)}]')
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(sheet, str(self.jsonres), row_pos, col_pos_v)
except Exception as e:
logger.error(
f"Execute method '{sys._getframe().f_code.co_name}' error.")
logger.error(e)
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
write_to_excel(sheet, str(self.result.text), row_pos, col_pos_v)
finally:
self.param = {}
return self.jsonres
def seturl(self, apidata, sheet, row_pos, col_pos_c, col_pos_v):
logger.info(f"执行函数:{sys._getframe().f_code.co_name}")
try:
with allure.step(f"[{mTime()}]['seturl'][{apidata['input']}]"):
logger.info(f"输入参数:[{apidata['input']}]")
path = str(apidata['input']).strip()
if path.startswith('http'):
self.url = path
self.return_value(self.url)
write_to_excel(sheet, 'PASS', row_pos, col_pos_c)
write_to_excel(sheet, str(apidata['input']), row_pos,
col_pos_v)
else:
write_to_excel(sheet, 'FAIL', row_pos, col_pos_c)
write_to_excel(sheet, str(apidata['input']), row_pos,
col_pos_v)
except Exception as e:
logger.error(
f"Execute method '{sys._getframe().f_code.co_name}' error.")
logger.error(e)
pass