数据驱动之Unittest+DDT
一、安装DDT
DDT是Data-Driven Tests的缩写,虽然Unittest没有自带数据驱动功能,但DDT与它可以完美的结合,来完成数据驱动测试。
在确保Python环境已经正常配置的前提下,启动命令行,运行命令:pip install ddt
接下来我们验证DDT是否安装正常,依旧是在命令行,进入Python环境,然后直接import ddt, 如果没有报异常,则说明DDT模块正常可用
二、代码示例
场景一:
借由DDT进行数据驱动测试,实现一组测试代码执行3组测试数据,打开三次浏览器,完成3次163邮箱的登陆测试,在工程中创建DataDrivenByDDTLogin.py,在文件中写入如下代码:
代码解析:
代码头部导入ddt模块(import ddt),测试类DataDrivenDDT前声明使用ddt(@ddt.ddt,测试方法test_dataDrivenByDDT前使用@ddt.data()来定义数据,并且定义的数据的个数和顺序必须与测试方法的形参一一对应,然后使用@unpack进行修饰,也就是对测试数据进行解包,将每组数据的第一个传给username,第二个传给password,并且执行代码后根据我们代码中的log设定,会生成内容到日志文件中
场景二:
借由DDT进行数据驱动测试,实现一组测试代码执行3组测试数据,完成检索并断言期望结果,在工程中创建DataDrivenByDDTSearch.py,在文件中写入如下代码:
# encoding = utf-8
from selenium import webdriver
import unittest
import time
import logging
import traceback
import ddt
from selenium.common.exceptions importNoSuchElementException
"""
初始化日志对象
"""
logging.basicConfig(
level=logging.INFO, #日志级别
format='%(asctime)s %(filename)s[line: %(lineno)d] %(levelname)s%(message)s', #时间、代码所在文件名、代码行号、日志级别名字、日志信息
datefmt='%a, %d %b %Y %H:%M:%S', #打印日志的时间
filename='D:\\Programs\\Python\\PythonUnittest\\Reports\\TestLog\ddt.log', #日志文件存放的目录及日志文件名
filemode='w' #打开日志的方式
)
@ddt.ddt
class DataDrivenDDT(unittest.TestCase):
defsetUp(self):
self.driver = webdriver.Chrome() #定义浏览器驱动
@ddt.data([u"阿里巴巴", u"腾讯"], [u"美团外卖", u"百度"], [u"饿了么", u"蚂蚁金服"])
@ddt.unpack
deftest_dataDrivenByDDT(self, testdata, expectdata):
url ="http://www.baidu.com"
self.driver.get(url)
self.driver.implicitly_wait(30)
try:
self.driver.find_element_by_id("kw").send_keys(testdata)
self.driver.find_element_by_id("su").click()
time.sleep(3)
self.assertTrue(expectdata in self.driver.page_source)
except NoSuchElementException as e:
logging.error(u"查找的页面元素不存在,异常堆栈信息:"+ str(traceback.format_exc()))
except AssertionError as e:
logging.info(u"搜索 '%s',期望 '%s' ,失败" %(testdata, expectdata))
except Exception as e:
logging.error(u"未知错误,错误信息:" +str(traceback.format_exc()))
else:
logging.info(u"搜索 '%s',期望 '%s' ,通过" %(testdata, expectdata))
deftearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main()
代码解析:
代码头部导入ddt模块(import ddt),测试类DataDrivenDDT前声明使用ddt(@ddt.ddt,测试方法test_dataDrivenByDDT前使用@ddt.data()来定义数据,并且定义的数据的个数和顺序必须与测试方法的形参一一对应,然后使用@unpack进行修饰,也就是对测试数据进行解包,将每组数据的第一个传给testdata,第二个传给expectdata,并且执行代码后根据我们代码中的log设定,会生成内容到日志文件中