接口测试学习-python第八课（数据驱动测试）

Posted 2020-11-10 myyard777

自动化分为三类，数据驱动即根据数据去进行测试；代码驱动即测试都是通过代码完成的；关键字驱动即依靠一些软件，利用关键字完成测试。本次主要记录数据驱动，自动化框架主要有以下几个部分：
①获取用例
②调用接口
③检验结果
④发送测试报告
⑤异常处理
⑥日志

以前进行接口封装的时候，有过将代码分别放在不同的文件夹下的经历。这次也以这样的模式来完成，主要做到以下几点：首先读取用例，其次根据用例完成接口测试，然后记录结果到用例文件中，最后发送邮件。同时各个部分关键结果处需要记录日志。

1、setting文件

setting文件一般放在config文件夹下。这个代码因为涉及到发送邮件，读取excel，写日志等。所以基础数据包括邮件的host，账号，密码等；还有文件地址和日志地址等。这些数据在后面的代码中都会使用到。

 1 import os
 2 
 3 
 4 # 发送邮件的一些固定参数，在sendmail函数中会使用到，可以在此处修改，此处的password是授权码
 5 MAIL_HOST = ‘smtp.163.com‘
 6 MAIL_USER = ‘********@163.com‘
 7 MAIL_PASSWORD = *****‘  # 授权码
 8 MAIL_TO = ‘[email protected]‘
 9 
10 BASE_PATH = os.path.dirname(
11     os.path.dirname(os.path.abspath(__file__))
12 )  # 找到APT这个文件夹的地址
13 LOG_PATH = os.path.join(BASE_PATH, ‘logs‘)  # 拼接出存放日志的路径
14 CASE_PATH = os.path.join(BASE_PATH, ‘cases‘)  # 拼接出存放用例的路径
15 
16 LEVEL = ‘debug‘  # 日志级别
17 LOG_NAME = ‘apt.log‘  # 日志文件名

2、日志模块

这个模块主要内容就是日志记录的功能。构造了一个可实例化的类，实例化后在其他模块可以直接调用生成相应等级的日志。

 1 import logging
 2 import os
 3 from logging import handlers
 4 from config import setting
 5 
 6 
 7 class MyLogger(object):
 8     def __init__(self,file_name,level=‘info‘,backCount=5,when=‘D‘):
 9         logger = logging.getLogger()
10         logger.setLevel(self.get_level(level))  # 设置日志的级别
11         # fl = logging.FileHandler(filename=‘a.log‘, mode=‘a‘, encoding=‘utf-8‘)
12         cl = logging.StreamHandler()  # 负责往控制台输出的
13         bl = handlers.TimedRotatingFileHandler(filename=file_name, when=when, interval=1,
14                                                backupCount=backCount, encoding=‘utf-8‘)
15         fmt = logging.Formatter(‘%(asctime)s - %(pathname)s[line:%(lineno)d]-%(levelname)s:%(message)s‘)
16         # 指定日志的格式
17         cl.setFormatter(fmt)  # 设置控制台输出的日志格式
18         bl.setFormatter(fmt)  # 设置文件里面写入的日志格式
19         logger.addHandler(cl)  # 把已经设置好的人放到办公室中
20         logger.addHandler(bl)  # 同上
21         self.logger = logger
22 
23     def get_level(self, sss):
24         level = {
25             ‘debug‘: logging.DEBUG,
26             ‘info‘: logging.INFO,
27             ‘warn‘: logging.WARN,
28             ‘error‘: logging.ERROR
29         }
30         sss = sss.lower()
31         return level.get(sss)
32 
33 path = os.path.join(setting.LOG_PATH,  setting.LOG_NAME)  # 拼好日志的绝对路径
34 apt_log = MyLogger(path, setting.LEVEL).logger
35 # 直接在这边实例化，并.logger，使用时就可以直接apt_log.warning了

3、发送邮件模块

用于记录发送邮件的功能，这里就用到了setting里面记录的那些基础数据，还用到了日志记录功能。

 1 import yagmail
 2 from config import setting
 3 from lib.log import apt_log
 4 
 5 
 6 def sendemile(title, content, attrs=None):  # 定义发送邮件函数
 7     m = yagmail.SMTP(host=setting.MAIL_HOST,  # host调用setting文件中设置的固定参数
 8                      user=setting.MAIL_USER,  # user调用setting文件中设置的固定参数
 9                      password=setting.MAIL_PASSWORD  # password调用setting文件中设置的固定参数
10                      )
11     m.send(to=setting.MAIL_TO,
12            subject=title,
13            contents=content,
14            attachments=attrs
15            )  # 邮件发送主体内容，收件人使用固定参数，其他内容在调用函数时传入
16     apt_log.info(‘邮件发送成功。‘)  # 写入发送邮件成功的日志

4、读取用例进行测试并记录测试结果模块

这是一个主体模块，完成了这个文件的主要功能。详细代码如下。

 1 import xlrd
 2 from lib.log import apt_log
 3 import requests
 4 from xlutils import copy
 5 
 6 
 7 class OpCase(object):
 8     def get_case(self, file_path):
 9         cases = []  # 存放所有的测试用例
10         if file_path.endswith(‘.xls‘) or file_path.endswith(‘.xlsx‘):  # 判断路径是否存在
11             try:
12                 book = xlrd.open_workbook(file_path)
13                 sheet = book.sheet_by_index(0)  # 打开用例文件
14                 for i in range(1, sheet.nrows):  # 从第二行开始遍历excel文件内容（第一行是标题）
15                     row_data = sheet.row_values(i)  # 获取每行内容
16                     cases.append(row_data[4:8])  # 在cases这个list中存每个用例的url,method,req_data,check
17                 apt_log.info(‘共读取%s条用例‘ % (len(cases)))  # 检查cases中的元素个数，日志记录读取了几条用例
18                 self.file_path = file_path  # 既然读取成功说明这个文件路径是正确的，那么这边定义下面写exccel就可以直接使用了
19             except Exception as e:
20                 apt_log.error(‘【%s】用例打开失败，错误信息：%s‘ % (file_path, e))
21         else:
22             apt_log.error(‘用例文件不合法，%s‘ % file_path)
23         return cases
24 
25     def dataToDict(self, data):  # 将用例中的req_data转为字典格式，可以直接用在request模块
26         res = {}
27         data = data.split(‘,‘)  # 这个地方要求用例中请求参数使用英文逗号分割的
28         # 这边分割入参后，入参变为[‘a=1‘, ‘b=2‘]这样的格式
29         for d in data:
30             k, v = d.split(‘=‘)
31             # 再次分割后变为[‘a‘,‘1‘]这样的模式分别用k代表a，v代表1
32             res[k] = v  # 存入字典中变成‘a‘:1，这样的模式可直接在request模块接口测试中作为入参使用
33         return res
34 
35     def my_request(self, url, method, data):
36         method = method.upper()  # 首先将请求方式变为大写模式，方便后面验证
37         data = self.dataToDict(data)  # 把a=1,b=2格式的请求参数变成字典格式
38         try:
39             if method == ‘POST‘:
40                 res = requests.post(url, data).text  # 如果请求方式是post按照post模式进行接口测试
41             elif method == ‘GET‘:
42                 res = requests.get(url, data).text  # 按照get方式传入url和data进行接口测试
43             else:
44                 apt_log.warning(‘暂不支持该请求‘)  # 其他的请求方式不支持，打印日志并返回提示
45                 res = ‘暂不支持该请求‘
46         except Exception as e:
47             msg = ‘【%s】接口调用失败,%s‘ % (url, e)  # 请求未成功提示接口调用失败，写入日志并返回提示
48             apt_log.error(msg)
49             res = msg
50         return res
51 
52     def check_res(self, res, check):
53         res.replace(‘": "‘, ‘=‘).replace(‘": ‘, ‘=‘)  # 返回的报文是json格式的，利用字符替换让数据变成"a=1,b=2"的格式
54         for c in check.split(‘,‘):  # 判断验证数据是否在返回的报文中
55             if c not in res:
56                 apt_log.info(‘结果校验失败，预期结果：【%s】，实际结果【%s】‘ % (c, res))  # 不匹配则测试校验失败，打印日志
57                 return ‘失败‘  # 返回失败的提示
58         return ‘通过‘  # 如果存在说明测试通过
59 
60     def write_excel(self, cases_res):  # 入参是每个用例的执行结果
61         book = xlrd.open_workbook(self.file_path)
62         new_book = copy.copy(book)
63         sheet = new_book.get_sheet(0)  # 复制测试用例excel文件
64         row = 1
65         for case_res in cases_res:  # 遍历写入数据
66             sheet.write(row, 8, case_res[0])  # 第八列写返回结果
67             sheet.write(row, 9, case_res[1])  # 第九列写测试是否通过
68             row += 1
69         new_book.save(self.file_path.replace(‘xlsx‘, ‘xls‘))  # 保存结果
70         

5、start文件

启动文件，用于集成所有的模块完成目标。

 1 import os
 2 import sys
 3 from lib.common import OpCase
 4 from lib.send_emile import sendemile
 5 from config import setting
 6 
 7 BASE_PATH = os.path.dirname(
 8     os.path.dirname(os.path.abspath(__file__))
 9 )  # 找到APT这个文件夹的地址
10 sys.path.insert(0, BASE_PATH)  # 把ATP目录加到环境变量中，用于在其他不能帮忙加环境变量的时候
11 
12 
13 class CaseRun(object):
14     def find_cases(self):
15         op = OpCase()  # 实例化了操作用例这个类
16         for f in os.listdir(setting.CASE_PATH):  # 获取到cases文件夹下的文件
17             abs_path = os.path.join(setting.CASE_PATH, f)  # 拼接测试用例文件的绝对路径
18             case_list = op.get_case(abs_path)  # 利用get_case函数获取到每个用例中的case
19             res_list = []  # 创建一个空的list来存放测试返回的报文和测试结果
20             pass_count = 0  # 用来计算成功用例个数
21             fail_count = 0  # 用来计算失败用例格式
22             for case in case_list:
23                 url, method, req_data, check = case  # get_case函数中提取每行用例的四个元素，分别定义一下
24                 res = op.my_request(url, method, req_data)  # 调用my_request函数进行测试接口返回结果
25                 status = op.check_res(res, check)  # 检查测试返回报文是否和预计测试结果相同，如果相同测试通过，否则测试失败
26                 res_list.append([res, status])  # 为了一次性写入测试结果，先将结果放入list中
27                 if status == ‘通过‘:
28                     pass_count += 1
29                 else:
30                     fail_count += 1
31             op.write_excel(res_list)  # 写入excel
32             msg = ‘‘‘  # 定义邮件正文
33             XX你好：
34                 本次共运行%s条用例，通过%s条，失败%s条。
35                 ‘‘‘ % (len(res_list), pass_count, fail_count)
36             sendemile(‘测试用例运行结果‘, content=msg, attrs=abs_path)  # 利用sendmail函数发送邮件，入参分别是主题，内容和附件
37 
38 
39 CaseRun().find_cases()  # 运行find_cases函数