def rh_html_check_result(self, input, response_json):
data = str(input).strip().replace("\n", "").replace("”", "\"").replace("‘", "\'").replace("’", "\'")\
# .replace("；", ";")
if data is None:
raise Exception("error format,prepareResult is none.")
flag = False
# eval()对特殊值处理
# null = ""
# true = True
# false = False
# 返回结果为list[]或者 json格式 比较
if response_json.startswith("[") or response_json.startswith("{"):
response_json = Core.eval_str(response_json) # str转成list
cmp_result = Core.rh_check_result(input, response_json)
if "[success]" in cmp_result:
flag = True
else:
return cmp_result
# 返回结果为text
else:
except_message = data.split(":=")[1]
if response_json.__contains__(except_message):
flag = True
else:
return "[fail]:response data do not contains %s." % data
if flag:
return "[success]:response data contains %s." % data
def change_para_html(self, html_cont, data):
if data == "":
return html_cont
replace_data = self.html_replace_data(data)
for data in replace_data:
if "=" not in data:
raise Exception("error format,not found =")
try:
if ":=" not in data:
raise Exception("error format")
sval = get_var(data.split(":=")[1])
name = data.split(":=")[0]
# 对参数做num，str区分 数字：222222， str：'222222'
if "," in sval:
sval = sval.split(",")
if sval != "" and isinstance(sval, str):
if Core.is_numeric(sval):
sval = int(sval)
elif "\'" in sval:
sval = sval.replace("\'", "")
elif sval.startswith("[") or sval.startswith("{"):
sval = Core.eval_str(sval)
# 对入参类型为bool, null的处理
if sval == 'false':
sval = False
elif sval == 'true':
sval = True
elif sval == 'null' or sval == "":
sval = None
scmd = '''html_cont[%s]="%s" ''' % (name, sval)
html_cont[name] = sval
except Exception, e:
raise Exception("error after %s, %s" % (scmd, str(e)))
def change_para_json(self, json_cont, data):
if data == "":
return json_cont
# try:
# t_js = json.loads(data)
# for ixe in t_js.keys():
# t_ixv = t_js.get(ixe)
# if isinstance(t_ixv, (basestring, dict, list)):
# t_ixv = get_var(t_ixv)
# json_cont[ixe] = t_ixv
# return json_cont
# except:
replace_data = self.json_replace_data(data)
for data in replace_data:
if "=" not in data:
raise Exception("error format,not found =")
try:
if ":=" not in data:
raise Exception("error format")
sval = get_var(data.split(":=")[1])
name = data.split(":=")[0]
# sval = sval.replace("'", "\"")
# 对参数做num，str区分 数字：222222， str：'222222'
if sval != "":
if sval.startswith("'") and sval.endswith("'"):
sval = str(sval[1:-1])
elif Core.is_numeric(sval):
if "." in sval:
sval = float(sval)
else:
sval = int(sval)
# 支持替换成空list或dict： a=[]; a={}
elif sval.startswith("[") or sval.startswith("{"):
# elif sval.replace(" ", "") == "[]" or sval.replace(" ", "") == "{}":
sval = Core.eval_str(sval)
# 嵌套json替换
obj = ""
for i in name.split("."):
if Core.is_numeric(i):
obj = "%s[%s]" % (obj, i)
else:
obj = "%s['%s']" % (obj, i)
# 对值为为bool, null的处理
if sval == 'false':
sval = False
elif sval == 'true':
sval = True
elif sval == 'null' or sval == "": # 若需要传 {"A":null} 则可以用默认参数或者 A:=null 或者 A:=
sval = None
elif sval == "\"\"": # 若需要传 {"A":""} 则替换值时需这样处理 A:=\"\"
sval = "" # 字符串处理
scmd = "%s%s = sval" % ("json_cont", obj)
exec scmd
except Exception, e:
raise Exception("error occur %s" % scmd)
def get_nestdict_value(self, keytag, dict_data):
if type(dict_data) not in [types.ListType,
types.DictType]: # 处理返回值为 "{}" ,"[]"情况
# dict_data = json.loads(dict_data)
dict_data = Core.eval_str(dict_data) # 效果同上
# 处理 "a": "[]" 情况
dict_data = self.dict_handle(dict_data)
sname = keytag.strip()
obj = scmd = realval = ""
for i in sname.split("."):
if Core.is_numeric(i):
obj = "%s[%s]" % (obj, i)
else:
if 'str(' in i:
i = i[i.find('(') + 1:-1]
obj = "%s['%s']" % (obj, i)
scmd = "%s%s" % ("dict_data", obj)
try:
realval = eval(scmd)
except:
return "[Failed]:cmd change error,eval(%s)" % scmd
return realval
def dict_handle(self, dict_data):
"""
该方法处理json嵌套中，嵌套的json和list是str类型。则需要eval处理。 eg: {"a": "[]"} --> {"a": []}
:param dict_data:
:return:
"""
for k, v in dict_data.items():
if type(v) in [
types.StringType, types.UnicodeType
] and (str(v).startswith("[") or str(v).startswith("{")):
v1 = Core.eval_str(v)
dict_data[k] = v1
else:
pass
return dict_data
def change_in_json_dic(self, sdic, new_dic, json_cont):
if new_dic != {}:
if sdic.get("input", "") != "":
for (sname, sval) in sdic.get("input", "").items():
try:
name = sname
value = new_dic[sval]
# 嵌套json替换
obj = ""
for i in name.split("."):
if Core.is_numeric(i):
obj = "%s[%s]" % (obj, i)
else:
obj = "%s['%s']" % (obj, i)
scmd = "%s%s = value" % ("json_cont", obj)
exec scmd
# json_cont[name] = value
except Exception, e:
raise Exception("error after change, %s" % scmd)
def jsonparse_pub(self, model="ride"):
j = 0
headers = {"Content-Type": 'application/json;charset=UTF-8'}
token = ""
if model == "ride":
path = os.getcwd()
else:
path = os.path.abspath('..')
Settings.CASE_GEN_DIC.clear()
for i in xrange(self.api_len): # (0,2)
interface_name = self.apis[i]
api = Core.rh_get_api_name(self.apis[i])
# 再次循环对比接口和数据ini文件，取到接口数据所在的ini文件，支持多个ini文件存储数据
for ini_name in self.ini_name_list:
cur_path = os.path.join(path, "config", "api", self.api_name)
ini_path = os.path.join(cur_path, ini_name)
ini_data = RHIniParser(ini_path)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data = self.get_ini_default_contant(ini_data)
if self.apis[i] in default_data[0].keys() or self.apis[i] in default_data[4].keys():
break
# 读取接口产品 [msp.01; msp; msp.03 ]
product_name = self.get_productname(i)
# 读取接口路径
model_url = self.get_modelurl(ini_data, api, product_name)
# 读取对应api接口的默认值
ini_contant = self.get_ini_default_contant(ini_data, api, product_name)
if ".db" not in api and ".dmldb" not in api:
# 对应api接口的默认值，请求方式，出入参设置，正则表达式，特定返回值
default_contant = ini_contant[0]
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
if "||" in self.data_source[i * 2 + 1 - j]: # 对rf的表格的取值，替换列的，注意替换项必须和url里面的一致
self.urlparam = self.data_source[i * 2 + 1 - j].split("||")[1]
# 替换urlinput设置的变量--url
interface_name = self.change_in_json_urldic(in_para_dic, self.dic_args, interface_name, self.urlparam)
# 替换in out设置的变量--data
self.change_in_json_dic(in_para_dic, self.dic_args, default_contant)
# 替换设置的参数值
self.change_para_json(default_contant,
self.data_source[i * 2 + 1 - j].split("||")[0]) # data_source[1] data_source[3]...
# 走html处理的，或者是json 的get请求，不需要dumps
if request_method == "get":
# #####==get请求参数要传dict，且dict中嵌套的list或者dict需要转换为str才能识别。=========#########
for k in default_contant:
if isinstance(default_contant[k], dict):
default_contant[k] = json.dumps(default_contant[k])
elif isinstance(default_contant[k], list):
default_contant[k] = str(default_contant[k])
else:
# post请求参数为json，做dumps处理
default_contant = json.dumps(default_contant)
# 处理需要登录中的Authorization
token = self.dic_args.get("Authorization", "")
# ============请求hearder组装START 如有需要，自行处理====================
headers = dict(headers, **{"authorization": token})
# iaas_cloud : token验证
# if 'cloud_manage' in self.api_name or "auto_engine" in self.api_name:
# headers = {"Content-Type": 'application/json', 'X-Auth-Token': token.encode('utf-8')}
# ============请求hearder组装END========================================
print "请求接口: " + self.version_url + model_url + interface_name
print default_contant
response_json = self.json_request(self.version_url + model_url + interface_name, default_contant, request_method, headers)
print response_json[0]
data = Core.rh_replace_data(self.data_source[i * 2 + 2 - j])
for i in data:
res = self.rh_json_check_result(i, response_json[0])
if "[success]" not in res:
logger.error("%s failed:\n***********post data:\n%s\n***********response data:%s" % (
api, default_contant, response_json[0]))
self.Verify.should_contain(res, "[success]")
# 获取正则表达式列表
default_re = default_re_set.split(";") # 存储特定返回值
out = self.get_out_json_dic(out_para_dic, response_json[0], default_re)
self.dic_args = self.get_new_dic(self.dic_args, out)
print "%s:%s" % (api, self.dic_args)
def htmlparse_pub(self, model="ride"):
j = 0
ini_data = ""
# ============请求头等初始化信息（需要的则自行添加）===========
headers = {}
token = ""
cookies = ""
upload_files = ""
handler_version_url = ""
# 此处为表单上传附件接口处理，若无，则为空 []
list_upload = ["uploadImage"]
# =====================END=============================================================================
# =====上一层目录-实现类调用==============
if model == "ride":
path = os.getcwd()
else:
path = os.path.abspath('..')
# ========URL固定不变的，每个版本替换port处理============================
self.version_url = self.change_url_port(self.version_url,
self.version_port)
# 执行案例前清空CASE_GEN_DIC
'''PS：经验证不能使用Settings.CASE_GEN_DIC = {}，否则后续其他用例调用值一直保持第一次的，
原因分析?: Settings.CASE_GEN_DIC = {} 对CASE_GEN_DIC的引用改变了，CASE_GEN_DIC的值未改变;clear则清空该引用中的值
'''
Settings.CASE_GEN_DIC.clear()
for i in xrange(self.api_len): # (0,2)
interface_name = self.apis[i]
api = Core.rh_get_api_name(self.apis[i])
# 再次循环对比接口和数据ini文件，取到接口数据所在的ini文件，支持多个ini文件存储数据
for ini_name in self.ini_name_list:
cur_path = r'%s%sconfig%sapi%s%s' % (path, fileSep, fileSep,
fileSep, self.api_name)
ini_path = r"%s%s%s" % (cur_path, fileSep, ini_name)
ini_data = RHIniParser(ini_path)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data = self.get_ini_default_contant(ini_data)
if self.apis[i] in default_data[0].keys(
) or self.apis[i] in default_data[4].keys(
) or self.apis[i] in default_data[5].keys():
break
# 读取接口产品 [msp.01; msp; msp.03 ]
product_name = self.get_productname(i)
model_url = self.get_modelurl(ini_data, api)
# model_url = self.get_modelurl(ini_data, api, product_name)
# 读取对应api接口的默认值
ini_contant = self.get_ini_default_contant(ini_data, api,
product_name)
if ".db" not in api and ".dmldb" not in api:
# 读取对应api接口的默认值，请求方式，出入参设置，正则表达式，特定返回值
default_contant = ini_contant[0]
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
# 原始数据转成dict, 若原本为dict，则直接返回
default_contant = self.transfor_to_dict(default_contant)
if "||" in self.data_source[
i * 2 + 1 - j]: # 对rf的表格的取值，替换列的，注意替换项必须和url里面的一致
self.urlparam = self.data_source[i * 2 + 1 -
j].split("||")[1]
# 替换urlinput设置的变量--url
interface_name = self.change_in_html_urldic(
in_para_dic, self.dic_args, interface_name, self.urlparam)
# ============ 一些需要自己特殊处理的接口方法可以执行书写 ================================================
# sq特殊处理
if interface_name in ["getVerifyCode.do", "Login.sso"]:
handler_version_url = self.version_url.replace(
r"/" + self.version_uri, "")
self.version_url = handler_version_url
# 同一流程里需要还原
else:
self.version_url = handler_version_url + r"/" + self.version_uri
# sq特殊处理 查询数据库存储对应的值到dic_args， 通过input引用
# if interface_name in ["staffBindWechat", "salesmanBindWechat", "agentBind", "staffBind"]:
# sleep(5) # 确保数据已经入库
# self.dic_args = self.query_db_into_dicargs(EnvSetting.ENV_SQL_DIC.get(interface_name, ""), "tablet", "checkCode", self.dic_args, str(default_contant["phoneNo"]).split())
# # sq特殊处理 短信发送间隔必须大于30s
# if interface_name in ["commonPayApply.do"]:
# sleep(30)
# ============ 一些需要自己特殊处理的接口方法可以执行书写-END ============================================
# 替换in out设置的变量--data
self.change_in_html_dic(in_para_dic, self.dic_args,
default_contant)
# 替换设置的参数值
self.change_para_html(
default_contant, self.data_source[i * 2 + 1 - j].split(
"||")[0]) # data_source[1] data_source[3]...
# ============请求hearder组装 如有需要，自行处理====================
token = self.dic_args.get("token", "")
# header需要放token的处理
if ("newWechat" in model_url) and "updatePwd" not in api:
headers = dict(headers,
**{"authorization": "Bearer " + token})
'''上传照片接口单独处理获取照片'''
if api in list_upload:
if "Windows" in platform.system():
local_files = json.loads(
ini_data.get_ini_info(
"UPLOAD", "UPLOAD_DEFAULT_FILES"))[api].get(
"windows", {})
else: # 不是Windows 就默认是Linux
local_files = json.loads(
ini_data.get_ini_info(
"UPLOAD",
"UPLOAD_DEFAULT_FILES"))[api].get("linux", {})
upload_files = self.upload_files_format(local_files, path)
print "请求接口: " + self.version_url + model_url + interface_name
# print default_contant
# 发送请求
response_html = self.html_request(
self.version_url + model_url + interface_name,
default_contant, cookies, upload_files, request_method,
headers)
print response_html
data = Core.rh_replace_data(self.data_source[i * 2 + 2 - j])
for i in data:
res = self.rh_html_check_result(i, response_html)
if "[success]" not in res:
logger.error(
"%s failed:\n***********post data:\n%s\n***********response data:%s"
% (api, default_contant, response_html))
self.Verify.should_contain(res, "[success]")
# 获取正则表达式列表
default_re = default_re_set.split(";")
# 存储特定返回值
out = self.get_out_html_dic(out_para_dic, response_html,
default_re)
self.dic_args = self.get_new_dic(self.dic_args, out)
print "%s:%s" % (api, self.dic_args)
def html_model_pub(data_source, version="1221", model="ride"):
dic_args = {}
result = Result()
Verify = _Verify()
upload_files = {}
ini_path = ""
urlparam = ""
default_data_dict = {}
# ============请求头信息（需要的则自行添加）===========
# 手动添加一下header防反爬 --ChunkedEncodingError: ("Connection broken: error(10054, '')", error(10054, ''))报错
headers = {
"Accept":
"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
"Accept-Encoding":
"gzip, deflate, br",
"Accept-Language":
"zh-CN,zh;q=0.9",
"Connection":
"keep-alive",
"User-Agent":
"Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"
}
headers_wechat = {
"Accept":
"application/json, text/plain, */*",
"Accept-Encoding":
"gzip, deflate, br",
"Accept-Language":
"zh-CN,zh;q=0.9,en;q=0.8",
"Connection":
"keep-alive",
"User-Agent":
"Mozilla/5.0 (iPhone; CPU iPhone OS 10_3 like Mac OS X) AppleWebKit/602.1.50 (KHTML, like Gecko) CriOS/56.0.2924.75 Mobile/14E5239e Safari/602."
}
authorization = ""
# ===================================================
j = 0
# return
if model == "ride":
path = os.path.abspath('.')
else:
path = os.path.abspath('../..')
# =============路径，产品，执行接口获取;eg：msp.apply.01;msp.apply:interface_a;interface_b=============
data_source_apis = data_source[0].split(":") # 用 ：对datasource的第一个元素进行拆分
apis = html_replace_data(data_source_apis[1]) # 入参是接口名 取出接口名
interface_path_product = html_replace_data(
data_source_apis[0]) # 入参是 msp.apply.01
api_name = interface_path_product[0].split(".")[0] # 取出了 msp
api_len = len(apis)
# ================获取接口属性=====================
api_def_dic = EnvSetting.ENV_DEF_DIC
version_dict = EnvSetting.VERSION
version_info = version_dict.get(version)
version_port = version_info["port"]
version_url = version_info.get("url", "")
env_pro_dic = api_def_dic.get(api_name)
main_url = env_pro_dic["url"]
# ========获取url截取点==========
main_url = change_url_port(main_url, version_port, version_url)
login_url = get_login_url(main_url)
# ==============================
ini_name_list = env_pro_dic["api_ini_name"]
# ====循环对比接口和数据ini文件，取到第一个接口数据所在的ini文件，主要是为了先获取登录参数，看第一个接口是否需要登录
for ini_name in ini_name_list:
cur_path = r'%s\config\api\%s' % (path, api_name)
ini_path = r"%s\%s" % (cur_path, ini_name)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data_dict = get_ini_default_contant(ini_path, None, None)[0]
if apis[0] in default_data_dict.keys():
break
# 执行案例前清空CASE_GEN_DIC
'''PS：经验证不能使用Settings.CASE_GEN_DIC = {}，否则后续其他用例调用值一直保持第一次的，
原因分析?: Settings.CASE_GEN_DIC = {} 对CASE_GEN_DIC的引用改变了，CASE_GEN_DIC的值未改变;clear则清空该引用中的值
'''
Settings.CASE_GEN_DIC.clear()
'''获取cookies (调试接口用)=====有登陆接口传{}即可=============================================='''
# default_cookies = RHIniParser(ini_path).get_ini_info("COOKIES", "SET_COOKIES")
# cookies = str_to_dict(default_cookies, ";", "=")
cookies = {}
'''==========================================================================================='''
'''========================判断系统是否需要登录=================================================='''
login_sign = RHIniParser(ini_path).get_ini_info("IS_LOGIN", "LOGIN_SIGN")
if login_sign is "Y":
# 获取登录请求地址
# 利用有序OrderedDict 有序读取url（有些登录需要验证码，需现请求获取验证码接口，防止验证码过期）
login_url_dict = json.loads(RHIniParser(ini_path).get_ini_info(
"IS_LOGIN", "LOGIN_URL"),
object_pairs_hook=OrderedDict)
for url in login_url_dict.keys():
ini_contant = get_ini_default_contant(ini_path, url, None)
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
post_data = transfor_to_dict(login_url_dict[url])
# 若为登录请求接口，替换对应参数兼容各个版本
change_login_dic(api_name, post_data, version_port)
# 替换urlinput设置的变量--url
url = change_in_html_urldic(in_para_dic, dic_args, url, urlparam)
# 替换in out设置的变量--data
change_in_html_dic(in_para_dic, dic_args, post_data)
response_html = html_request(login_url + url, post_data,
request_method)
# 获取正则表达式列表
default_re = default_re_set.split(";") # 存储特定返回值
out = get_out_html_dic(out_para_dic, response_html, default_re)
dic_args = get_new_dic(dic_args, out)
'''==========================================================================================='''
for i in xrange(api_len): # (0,2)
interface_name = apis[i]
api = Core.rh_get_api_name(apis[i])
# 再次循环对比接口和数据ini文件，取到接口数据所在的ini文件，支持多个ini文件存储数据
for ini_name in ini_name_list:
cur_path = r'%s\config\api\%s' % (path, api_name)
ini_path = r"%s\%s" % (cur_path, ini_name)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data_dict = get_ini_default_contant(ini_path, None,
None)[0]
default_dbdata_dict = get_ini_default_contant(
ini_path, None, None)[4]
if apis[i] in default_data_dict.keys(
) or apis[i] in default_dbdata_dict.keys():
break
# 读取接口路径, 产品 [msp.apply.01; msp.apply; msp.tools.03 ]
if len(interface_path_product) == 1:
path_list = interface_path_product[0].split(".")
model = path_list[1]
if len(path_list) == 3:
product_name = path_list[2]
else:
product_name = None
else:
path_list = interface_path_product[i].split(".")
model = path_list[1]
if len(path_list) == 3:
product_name = path_list[2]
else:
product_name = None
model_url = json.loads(
RHIniParser(ini_path).get_ini_info("MODEL_URL",
"MODEL_URL_DATA"))[model]
ini_contant = get_ini_default_contant(ini_path, api, product_name)
# 若为移动端接口，则一般为默认443或则80端口，需要把端口号去掉 --如有需要处理的，自行添加。
if "newWechat" in model or "bplan" in model:
main_url = main_url.replace(version_port, "")
if ".db" not in api and ".dmldb" not in api:
# 读取对应api接口的默认值，请求方式，出入参设置，正则表达式，特定返回值
default_contant = ini_contant[0]
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
# 原始数据转成dict
default_contant = transfor_to_dict(default_contant)
if "||" in data_source[i * 2 + 1 -
j]: # 对rf的表格的取值，替换列的，注意替换项必须和url里面的一致
urlparam = data_source[i * 2 + 1 - j].split("||")[1]
# 替换urlinput设置的变量--url
interface_name = change_in_html_urldic(in_para_dic, dic_args,
interface_name, urlparam)
#szx特殊处理
if interface_name == 'savePolicyInfo.do' and product_name[
-2:] == '01':
wtr_num = 0
scanmock_result = query_scanmock_result(
str(default_contant["applyBarCode"])[:-2])
while (not scanmock_result.flag) and wtr_num <= 5:
scanmock_result = query_scanmock_result(
str(default_contant["applyBarCode"])[:-2])
# print 'waiting %s' % Settings.EXTREME_TIMEOUT
sleep(Settings.EXTREME_TIMEOUT)
wtr_num += 1
if api == 'addApplyInfo/passSaveApplyInfo.do':
sleep(20)
# 替换设置的参数值
change_para_html(default_contant, data_source[i * 2 + 1 - j].split(
"||")[0]) # data_source[1] data_source[3]...
#szx特殊处理
if interface_name == 'savePolicyInfo.do' and product_name[
-2:] == '02':
dic_args = uw_query_ins_or_prem(
str(default_contant["applyBarCode"]), dic_args)
# 替换in out设置的变量--data
change_in_html_dic(in_para_dic, dic_args, default_contant)
# 更新token， 放入请求hearder中
authorization = change_in_token_dic(in_para_dic, dic_args,
authorization)
# ============请求hearder组装 如有需要，自行处理====================
# 微信快易投需要用token验证
if "newWechat" in model:
headers = dict(headers_wechat,
**{"authorization": "Bearer " + authorization})
'''上传照片接口单独处理获取照片'''
if api.__contains__("imageUploding"):
local_files = json.loads(
RHIniParser(ini_path).get_ini_info(
"UPLOAD", "UPLOAD_DEFAULT_FILES"))[api]
upload_files = upload_files_format(local_files, path)
print "请求接口: " + main_url + model_url + interface_name
# print default_contant
# 发送请求
response_html = html_request(main_url + model_url + interface_name,
default_contant, cookies,
upload_files, request_method, headers)
data = Core.rh_replace_data(data_source[i * 2 + 2 - j])
for i in data:
res = rh_html_check_result(i, response_html)
if "[success]" not in res:
logger.error(
"%s failed:\n***********post data:\n%s\n***********response data:%s"
% (api, default_contant, response_html))
Verify.should_contain(res, "[success]")
# 获取正则表达式列表
default_re = default_re_set.split(";")
# 存储特定返回值
out = get_out_html_dic(out_para_dic, response_html, default_re)
dic_args = get_new_dic(dic_args, out)
print "%s:%s" % (api, dic_args)
# ===================================================================================
# ============若无DB检查点，可以暂时忽略该部分代码，若需要添加，自行修改====================
elif ".db" in api:
j += 2
# 对于DB校验sql取数，产品取db前一个接口对应的产品。
# 获取检验的sql
sql_list = ini_contant
# 避免程序入库事务还未完成就执行DB校验，故等待1秒
sleep(1)
# 注意 此处传过去的sql为list（多条） 格式：saveApproval.slis.db
if 'msp' in api_name:
result = check_db(sql_list, version,
api.split('.')[1],
dic_args["applyBarCode"].split())
else:
result = check_db(sql_list, version, api.split('.')[1])
if not result.flag:
logger.info(result.msg, "%s failed" % api)
print result.msg, "%s failed" % api
Verify.should_be_true(result.flag)
print "%s success" % api
logger.info("%s success" % api)
# ==========若有需要，执行回滚脚本======================================#
else:
j += 2
# 对于DB回滚sql取数，产品取db前一个接口对应的产品。
# 获取需要回滚的sql
rollback_sqllist = ini_contant
# 避免程序入库事务还未完成就执行DB校验，故等待1秒
sleep(1)
# 执行回滚操作
try:
dml_db(rollback_sqllist, version, api.split('.')[1])
print "%s success" % api
logger.info("%s success" % api)
except Exception, e:
raise Exception("error occur in sql rollback, %s", str(e))
def xml_model_pub(data_source, model="ride"):
dic_args = {}
result = Result()
Verify = _Verify()
j = 0
if model == "ride":
path = os.path.abspath('.')
else:
path = os.path.abspath('../..')
data_source_apis = data_source[0].split(":")
apis = xml_replace_data(data_source_apis[1])
product_path = xml_replace_data(data_source_apis[0])
project_name = product_path[0].split(".")[0]
api_len = len(apis)
ini_path = "%s/config/api/bank/%s/para_configuration.ini" % (path,
project_name)
ybt_def_dic = EnvSetting.ENV_DEF_DIC
ybt_pro_dic = ybt_def_dic.get(project_name)
Settings.CASE_GEN_DIC = {}
for i in xrange(api_len):
api = Core.rh_get_api_name(apis[i])
if ".xml" in api:
if len(product_path) == 1:
project_all_path = "/".join(product_path[0].split("."))
model_path = '%s/config/api/bank/%s/' % (path,
project_all_path)
else:
project_all_path = "/".join(product_path[i - j / 2].split("."))
model_path = '%s/config/api/bank/%s/' % (path,
project_all_path)
xml = model_path + api
in_para_dic = Core.rh_load_default_api_dic(api, ini_path, "PARA",
"DICT_IN_OUT")
soup = get_soup(xml)
change_in_soup_dic(in_para_dic, dic_args, soup)
change_para_soup(soup, data_source[i * 2 + 1 - j])
response_soup = soup_post(ybt_pro_dic["url"], soup)
data = Core.rh_replace_data(data_source[i * 2 + 2 - j])
for i in data:
res = rh_soup_check_result(i, response_soup)
if "[success]" not in res:
logger.error(
"%s failed:\n***********post data:\n%s\n***********response data:%s"
% (api, soup, response_soup))
Verify.should_contain(res, "[success]")
out_para_dic = Core.rh_load_default_api_dic(
api, ini_path, "PARA", "DICT_IN_OUT")
out = get_out_soup_dic(out_para_dic, response_soup)
dic_args = get_new_dic(dic_args, out)
print "%s:%s" % (api, dic_args)
else:
j += 2
# if project_name=="ccb":
# p_num = dic_args["InsPolcy_No"]
# else:
p_num = dic_args["PolNumber"]
if api == "bia_to_core.db":
result = bia_to_core(p_num, ybt_pro_dic["db_bia_usr"],
ybt_pro_dic["db_bia_pwd"],
ybt_pro_dic["db_ip"],
ybt_pro_dic["db_name"])
elif api == "fin_check.db":
result = fin_check(p_num, ybt_pro_dic["db_fin_usr"],
ybt_pro_dic["db_fin_pwd"],
ybt_pro_dic["db_ip"],
ybt_pro_dic["db_name"])
elif api == "biab_to_core.db":
result = biab_to_core(p_num, ybt_pro_dic["db_bia_usr"],
ybt_pro_dic["db_bia_pwd"],
ybt_pro_dic["db_ip"],
ybt_pro_dic["db_name"])
elif api == "upload_file":
result = upload_file(ybt_pro_dic,
dic_args,
path,
name="pub.jpg")
if not result.flag:
print result.msg
"%s failed" % api
Verify.should_be_true(result.flag)
logger.info("%s success" % api)
print "%s success" % api
def json_model_pub(data_source, version="1221", model="ride"):
dic_args = {}
result = Result()
Verify = _Verify()
upload_files = {}
ini_path = ""
token = ""
urlparam = ""
default_data_dict = {}
# ============请求头信息（需要的则自行添加）===========
headers = {}
# ===================================================
j = 0
if model == "ride":
path = os.path.abspath('.')
else:
path = os.path.abspath('../..')
# =========路径，产品，执行接口获取； eg：msp.apply.01;msp.apply:interface_a;interface_b ==========
data_source_apis = data_source[0].split(":") # 用 ：对datasource的第一个元素进行拆分
apis = json_replace_data(data_source_apis[1]) # 入参是接口名 取出接口名
interface_path_product = json_replace_data(
data_source_apis[0]) # 入参是 msp.apply.01
api_name = interface_path_product[0].split(".")[0] # 取出了 msp
api_len = len(apis)
# ================获取接口属性=====================
api_def_dic = EnvSetting.ENV_DEF_DIC
version_dict = EnvSetting.VERSION
version_info = version_dict.get(version)
version_port = version_info["port"]
version_url = version_info.get("url", "")
env_pro_dic = api_def_dic.get(api_name)
main_url = env_pro_dic["url"]
# ========获取url截取点==========
main_url = change_url_port(main_url, version_port, version_url)
login_url = get_login_url(main_url)
# ==============================
ini_name_list = env_pro_dic["api_ini_name"]
# ====循环对比接口和数据ini文件，取到第一个接口数据所在的ini文件，支持多个ini文件存储数据,PS:同一案例中的接口需放在同一个ini文件===
for ini_name in ini_name_list:
cur_path = r'%s\config\api\%s' % (path, api_name)
ini_path = r"%s\%s" % (cur_path, ini_name)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data_dict = get_ini_default_contant(ini_path, None, None)[0]
if apis[0] in default_data_dict.keys():
break
# 执行案例前清空CASE_GEN_DIC
'''PS：经验证不能使用Settings.CASE_GEN_DIC = {}，否则后续其他用例调用值一直保持第一次的，
原因分析?: Settings.CASE_GEN_DIC = {} 对CASE_GEN_DIC的引用改变了，CASE_GEN_DIC的值未改变;clear则清空该引用中的值
'''
Settings.CASE_GEN_DIC.clear()
'''==========================================================================================='''
'''====================【对于固定登录的可以抽出来单独处理】--判断系统是否需要登录================='''
login_sign = RHIniParser(ini_path).get_ini_info("IS_LOGIN", "LOGIN_SIGN")
if login_sign is "Y":
# 获取登录请求地址
# 利用有序OrderedDict 有序读取url（有些登录需要验证码，需现请求获取验证码接口，防止验证码过期）
login_url_dict = json.loads(RHIniParser(ini_path).get_ini_info(
"IS_LOGIN", "LOGIN_URL"),
object_pairs_hook=OrderedDict)
for url in login_url_dict.keys():
ini_contant = get_ini_default_contant(ini_path, url, None)
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
# 走html-urlencode格式
if "highest" in url or "rest.jsf" in url or "j_security_check" in url:
post_data = transfor_to_dict(login_url_dict[url])
# 登录走json参数格式
else:
post_data = json.dumps(login_url_dict[url])
# 若为登录请求接口，替换对应参数兼容各个版本
change_login_dic(api_name, post_data, version_port)
# 替换urlinput设置的变量--url
url = change_in_json_urldic(in_para_dic, dic_args, url, urlparam)
# 替换in out设置的变量--data
change_in_json_dic(in_para_dic, dic_args, post_data)
response_json = json_request(login_url + url, post_data,
request_method)
# 获取正则表达式列表
default_re = default_re_set.split(";") # 存储特定返回值
out = get_out_json_dic(out_para_dic, response_json[0], default_re)
dic_args = get_new_dic(dic_args, out)
'''==========================================================================================='''
for i in xrange(api_len): # (0,2)
interface_name = apis[i]
api = Core.rh_get_api_name(apis[i])
# 再次循环对比接口和数据ini文件，取到接口数据所在的ini文件，支持多个ini文件存储数据
for ini_name in ini_name_list:
cur_path = r'%s\config\api\%s' % (path, api_name)
ini_path = r"%s\%s" % (cur_path, ini_name)
# 循环读取ini文件，找到接口数据所在的ini文件
default_data_dict = get_ini_default_contant(ini_path, None,
None)[0]
default_dbdata_dict = get_ini_default_contant(
ini_path, None, None)[4]
if apis[i] in default_data_dict.keys(
) or apis[i] in default_dbdata_dict.keys():
break
# 读取接口路径, 产品 [msp.apply.01; msp.apply; msp.tools.03 ]
if len(interface_path_product) == 1:
path_list = interface_path_product[0].split(".")
model = path_list[1]
if len(path_list) == 3:
product_name = path_list[2]
else:
product_name = None
else:
path_list = interface_path_product[i].split(".")
model = path_list[1]
if len(path_list) == 3:
product_name = path_list[2]
else:
product_name = None
model_url = json.loads(
RHIniParser(ini_path).get_ini_info("MODEL_URL",
"MODEL_URL_DATA"))[model]
# 读取对应api接口的默认值
ini_contant = get_ini_default_contant(ini_path, api, product_name)
if ".db" not in api and ".dmldb" not in api:
# 对应api接口的默认值，请求方式，出入参设置，正则表达式，特定返回值
default_contant = ini_contant[0]
request_method = ini_contant[1]
in_para_dic = ini_contant[2]
out_para_dic = ini_contant[2]
default_re_set = ini_contant[3]
if "||" in data_source[i * 2 + 1 -
j]: # 对rf的表格的取值，替换列的，注意替换项必须和url里面的一致
urlparam = data_source[i * 2 + 1 - j].split("||")[1]
if "highest" in interface_name or "rest.jsf" in interface_name or "j_security_check" in interface_name:
default_contant = transfor_to_dict(default_contant)
# 替换urlinput设置的变量--url
interface_name = change_in_json_urldic(in_para_dic, dic_args,
interface_name, urlparam)
# 替换in out设置的变量--data
change_in_json_dic(in_para_dic, dic_args, default_contant)
# 替换设置的参数值
change_para_json(default_contant, data_source[i * 2 + 1 - j].split(
"||")[0]) # data_source[1] data_source[3]...
# 走html处理的，或者是json 的get请求，不需要dumps
if "highest" in interface_name or "j_security_check" in interface_name or request_method == "get":
# #####==get请求参数要传dict，且dict中嵌套的list或者dict需要转换为str才能识别。=========#########
for k in default_contant:
if isinstance(default_contant[k], dict):
default_contant[k] = json.dumps(default_contant[k])
elif isinstance(default_contant[k], list):
default_contant[k] = str(default_contant[k])
else:
# post请求参数为json，做dumps处理
default_contant = json.dumps(default_contant)
# 更新token， 放入请求hearder中
token = change_in_token_dic(in_para_dic, dic_args, token)
# ============请求hearder组装START 如有需要，自行处理====================
# iaas_cloud : token验证
if 'cloud_manage' in api_name or "auto_engine" in api_name:
headers = {
"Content-Type": 'application/json',
'X-Auth-Token': token.encode('utf-8')
}
else:
headers = {"Content-Type": 'application/json'}
# ============请求hearder组装END========================================
print "请求接口: " + main_url + model_url + interface_name
# print default_contant
response_json = json_request(main_url + model_url + interface_name,
default_contant, request_method,
headers)
data = Core.rh_replace_data(data_source[i * 2 + 2 - j])
for i in data:
if re.search('code', str(i)) != None:
status_code = rh_get_rf_code(i)
if int(status_code) == response_json[1]:
logger.info("code(%s) check success" % status_code)
else:
logger.error(
"%s failed:\n***********post code:\n%s\n***********response code:%s"
% (api, status_code, response_json[1]))
raise Exception("api(%s) fail " % api)
else:
res = rh_json_check_result(i, response_json[0])
if "[success]" not in res:
logger.error(
"%s failed:\n***********post data:\n%s\n***********response data:%s"
% (api, default_contant, response_json[0]))
Verify.should_contain(res, "[success]")
# 获取正则表达式列表
default_re = default_re_set.split(";") # 存储特定返回值
out = get_out_json_dic(out_para_dic, response_json[0], default_re)
dic_args = get_new_dic(dic_args, out)
print "%s:%s" % (api, dic_args)
# ===================================================================================
# ============若无DB检查点，可以暂时忽略该部分代码，若需要添加，自行修改====================
elif ".db" in api:
j += 2
# 对于DB校验sql取数，产品取db前一个接口对于的产品。
# 获取检验的sql
sql_list = ini_contant
# 避免程序入库事务还未完成就执行DB校验，故等待1秒
sleep(1)
# 注意 此处传过去的sql为list（多条） 格式：saveApproval.slis.db
if 'msp' in api_name:
result = check_db(sql_list, version,
api.split('.')[1],
dic_args["applyBarCode"].split())
else:
result = check_db(sql_list, version, api.split('.')[1])
if not result.flag:
logger.info(result.msg, "%s failed" % api)
print result.msg, "%s failed" % api
Verify.should_be_true(result.flag)
print "%s success" % api
logger.info("%s success" % api)
# ==========若有需要，执行回滚脚本======================================#
else:
j += 2
# 对于DB回滚sql取数，产品取db前一个接口对应的产品。
# 获取需要回滚的sql
rollback_sqllist = ini_contant
# 避免程序入库事务还未完成就执行DB校验，故等待1秒
sleep(1)
# 执行回滚操作 格式：saveApproval.slis.dmldb
try:
# 特殊处理
if "applyInfoIntefaceToFH.slis.dmldb" in api:
dml_db(rollback_sqllist, version,
api.split('.')[1], dic_args["applyBarCode"].split())
else:
dml_db(rollback_sqllist, version, api.split('.')[1])
print "%s success" % api
logger.info("%s success" % api)
except Exception, e:
raise Exception("error occur in sql rollback, %s", str(e))
