[ddt02篇]十年测试老鸟帮您解析:ddt结合txt,excel,csv,mysql实现自动化测试数据驱动
一、前言:
阅读此文之前请先阅读:
[ddt01篇]十年测试老鸟帮您解析:ddt数据驱动入门基础应用:https://www.cnblogs.com/csmashang/p/12679448.html
二、ddt数据驱动框架结合txt文件实现数据驱动
test_demo.py代码如下:
import unittest
from ddt import ddt, data, unpack
#读取txt文件中的内容,strip()方法去掉首位的指定字符。
def read_txt():
list = []
file = open('param.txt','r',encoding='utf-8')
for line in file.readlines():
list.append(line.strip('\n').split(','))
return list
@ddt
class aaa(unittest.TestCase):
def setUp(self):
pass
@data(*read_txt())
def test1_data(self,args1):
print(args1)
def tearDown(self):
pass
if __name__ == '__main__':
unittest.main()![点击并拖拽以移动](data:image/gif;base64,R0lGODlhAQABAPABAP///wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==)
param.txt文件中内容如下:
诸葛,司马
马超,关羽
三、使用ddt结合excel(.csv格式也可以)文件实现数据驱动
test_demo.py代码如下:
import time
import unittest
import ddt
from selenium import webdriver
from selenium.webdriver.common.by import By
from a.excel_util import ExcelUtil
@ddt.ddt
class TestDemo(unittest.TestCase):
eu = ExcelUtil()
def setUp(self) -> None:
self.driver = webdriver.Chrome()
self.driver.get("https://www.baidu.com")
self.driver.maximize_window()
self.driver.implicitly_wait(10)
@ddt.data(*eu.read_excel("D:\\testdata.xlsx","Sheet1"))
def test_data_driver_by_jsonfile(self,value):
try:
testData,expectData = tuple(value)
self.driver.find_element(By.ID,"kw").send_keys(testData)
time.sleep(3)
self.driver.find_element(By.ID,"su").click()
time.sleep(5)
self.assertTrue(expectData in self.driver.page_source)
except:
print("搜索:%s,期望:%s,失败"%(testData,expectData))
pass
else:
print("搜索:%s,期望:%s,通过"%(testData,expectData))
pass
def tearDown(self) -> None:
self.driver.quit()
if __name__ == '__main__':
unittest.main()
excel_util.py读取excel(csv)数据文件如下:
import xlrd
class ExcelUtil:
#读取excel数据
def read_excel(self,excel_path,sheet_name):
xls = xlrd.open_workbook(excel_path)
sheet = xls.sheet_by_name(sheet_name)
dataList=[]
for line in range(1,sheet.nrows):
tempList = []
tempList.append(sheet.cell_value(line,1))
tempList.append(sheet.cell_value(line,2))
dataList.append(tempList)
return dataList
if __name__ == '__main__':
data = ExcelUtil().read_excel("d:\\testdata.xlsx","Sheet1")
print(data)![点击并拖拽以移动](data:image/gif;base64,R0lGODlhAQABAPABAP///wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==)
D:\\testdata.xlsx文件中Sheet1的工作表内容如下:
实例讲解:@ddt.data()从eu.read_excel("D:\\testdata.xlsx","Sheet1")方法中接收一个可迭代的数组对象,以此来判断需要执行的次数。如果@ddt.data()括号中传的是一个方法,方法前需要加星号(*)修饰。*表示出掉最外层的括号。
四、使用ddt结合mysql数据库进行数据驱动
1.在dos窗口安装pymysql模块来操作数据库。命令:pip install pymysql
2.安装mysql数据库(安装过程省略),创建database.py文件,使用pymysql创建数据库datadriver,创建表testdata,并向testdata表中插入三条数据。
database.py代码如下:
import time
import pymysql as pymysql
class DataBase:
#删除datadriver数据库的SQL语句
drop_database = " drop database if exists datadriver;"
#创建datadriver的SQL语句:
create_database = "create database datadriver default charset utf8 COLLATE utf8_general_ci;"
#创建testdata表的SQL语句:
create_table = """
create table testdata(
id int not null auto_increment comment '主键',
searchContent varchar(40) unique not null comment '百度搜索的内容',
assertContent varchar(30) not null comment '断言的内容',
primary key(id)
);
"""
def __init__(self,host,port,username,password,db):
self.host = host
self.port = port
self.username = username
self.password = password
self.db = db
#连接数据库创建数据库和表
def create_databse_and_table(self):
try:
#连接Mysql数据库。
conn = pymysql.connect(host=self.host,port=self.port,user=self.username,password=self.password)
#获取数据库游标
cur = conn.cursor()
#删除数据库
cur.execute(self.drop_database)
time.sleep(3)
#执行创建数据库的SQL语句
cur.execute(self.create_database)
time.sleep(3)
#选中创建好的datadriver数据库。
conn.select_db("datadriver")
#执行创建表的SQL语句
cur.execute(self.create_table)
# 提交操作
conn.commit()
except Exception:
raise
else:
cur.close() #关闭游标
conn.close() #关闭连接
print("创建数据库和表成功!")
#插入数据
def insert_data(self):
try:
# 连接Mysql数据库。
conn = pymysql.connect(host=self.host, port=self.port, user=self.username, password=self.password,db=self.db)
# 获得数据库游标
cur = conn.cursor()
# 执行插入数据的SQL语句
sql = "insert into testdata(searchContent,assertContent) values(%s,%s)"
cur.executemany(sql,[('神奇动物在哪里','叶茨'),('疯狂动物城','霍华德'),('杜兰特','凯文')])
# 提交操作
conn.commit()
except Exception:
raise
else:
cur.close() # 关闭游标
conn.close() # 关闭连接
print("数据插入成功!")
#查询数据
def select_data(self):
result = ""
try:
conn = pymysql.connect(host=self.host,port=self.port,user=self.username,password=self.password,db=self.db)
cur = conn.cursor()
cur.execute("select searchContent,assertContent from testdata;")
result=cur.fetchall()
except Exception:
raise
else:
cur.close() # 关闭游标
conn.close() # 关闭连接
print("数据查询成功!")
return result
if __name__ == '__main__':
db = DataBase("localhost",3306,"root","admin","datadriver")
db.create_databse_and_table()
db.insert_data()
print(db.select_data())
# 以下代码连接Mysql数据库。最好使用关键字传参,否则可能会出现参数顺序问题,关键字包括:host,port,user,password,db
conn = pymysql.connect(host=self.host, port=self.port, user=self.username, password=self.password,db="datadriver")
#以下代码在传值时3306必须传数字类型,不能传入"3306"这种字符串类型。
db = DataBase("localhost",3306,"root","admin")
3.使用ddt结合mysql数据库进行数据驱动
test_demo.py文件内容如下:
import time import unittest import warnings import ddt from selenium import webdriver from selenium.webdriver.common.by import By from a.database import DataBase def get_data(): db = DataBase("localhost", 3306, "root", "admin","datadriver") data = db.select_data() return data @ddt.ddt class TestDemo(unittest.TestCase): def setUp(self) -> None: warnings.simplefilter('ignore',ResourceWarning) #处理资源警告。 self.driver = webdriver.Chrome() self.driver.get("https://www.baidu.com") self.driver.maximize_window() self.driver.implicitly_wait(10) @ddt.data(*get_data()) def test_data_driver_by_jsonfile(self,data): try: testData,expectData = data self.driver.find_element(By.ID,"kw").send_keys(testData) time.sleep(3) self.driver.find_element(By.ID,"su").click() time.sleep(5) self.assertTrue(expectData in self.driver.page_source) except: print("搜索:%s,期望:%s,失败"%(testData,expectData)) pass else: print("搜索:%s,期望:%s,通过"%(testData,expectData)) pass def tearDown(self) -> None: self.driver.quit() if __name__ == '__main__': unittest.main()
五、使用ddt中读取json/yaml文件实现数据驱动。
因内容太多,这里就不在详述了,如果你觉得此文对你有帮助,如果你想要获取[ddt中读取json/yaml文件实现数据驱动]的资料,如果你对软件测试、接口测试、自动化测试、
面试经验交流感兴趣欢迎加入:软件测试技术群:695458161,群里发放的免费资料都是笔者十多年测试生涯的精华。还有同行大神一起交流技术哦。
作者:来自公众号:软测之家
出处:https://www.cnblogs.com/csmashang/p/12679477.html
原创不易,欢迎转载,但未经作者同意请保留此段声明,并在文章页面明显位置给出原文链接。