Commit 636d41688769a51e5550ff5eda38bf1727218c4a

Authored by lixi
1 parent f960ecf0

1

Showing 2 changed files with 83 additions and 70 deletions
commons/MySession.py
1 1 #!/usr/bin/python
2 2 # -*- coding: UTF-8 -*-
3 3 import requests
  4 +import re
4 5 from commons import common as com
5 6 from commons.Logging import log
6 7 from commons.clientSession import cliSession
... ... @@ -36,16 +37,17 @@ class mysession(requests.Session):
36 37 body = "userName=sg_wenze&password=111111"
37 38 body_client= {"userName":"sg_wenze","password":"111111"}
38 39  
39   - def __init__(self):
  40 + def __init__(self,host=None):
40 41 "如下代码,可以通过配置文件来控制测试环境和灰度环境,http和https"
41 42 super().__init__()
  43 + self.user={}
42 44 self.url = mysession.url.replace("http://test.", com.get_global_config("global_data", "environment", "en"))
43 45 self.header = mysession.header
44 46 self.body = mysession.body
45 47 self.url_client = mysession.url_client.replace("http://test.", com.get_global_config("global_data", "environment", "en"))
46 48 self.header_client = mysession.header_client
47 49 self.body_client = mysession.body_client
48   - self.timeout = (6,6)
  50 + self.timeout = (10,10)
49 51 self.max_retries = 3
50 52 self.keep_alive = False
51 53 self.ssl_verify = False
... ... @@ -55,36 +57,48 @@ class mysession(requests.Session):
55 57 self.allow_redirects = False
56 58 self.firmid={"group":"1","hd":"2","cd":"3","qqhe":"4","mdj":"5","gy":"6","cc":"7","sg":"8","sy":"9"}
57 59 self.market={"sy":"沈阳","heb":"哈尔滨","sg":"寿光","gy":"贵阳","cc":"长春","hs":"杭水","hg":"杭果"}
58   - self.user={}
  60 + self.host= None if host==None else eval(com.get_global_config("global_data", "host_ip", host))
  61 +
59 62  
60 63 def cliLogin(self,user="sy_userName_01"):
61 64 self.webHeaders, self.clientHeaders, self.userInfo = cliSession().loginUser(user=user)
62 65 return self
63 66  
64   - def get_session(self, account, **kwargs):
65   - "如下代码,可以通过配置文件来控制登录的账户session"
66   - self.body = self.body.replace("sg_wenze",
67   - com.get_global_config("global_data", "account", account).split("&")[0])
68   - self.body = self.body.replace("111111",
69   - com.get_global_config("global_data", "account", account).split("&")[1])
70   - self.se = requests.session()
71   - co = requests.cookies.RequestsCookieJar()
72   - #加入UAP_firmId属性
73   - firm=account.split("_")[0]
74   - co.set("UAP_firmId", self.firmid[firm])
75   - self.se.cookies.update(co)
76   - # 进行登录请求
77   - re = self.se.post(url=self.url, headers=self.header, data=self.body, proxies=self.proxies, **kwargs)
78   - self.UAP_accessToken=self.se.cookies["UAP_accessToken"]
79   - self.UAP_refreshToken=self.se.cookies["UAP_refreshToken"]
80   - return self.se
  67 + # def get_session(self, account, **kwargs):
  68 + # "如下代码,可以通过配置文件来控制登录的账户session"
  69 + # self.body = self.body.replace("sg_wenze",
  70 + # com.get_global_config("global_data", "account", account).split("&")[0])
  71 + # self.body = self.body.replace("111111",
  72 + # com.get_global_config("global_data", "account", account).split("&")[1])
  73 + # self.se = requests.session()
  74 + # co = requests.cookies.RequestsCookieJar()
  75 + # #加入UAP_firmId属性
  76 + # firm=account.split("_")[0]
  77 + # co.set("UAP_firmId", self.firmid[firm])
  78 + # self.se.cookies.update(co)
  79 + # # 进行登录请求
  80 + # re = self.se.post(url=self.url, headers=self.header, data=self.body, proxies=self.proxies, **kwargs)
  81 + # self.UAP_accessToken=self.se.cookies["UAP_accessToken"]
  82 + # self.UAP_refreshToken=self.se.cookies["UAP_refreshToken"]
  83 + # return self.se
  84 + #
  85 + # def get_login_info(self, account, **kwargs):
  86 + # "用于获取用户信息"
  87 + # self.body_client.update({"userName":com.get_global_config("global_data", "account", account).split("&")[0]})
  88 + # self.body_client.update({"password":pwdCry(com.get_global_config("global_data", "account", account).split("&")[1])})
  89 + # tmp = requests.post(url=self.url_client, headers=self.header_client, json=self.body_client, proxies=self.proxies, **kwargs)
  90 + # return tmp
81 91  
82   - def get_login_info(self, account, **kwargs):
83   - "用于获取用户信息"
84   - self.body_client.update({"userName":com.get_global_config("global_data", "account", account).split("&")[0]})
85   - self.body_client.update({"password":pwdCry(com.get_global_config("global_data", "account", account).split("&")[1])})
86   - tmp = requests.post(url=self.url_client, headers=self.header_client, json=self.body_client, proxies=self.proxies, **kwargs)
87   - return tmp
  92 + def url_pro(self, url, host):
  93 + # url = url.replace(" ", "")
  94 + if host!=None:
  95 + if "http:" in url:
  96 + d1 = re.match(r"http://(.+?)/", url).group(1).split(":")[0]
  97 + url = re.sub(r"http://(.+?)/", r"http://" + host[d1] + "/", url)
  98 + elif "https:" in url:
  99 + d1 = re.match(r"https://(.+?)/", url).group(1).split(":")[0]
  100 + url = re.sub(r"https://(.+?)/", r"https://" + host[d1] + "/", url)
  101 + return url
88 102  
89 103 def get_session_client(self, account, **kwargs):
90 104 "get_session和get_session_client的方法只能用一个"
... ... @@ -151,6 +165,8 @@ class mysession(requests.Session):
151 165 """
152 166 # 记录日志
153 167 log.info("{0:=^86}".format(''))
  168 + log.info(url)
  169 + url=self.url_pro(url,self.host)
154 170 log.info("{}\n{}\n".format(url, kwargs))
155 171 # 进行请求
156 172 re = super().request(method , url, **kwargs,timeout=self.timeout)
... ... @@ -262,15 +278,15 @@ class mysession(requests.Session):
262 278  
263 279 my = mysession()
264 280 my.set_mark()
265   -
266 281 # 沈阳客户端session
267 282 sessionSy = mysession().cliLogin("sy_userName_01")
268 283 # 哈尔滨客户端session
269 284 sessionHeb = mysession().cliLogin("hd_userName_01")
270   -# print(sessionSy.userInfo)
271   -# print(sessionHeb.userInfo)
272   -
273   -sy1=mysession().get_session_client("sy_user01")
274   -heb=mysession().get_session_client("heb_user01")
275   -# sg=mysession().get_session_client("sg_user01")
276   -
  285 +#获取对应市场session
  286 +sy1=mysession("host1").get_session_client("sy_user01")
  287 +heb=mysession("host1").get_session_client("heb_user01")
  288 +hg=mysession("host2").get_session_client("hg_user01")
  289 +# 检测登录接口
  290 +sy1.check_login("sy_user01")
  291 +# 检测登录接口
  292 +hg.check_login("hg_user01")
277 293 \ No newline at end of file
... ...
1 1 #!/usr/bin/python
2 2 # -*- coding: UTF-8 -*-
3   -import os,time
  3 +import os, time
4 4 import sys
5 5 import unittest
6 6 import re
... ... @@ -13,72 +13,69 @@ from commons.MySession import my
13 13 from commons.scripts import delReport
14 14  
15 15  
16   -
17 16 def Create_Testcase_suite(path=""):
18 17 '''创建测试套件'''
19   - if path!="":
20   - path+="/"
  18 + if path != "":
  19 + path += "/"
21 20 print(path)
22   - testunit=unittest.TestSuite()
  21 + testunit = unittest.TestSuite()
23 22 test_Loader = DiscoveringTestLoader()
24   - discover=test_Loader.discover("./testcase/{}".format(path),pattern='test_*.py',top_level_dir=None)
  23 + discover = test_Loader.discover("./testcase/{}".format(path), pattern='test_*.py', top_level_dir=None)
25 24 # print(discover)
26 25 for test_suite in discover:
27 26 testunit.addTests(test_suite)
28   -# print(testunit)
  27 + # print(testunit)
29 28 return testunit
30 29  
31   -
32   -def Run_Testcase(testsuit,head=""):
  30 +
  31 +def Run_Testcase(testsuit, head=""):
33 32 '''运行测试用例并生成报告'''
34 33 if (head in my.market.keys()):
35   - head="["+ my.market[head] + "]"
36   - now = time.strftime("%Y-%m-%d %H_%M_%S",time.localtime())
37   - path=os.path.dirname(os.path.abspath(sys.argv[0]))
38   - report_file=path+"/report/"+now+"_result.html"
39   - #创建报告文件
40   - fp=open(report_file,'wb')
  34 + head = "[" + my.market[head] + "]"
  35 + now = time.strftime("%Y-%m-%d %H_%M_%S", time.localtime())
  36 + path = os.path.dirname(os.path.abspath(sys.argv[0]))
  37 + report_file = path + "/report/" + now + "_result.html"
  38 + # 创建报告文件
  39 + fp = open(report_file, 'wb')
41 40  
42   - runner=HTMLTestRunner_cn.HTMLTestRunner(
  41 + runner = HTMLTestRunner_cn.HTMLTestRunner(
43 42 stream=fp,
44 43 title=u'进门收费{}--接口测试报告'.format(head),
45 44 description=u'用例简要执行情况如下:(注:报告详细信息需要下载report.html并用浏览器打开)',
46   - verbosity = 2)
47   - #执行用例
  45 + verbosity=2)
  46 + # 执行用例
48 47 runner.run(testsuit)
49   - #关闭文件
  48 + # 关闭文件
50 49 fp.close()
51 50 delReport.delReport(path)
52 51 return report_file
53   -
  52 +
  53 +
54 54 def Send_email(filename):
55 55 '''判断邮件发送逻辑'''
56   - l=eval(com.get_global_config("global_data", "email","to_list").lower())
57   - c=eval(com.get_global_config("global_data", "email","cc_list").lower())
58   - if type(l)!=type([]):
  56 + l = eval(com.get_global_config("global_data", "email", "to_list").lower())
  57 + c = eval(com.get_global_config("global_data", "email", "cc_list").lower())
  58 + if type(l) != type([]):
59 59 raise Exception("error,pls input list type send-email address")
60   - elif len(l)==0:
  60 + elif len(l) == 0:
61 61 print("\n To_list of send-email is null,won't send email!")
62   - elif len(l)!=0:
  62 + elif len(l) != 0:
63 63 for i in l:
64 64 print("\n check send-email format : {}".format(i))
65   - if re.match(r'^[0-9a-zA-Z_]{1,19}@[0-9a-zA-Z]{1,13}\.[com,cn,net]{1,3}$',i):
  65 + if re.match(r'^[0-9a-zA-Z_]{1,19}@[0-9a-zA-Z]{1,13}\.[com,cn,net]{1,3}$', i):
66 66 pass
67 67 else:
68 68 raise Exception("error,pls check your send-email format")
69   - #发送邮件
70   - em.send_email(filename,send_to=l,cc_to=c)
  69 + # 发送邮件
  70 + em.send_email(filename, send_to=l, cc_to=c)
71 71 else:
72 72 print("\n Haven't sent the email,pls check send-email address!")
73   -
74 73  
75 74  
76 75 if __name__ == "__main__":
77   - #检测数据库
  76 + # 检测数据库
78 77 db.mysql_conn_test()
79   - #检测登录接口
80   - my.check_login("sy_user01")
81   - #创建测试套,运行测试用例,生成报告
82   - report=Run_Testcase(Create_Testcase_suite("sy"))
83   - #发送邮件
84   - Send_email(report)
  78 + # 创建测试套,运行测试用例,生成报告
  79 + report = Run_Testcase(Create_Testcase_suite())
  80 + # 发送邮件
  81 + Send_email(report)
85 82 \ No newline at end of file
... ...