python自动化学习笔记10-数据驱动DDT与yml的应用

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

在测试工作中，针对某一API接口，或者某一个用户界面的输入框，需要设计大量相关的用例，每一个用例包含实际输入的各种可能的数据。通常的做法是，将测试数据存放到一个数据文件里，然后从数据文件读取，在脚本中循环输入测试数据，并对结果进行验证。
我们用Python的unittest+DDT结合的方式；

DDT包含类的装饰器ddt和两个方法装饰器data（直接输入测试数据），file_data（可以从json或者yaml中获取测试数据）

unittest框架：

import unittestdef add(a,b):
res=a+b
return resclass MyTest(unittest.TestCase):
#初始化
def setUp(self):
print('setup')
#结束
def tearDown(self):
print('tearDown') def test_add(self):
res=add(1,2)
self.assertEqual(res,3)if __name__ == '__main__':
unittest.main()

Unittest框架包含一个test Fixture，test Fixture由三部分组成，setup，testcase和teardown。Setup过程，是测试用例执行前的初始化过程，teardown过程，是在测试用例执行后，对资源进行释放与回收的过程；而testcase是具体的测试用例。
使用ddt模块需要先安装，并进行导入；

首先在头部导入ddt；

其次在测试类前声明使用ddt（@ddt.ddt）;

第三步，在测试方法前，使用@ddt.data和@unpack进行修饰。

第四步、在data中填加测试数据

如下：

import unittest
import ddtdef add(a,b):
res=a+b
return res#声明要使用ddt
@ddt.ddt
class MyTest(unittest.TestCase):
#初始化
def setUp(self):
print('setup') #ddt.data中添加测试数据
@ddt.data([1,2,3],[0,0,0],[999,1,1000])
@ddt.unpack
def test_add(self,a,b,e):
res=add(a,b)
self.assertEqual(res,e)
#结束
def tearDown(self):
print('tearDown')if __name__ == '__main__':
unittest.main()

执行查看结果：

可以看到我们程序中并没有写循环操作，测试框架自动的将测试数据分成多条测试用例来执行；

当测试数据较多的时候，直接写在ddt.data中又会使代码看起来比较混乱，所以我们可以使用ddt.file_data()直接读取yml文件来读取测试数据，如下：

yaml文件中我们定义了一个二维数组：

Python文件代码：

import unittest
import ddt

def add(a,b):
res=a+b
return res

#声明要使用ddt
@ddt.ddt
class MyTest(unittest.TestCase):
#初始化
def setUp(self):
print('setup')

#ddt.data中添加测试数据
# @ddt.data([1,2,3],[2,3,5])
@ddt.file_data('data.yml')
@ddt.unpack
def test_add(self,value):
print(value)
a=value[0]
b=value[1]
e=value[2]
res=add(a,b)
self.assertEqual(res,e)
#结束
def tearDown(self):
print('tearDown')

if __name__ == '__main__':
unittest.main()

上述代码我们加了print（value），为的是看一下value的取值，执行查看结果：

我们可以看到value的取值是一个list，读取yaml文件是一个二维数组[[1,2,3],[2,3,5]],循环取值后，取出一维数组[1,2,3]，通过角标获取出对应的值，执行了需要测试的函数；

当yaml文件中存在字典格式时我们就需要传入一个**kwargs参数来保存相应的值；

示例如下：

以公司的一个登录接口为例，将测试用例写在yml文件中，Python中调用yml文件获取测试数据：

import unittest
import ddt
import requests
def add(a,b):
res=a+b
return res#声明要使用ddt
@ddt.ddt
class MyTest(unittest.TestCase):
#初始化
def setUp(self):
print('setup') #ddt.data中添加测试数据
# @ddt.data([1,2,3],[2,3,5])
# @ddt.file_data('data.yml')
# @ddt.unpack
# def test_add(self,value):
# print(value)
# a=value[0]
# b=value[1]
# e=value[2]
# res=add(a,b)
# self.assertEqual(res,e)
@ddt.file_data('jzjz_data.yml')
@ddt.unpack
def test_login(self,**kwargs):
#由于获取yml文件数据后是一个字典表，所以存入**kwargs中，从kwargs中获取各参数的值
#一般接口调用都会用到的参数信息，data，header，cookie，is_json 设置了默认值，如果测试数据中没有传，则传入默认值
url=kwargs.get('url')
method=kwargs.get('method')
data=kwargs.get('data',{})
check=kwargs.get('check')
header=kwargs.get('header',{})
is_json=kwargs.get('is_json',0)
cookie=kwargs.get('cookie',{})
#如果是get方法则走该分支
if method.upper=='GET':
res=requests.get(url=url,params=data,cookies=cookie,headers=header)
self.assertIn(check,res)
#如果是post请求走这个分支
else:
#如果传入的参数是json类型的，则类型写为json
if is_json==1:
res=requests.post(url=url,json=data,cookies=cookie,headers=header)
#如果传入的不是json，data正常传递给data参数
else:
#因为返回类型是页面的形式，我们返回值转化成为text，如果是json类型的，可以转化成json类型
res=requests.post(url=url,data=data,cookies=cookie,headers=header).text
self.assertIn(check,res) #结束
def tearDown(self):
print('tearDown')if __name__ == '__main__':
unittest.main()

yml文件的内容：

执行Python文件，查看结果：

setup
.tearDown
setup
.tearDown
setup
.tearDown
setup
.
tearDown
----------------------------------------------------------------------
Ran 4 tests in 36.532s

测试用例直行通过，实际结果与预期结果一致；

在实际测试中，我们可能会用到发送邮件和生成测试报告，之前的学习中也已经说过了，这里简单写一下生成测试报告的过程：

if __name__ == '__main__':
# unittest.main()
#创建一个测试套件
suite=unittest.TestSuite()
#套件中加入所有的测试用例
suite.addTests(unittest.makeSuite(MyTest))
#创建测试报告
result=BeautifulReport(suite)
#生成报告
result.report(description='登录接口',filename='login_report.html',log_path=r'C:/besttest/newpython/besttest_code/练习/day11笔记')

执行查看结果，生成了测试报告：