利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

时间:2021-09-11 15:33:56

知道ddt的基本使用方法之后,练习把之前用excel文件来维护的接口测试用例改用unittest+ddt来实现。

这里我选用yaml文件来管理接口参数,开始本来想用json,但是json无法添加注释,可读性不好。

1.向yaml文件中添加数据

下面截图是接口文档中的各个接口,每个接口都有一个固定的序号,所以在设计每个接口的测试数据时,以序号来区分不同接口

利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

yaml文件内容如下,需要注意的是yaml的语法:

(1)键值对用冒号分割,但是冒号后需要加一个空格

(2)禁止使用tab缩进,只能使用空格键;缩进长度没有限制,只要元素对齐就表示这些元素属于一个层级

(3)字符串可以不用引号标注,也可以加引号,如果想把数字变为字符串,加引号即可

(4)使用#表示注释

详情可以参考博客:https://blog.csdn.net/vincent_hbl/article/details/75411243

利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

2.简单demo:python读取yaml文件,取出接口参数

import yaml

fp = open('../dataconfig/信息互动模块接口.yaml', encoding='utf-8')   # 有中文字符的话,加编码格式
testdata = yaml.load(fp)
t = testdata['5.2.1.4']
print(t)

利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

3.完整流程

(1)封装读取yaml文件方法

handle_yaml.py

# coding: utf-8
# author: hmk import yaml
import os class HandleYaml:
def __init__(self, file_path=None):
if file_path:
self.file_path = file_path
else:
root_dir = os.path.dirname(os.path.abspath('.'))
# os.path.abspath('.')表示获取当前文件所在目录;os.path.dirname表示获取文件所在父目录;所以整个就是项目的所在路径
self.file_path = root_dir + '/dataconfig/信息互动模块接口.yaml' # 获取文件所在的相对路径(相对整个项目)
# self.data = self.get_data() def get_data(self):
fp = open(self.file_path, encoding='utf-8')
data = yaml.load(fp)
return data if __name__ == '__main__':
test = HandleYaml()
p = test.get_data()
print(p['5.2.1.1'])

(2)封装requests请求方法

# coding: utf-8
# author: Archer import requests
import json class RunMethod:
def post_main(self, url, data, header=None):if header is not None:
res = requests.post(url=url, data=data, headers=header)
else:
res = requests.post(url=url, data=data)
# print(res.status_code)
# return res.json()
return res #为了方便后面断言,这里不再对服务器响应进行json格式编码 def get_main(self, url, data=None, header=None):if header is not None:
res = requests.get(url=url, params=data, headers=header)
else:
res = requests.get(url=url, params=data)
print(res.status_code)
# return res.json()
return res def run_main(self, method, url, data=None, header=None):if method == 'POST':
res = self.post_main(url, data, header)
else:
res = self.get_main(url, data, header)
return res
# return json.dumps(res, indent=2, sort_keys=False, ensure_ascii=False) # 使用json模块格式化显示结果

(3)一个接口测试用例

# coding: utf-8
# author: Archer import unittest
import ddt
from base.run_method import RunMethod
from utils.handle_yaml import HandleYaml get_data = HandleYaml() # 从yaml文件中取出该接口的参数
params = get_data.get_data()['5.2.1.4'] @ddt.ddt
class Test(unittest.TestCase):
"""加载咨询详情接口""" def setUp(self):
self.url = 'http://localhost:8088/ApprExclusiveInterface/api/enterprise/info/consult/loadDetail.v'
self.run = RunMethod() @ddt.data(*params)
def test(self, value):
r = self.run.run_main("GET", self.url, value)
print(r)
self.assertTrue(value['assert'] in r.text) if __name__ == '__main__':
unittest.main()

利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

(4)利用HTMLTestRunner生成测试报告

run_report.py

# coding: utf-8
# author: hmk from HTMLTestRunner import HTMLTestRunner
import unittest
import time, os root_dir = os.path.dirname(os.path.abspath('.')) # 获取当前文件所在目录的父目录的绝对路径,也就是项目所在路径E:\DDT_Interface
case_dir = root_dir + '/test_case/' # 根据项目所在路径,找到用例所在的相对项目的路径
print(root_dir)
print(case_dir)
"""定义discover方法"""
discover = unittest.defaultTestLoader.discover(case_dir,
pattern='test*.py',
top_level_dir=None)
"""
1.case_dir即测试用例所在目录
2.pattern='test_*.py' :表示用例文件名的匹配原则,“*”表示任意多个字符
3.top_level_dir=None:测试模块的顶层目录。如果没顶层目录(也就是说测试用例不是放在多级目录
中),默认为 None
""" if __name__ == "__main__":
"""直接加载discover"""
now = time.strftime("%Y-%m-%d %H_%M_%S")
filename = root_dir + '/report/' + now + '_result.html' # 定义报告存放路径
print(filename)
fp = open(filename, 'wb')
runner = HTMLTestRunner(stream=fp,
title='个人网企业网接口测试报告',
description='测试结果如下: ')
runner.run(discover)
fp.close()

利用unittest+ddt进行接口测试(二):使用yaml文件管理测试数据

ok ,unittest+ddt进行接口测试就完成了,还有很多不足,yaml配置文件还可以继续设计优化,例如可以把请求url也加进去。

其实感觉如何设计测试用例,组织测试数据也是一件很有意思的事情,很多事情都必须先有一个良好的设计思路才会进行的更顺畅。总之勤于思考,多参考他人的思路。不是有句话吗,学而不思则罔,思而不学则殆。


2018-04-29 12:53:22