python-单元测试优化，加入日志

HttpRequests.py

 1 #-*- coding:utf-8 -*-
 2 
 3 import requests
 4 class HttpRequests():
 5     def http_requests(self,url,params,http_mothed,cookies=None):
 6         if http_mothed=='get':
 7             res=requests.get(url,params,cookies=cookies)
 8             return res
 9         else:
10             res=requests.post(url,params,cookies=cookies)
11             return res

my_log.py

 1 #-*- coding:utf-8 -*-
 2 
 3 
 4 import logging
 5 from common import project_path
 6 from common.read_config import ReadConfig
 7 class MyLog:
 8 
 9  def my_log(self,level,msg):
10   # 创建一个日志收集器并命名
11     my_logger=logging.getLogger(ReadConfig().read_config(project_path.conf_path,'LOGS','logger_name'))
12     my_logger.setLevel(ReadConfig().read_config(project_path.conf_path,'LOGS','log_level'))# 设置收集器的级别
13 
14   # 设置输出格式
15     formatter = logging.Formatter(ReadConfig().read_config(project_path.conf_path,'LOGS','formatter'))
16 
17   # 指定输出渠道，输出到控制台
18     ch=logging.StreamHandler()#输入到控制台
19     ch.setLevel(ReadConfig().read_config(project_path.conf_path,'LOGS','console_handle'))# 设置输出渠道的级别；
20   # 如果收集器中的级别与输出渠道级别不同，取两者的交集
21     ch.setFormatter(formatter)#日志输出时，按指定的输出格式输出
22 
23   #输出到本地文件
24     fh=logging.FileHandler(project_path.log_path,encoding='utf-8')#输出到本地log_file文档中；日志中有中文，需要设置编码
25     fh.setLevel(ReadConfig().read_config(project_path.conf_path,'LOGS','file_handle'))#输出级别为debug
26     fh.setFormatter(formatter)#日志输出时，按指定的输出格式输出
27 
28   #对接:收集器与输出渠道
29     my_logger.addHandler(ch)
30     my_logger.addHandler(fh)
31 
32     if level == 'DEBUG':
33         my_logger.debug(msg)
34     elif level == 'INFO':
35         my_logger.info(msg)
36     elif level == 'WARNING':
37         my_logger.warning(msg)
38     elif level == 'ERROR':
39         my_logger.error(msg)
40     elif level == 'CRITICAL':
41         my_logger.critical(msg)
42 
43     #将渠道移除
44     my_logger.removeHandler(ch)
45     my_logger.removeHandler(fh)
46     fh.close()
47 
48 
49  def debug(self,msg):
50     self.my_log('DEBUG',msg)
51  def info(self,msg):
52     self.my_log('INFO',msg)
53  def warning(self,msg):
54     self.my_log('WARNING',msg)
55  def error(self,msg):
56     self.my_log('ERROR',msg)
57  def crtical(self,msg):
58     self.my_log('CRTICAL',msg)
59 
60 
61 if __name__ == '__main__':
62   my_logger=MyLog()
63   my_logger.debug('haha')
64   my_logger.info('heng')

project_path.py

 1 # @File : project_path.py
 2 
 3 #-*- coding:utf-8 -*-
 4 #存放路径变量
 5 import os
 6 base_path=os.path.split(os.path.split(os.path.realpath(__file__))[0])[0] #api_auto目录
 7 
 8 #配置文件路径
 9 conf_path=os.path.join(base_path,'conf','test_api.conf')
10 
11 #测试报告路径
12 report_path=os.path.join(base_path,'test_result','report','test_api.html')
13 
14 #测试用例的路径
15 case_path=os.path.join(base_path,'test_case','http_requests.xlsx')
16 
17 #日志的路径
18 log_path=os.path.join(base_path,'test_result','log','log_file.txt')

read_config.py

 1 # @File : read_config.py
 2 
 3 #-*- coding:utf-8 -*-
 4 
 5 import configparser
 6 class ReadConfig():
 7     def read_config(self,file_name,section,option):
 8         cf = configparser.ConfigParser()#创建对象
 9         cf.read(file_name,encoding ='utf-8')#打开配置文件
10         value=cf.get(section,option)
11         return value
12 if __name__ == '__main__':
13     value=ReadConfig().read_config('test_api.conf','student1','name')
14     print(value)

read_excel.py

 1 #-*- coding:utf-8 -*-
 2 
 3 from openpyxl import load_workbook
 4 import configparser
 5 
 6 
 7 class TestExcel():
 8 
 9     def get_TestExcel(self,file_name,sheet_name,button):
10         workbook = load_workbook(file_name)#打开表格
11         sheet = workbook[sheet_name]#定位表单
12 
13         test_data = []#把所有行的数据放到列表中
14 
15         for i in range(2,sheet.max_row+1):
16             sub_data = {}#把每行的数据放到字典中
17             for j in range(1,sheet.max_column+1):
18 
19                 sub_data[sheet.cell(1,j).value] = sheet.cell(i,j).value
20 
21             test_data.append(sub_data)#把每一行数据添加到test_data列表里
22 
23         final_data=[]#最后的测试数据
24         if button == 'all':
25             final_data=test_data
26         else:
27             for item in test_data: #遍历test_data里面的数据
28                 if item['case_id'] in eval(button):
29                     #如果表格中case_id在button里面，就把这条用例加到final_data里面
30                     final_data.append(item)
31 
32         return final_data

test_HttpRequests.py

 1 #-*- coding:utf-8 -*-
 2 
 3 #存放测试用例
 4 import unittest
 5 
 6 from ddt import ddt,data
 7 
 8 from common.HttpRequests import HttpRequests
 9 from common.read_config import ReadConfig
10 from common.read_excel import TestExcel
11 from common.my_log import MyLog
12 from common import project_path
13 
14 COOKIES=None
15 
16 #利用配置类，从配置文件case.conf里面读取section：case;option:button的values值
17 button=ReadConfig().read_config(project_path.conf_path,'case','button')
18 
19 #调用task_04_do_excel这个模块里面的TestExcel类里面的get_TestExcel方法
20 #此方法需要 文件名，表单名，配置的值 这几个参数
21 test_data=TestExcel().get_TestExcel(project_path.case_path,'Sheet1',button)
22 
23 @ddt
24 class TestHttpRequests(unittest.TestCase):
25 
26 
27     def setUp(self):
28         # self.url='http://47.107.168.87:8080/futureloan/mvc/api/member/login'
29         pass
30 
31     @data(*test_data)
32     def test_api(self,item):#登录成功 ：手机号正确、密码正确；用例里面不可以传参数;但是如果用@data装饰以后，要用等量的变量接收，必须赋值
33         global COOKIES#声明全局变量
34         my_logger=MyLog()
35 
36         my_logger.info('目前正在执行第{}条用例：{}'.format(item['case_id'],item['title']))
37         my_logger.info('------开始检查url地址----------')
38         my_logger.info('url:{}'.format(item['url']))
39         my_logger.info('------开始检查params----------')
40         my_logger.info('params:{}'.format(item['params']))
41         my_logger.info('---------开始http请求----------')
42 
43         res=HttpRequests().http_requests(item['url'],eval(item['params']),item['http_method'],COOKIES)
44         my_logger.info('---------结束http请求----------')
45 
46 
47         try:
48             self.assertEqual(item['excepted'],res.json()['msg'])
49             # TestReslt='pass'
50         except AssertionError as e:
51             # TestReslt='Failed'
52             my_logger.error('断言结果是：{}'.format(e))
53             raise e
54         # finally:
55         #     my_logger.info('本次用例的测试结论是：{}'.format(e))
56         #     my_logger.info('看看TestReslt的值')
57 
58         if res.cookies:
59             COOKIES=res.cookies#当res.cookies非空时，修改COOKIES的值

test_runner.py

 1 #-*- coding:utf-8 -*-
 2 
 3 
 4 import unittest
 5 from common import project_path
 6 import HTMLTestRunnerNew
 7 
 8 
 9 from common.test_HttpRequests import TestHttpRequests
10 
11 
12 suit=unittest.TestSuite()##装用例的地方
13 loader=unittest.TestLoader()
14 suit.addTest(loader.loadTestsFromTestCase(TestHttpRequests))
15 
16 
17 #执行用例
18 with open(project_path.report_path,'bw+') as file:
19     Runner=HTMLTestRunnerNew.HTMLTestRunner(stream=file,verbosity=2,
20                                             title='HTTP请求作业-单元测试报告',description='测试一下HTTP请求',tester='huimin'
21                                             )
22     Runner.run(suit)

test_api.conf

#button = all 执行表格中所有用例
#button=[x,y] int(x)<int(y),执行x-y行之间的用例

[case]
button=all

[LOGS]
log_level=DEBUG
console_handle=ERROR
file_handle=DEBUG
formatter=%%(asctime)s - %%(filename)s - %%(name)s - %%(levelname)s - 日志信息： %%(message)s
logger_name=Mina

http_requests.xlsx

执行结果