七、数据驱动(unittest ddt)☆
ddt:data-driver tests
数据驱动解决的问题是:
1)、代码和数据分离,避免代码冗余
2)、不写重复的代码逻辑;
在python解释器中需要安装 ddt 这个包才能用:
要检查是否安装上,在cmd当中 输入 pip list命名,有ddt说明安装成功。
语法:
1、使用数据驱动,要在class前加上修饰器 @ddt
说明:方法里面使用 print ,为了方便,模拟测试用例,主要是为了学习数据驱动,实际中方法里面写的是测试用例的代码:
import unittest
from ddt import ddt, data
@ddt
class TestDemo(unittest.TestCase):
# 单一参数
@data('17611110000', '17611112222')
def test_1(self, phone):
print('测试一电话号码:', phone)
if __name__ == '__main__':
unittest.main()
else:
pass
1)、结合 selenium 使用 ddt
"""
unittest + selenium
"""
import unittest
from time import sleep
from ddt import ddt, data
from selenium import webdriver
@ddt
class TestBaidu(unittest.TestCase):
def setUp(self) -> None:
self.driver = webdriver.Chrome()
self.driver.get('https://www.sogou.com/')
def tearDown(self) -> None:
sleep(3)
self.driver.quit()
# 单一参数
@data('易烊千玺', '王嘉尔')
def test_01(self, name):
self.driver.find_element_by_id('query').send_keys(name)
self.driver.find_element_by_id('stb').click()
if __name__ == '__main__':
unittest.main()
self:相当于java中的this,当前对象的引用,self.driver定义了driver这个变量。
2、在实际中不可能是单一参数进行传参,将会使用多个参数进行传参:
注意事项:
1)、多个数据传参的时候@data里面是要用列表形式
2)、会用到 @unpack 装饰器 进行拆包,把对应的内容传入对应的参数;
import unittest
from ddt import ddt, data, unpack
@ddt
class TestDemo(unittest.TestCase):
# 多参数数据驱动
@data(['admin', '123456'])
# unpack 是进行拆包,不然会把列表里面的数据全部传到username这个一个参数,我们要实现列表中的两个数据分别传入对应的变量中
@unpack
def test_2(self, username, password):
print('测试二:', username, password)
if __name__ == '__main__':
unittest.main()
else:
pass
但是以上步骤都是数据在代码当中的,假如要测试n个手机号这样的数据,全部写在 @data 装饰器里面就很麻烦,这就引出了数据驱动里面的代码和数据的分离。
3、将数据放入一个文本文件中,从文件读取数据, 如JSON、 excel、 xml、 txt等格式文件 ,这里演示的是json文件类型。
json文件处理, 这个链接介绍了json文件和Python文件基本操作
(1)、在json文件驱动
[
{
"username": "admin",
"password": "123456"
},
{
"username": "normal",
"password": "45678"
}
]
(2)、在测试代码中读取json文件
import json
import unittest
from ddt import ddt, data, unpack
# 用json多个参数读取
def reads_phone():
with open('user.json', encoding='utf-8') as f:
result = json.load(f) # 列表
return result
@ddt
class TestDemo(unittest.TestCase):
# 多参数数据驱动
@data(*reads_phone())
# unpack 是进行拆包,不然会把列表里面的数据全部传到username这个一个参数,我们要实现列表中的两个数据分别传入对应的变量中
@unpack
def test_2(self, username, password):
print('测试二:', username, password)
if __name__ == '__main__':
unittest.main()
else:
pass
注意事项:
1、with open里面默认是 ”r“
2、@data 里面的 * 含义是实现每个json对象单个传入方法执行,不然会吧json文件里面所用数据全部传入
> * 是元祖;
> ** 是字典;
3、参数不能传错,要对应
执行结果:
(3)、txt文件驱动
一行表示一组:
admin,123456
normal,456789
import unittest
def read():
lis = []
with open('readtext.txt', 'r', encoding='utf-8') as f:
for line in f.readlines():
# lis.append(line) # ['admin,123456\n', 'normal,456789\n']
# lis.append(line.strip('\n')) ['admin,123456', 'normal,456789'] 两个字符串
lis.append(line.strip('\n').split(',')) # [['admin', '123456'], ['normal', '456789']]
return lis
class TestDome(unittest.TestCase):
def test_01(self):
li = read()
print(li)
if __name__ == '__main__':
unittest.main()
"""
split():一个字符串里面用某个字符分割,返回列表
strip():去掉两边的字符或者字符串,默认删除空白符(包括'\n', '\r', '\t', ' ')
"""
(4)、csv 文件驱动
写法一:
"""
编写 csvv.py脚本读取csv中的测试数据
"""
import csv
class ReadCsv():
def read_csv(self):
lis = []
# 用csv的API的reader方法!!!!
data = csv.reader(open('testdata.csv', 'r')) #!!!!
next(data, None)
for line in data:
lis.append(line)
# lis.append(line[0]) # 二维数组可以省略行,列不可以省略
# lis.append(line[1])
return lis
# 实例化类
readCsv = ReadCsv()
# 打印类中的方法
print(readCsv.read_csv())
写法二:推荐
def csvTest():
li = []
with open('user.csv', 'r', encoding='utf-8') as f:
filename = csv.reader(f)
next(filename, None)
for r in filename:
li.append(r)
return li
(5) 、yaml文件驱动
-
username: admin9
password: 123456
-
username: normal
password: 789456
对应的json文件
[
{
"username": "admin9",
"password": 123456
},
{
"username": "normal",
"password": 7894
}
]
写法:
"""
使用yaml数据驱动
"""
import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt, data, unpack, file_data
@ddt
class YamlTest(unittest.TestCase):
def setUp(self) -> None:
self.driver = webdriver.Chrome()
self.driver.get('file:///D:/%E6%A1%8C%E9%9D%A2/page/%E6%B3%A8%E5%86%8CA.html')
self.driver.maximize_window()
def tearDown(self) -> None:
driver = self.driver
sleep(3)
driver.quit()
# file_data 传入多个参数的时候,@unpack 的解包不起作用
@unittest.skip
@file_data('../user.yaml')
@unpack
def test_yaml01(self, username, password):
driver = self.driver
driver.find_element_by_id('userA').send_keys(username)
driver.find_element_by_id('passwordA').send_keys(password)
# 注意:传的参数名称要与yaml文件对应
# 在yaml数据中文件中采用对象(键值对)的方式来定义数据内容
@file_data('../user1.yaml')
def test_yaml02(self, username, password):
driver = self.driver
driver.find_element_by_id('userA').send_keys(username)
driver.find_element_by_id('passwordA').send_keys(password)
if __name__ == '__main__':
unittest.main()
注意:file_date 装饰器,可以直接读取yaml和json文件
(5)、Excel文件驱动
建立excel表的时候需要退出pychram在根目录下创建excel表保存,否则会报错
def read_excel():
xlsx = openpyxl.load_workbook("../excel.xlsx")
sheet1 = xlsx['Sheet1']
print(sheet1.max_row) # 行
print(sheet1.max_column) # 列
print('=======================================================')
allList = []
for row in range(2, sheet1.max_row + 1):
rowlist = []
for column in range(1, sheet1.max_column + 1):
rowlist.append(sheet1.cell(row, column).value)
allList.append(rowlist)
return allList
用excel登录csdn操作:
"""
测试excel数据驱动
"""
import unittest
from time import sleep
import openpyxl as openpyxl
from ddt import ddt, data, unpack
from selenium import webdriver
# 读取excel表中的数据,使用xlrd,openpyxl
def read_excel():
xlsx = openpyxl.load_workbook("../excel.xlsx")
sheet1 = xlsx['Sheet1']
print(sheet1.max_row) # 行
print(sheet1.max_column) # 列
print('=======================================================')
allList = []
for row in range(2, sheet1.max_row + 1):
rowlist = []
for column in range(1, sheet1.max_column + 1):
rowlist.append(sheet1.cell(row, column).value)
allList.append(rowlist)
return allList
@ddt
class ExcelText(unittest.TestCase):
def setUp(self) -> None:
self.driver = webdriver.Chrome()
self.driver.get('https://passport.csdn.net/login?code=applets')
self.driver.maximize_window()
def tearDown(self) -> None:
driver = self.driver
sleep(3)
driver.quit()
@data(*read_excel())
@unpack
def test_excel01(self, flag, username, password):
print(flag, username, password)
driver = self.driver
sleep(2)
driver.find_element_by_xpath('/html/body/div[2]/div/div[2]/div[2]/div[1]/div/div[1]/span[4]').click()
driver.find_element_by_xpath('/html/body/div[2]/div/div[2]/div[2]/div[1]/div/div[2]/div/div[1]/div/input').send_keys(username)
driver.find_element_by_xpath('/html/body/div[2]/div/div[2]/div[2]/div[1]/div/div[2]/div/div[2]/div/input').send_keys(password)
driver.find_element_by_xpath('/html/body/div[2]/div/div[2]/div[2]/div[1]/div/div[2]/div/div[4]/button').click()
if __name__ == '__main__':
unittest.main()
本文内容不用于商业目的,如涉及知识产权问题,请权利人联系51Testing小编(021-64471599-8017),我们将立即处理