Python Selenium 之数据驱动测试的实现


Posted in Python onAugust 01, 2019

数据驱动模式的测试好处相比普通模式的测试就显而易见了吧!使用数据驱动的模式,可以根据业务分解测试数据,只需定义变量,使用外部或者自定义的数据使其参数化,从而避免了使用之前测试脚本中固定的数据。可以将测试脚本与测试数据分离,使得测试脚本在不同数据集合下高度复用。不仅可以增加复杂条件场景的测试覆盖,还可以极大减少测试脚本的编写与维护工作。

下面将使用Python下的数据驱动模式(ddt)库,结合unittest库以数据驱动模式创建百度搜索的测试。

ddt库包含一组类和方法用于实现数据驱动测试。可以将测试中的变量进行参数化。

可以通过python自带的pip命令进行下载并安装:pip install ddt . 更多关于ddt的信息可以参考:

https://pypi.org/project/ddt/

一个简单的数据驱动测试

为了创建数据驱动测试,需要在测试类上使用@ddt装饰符,在测试方法上使用@data装饰符。@data装饰符把参数当作测试数据,参数可以是单个值、列表、元组、字典。对于列表,需要用@unpack装饰符把元组和列表解析成多个参数。

下面实现百度搜索测试,传入搜索关键词和期望结果,代码如下:

import unittest
from selenium import webdriver
from ddt import ddt, data, unpack

@ddt
class SearchDDT(unittest.TestCase):
  '''docstring for SearchDDT'''
  def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.implicitly_wait(30)
    self.driver.maximize_window()
    self.driver.get("https://www.baidu.com")

  # specify test data using @data decorator
  @data(('python', 'PyPI'))
  @unpack
  def test_search(self, search_value, expected_result):
    search_text = self.driver.find_element_by_id('kw')
    search_text.clear()
    search_text.send_keys(search_value)

    search_button = self.driver.find_element_by_id('su')
    search_button.click()

    tag = self.driver.find_element_by_link_text("PyPI").text
    self.assertEqual(expected_result, tag)

  def tearDown(self):
    self.driver.quit()

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

在test_search()方法中,search_value与expected_result两个参数用来接收元组解析的数据。当运行脚本时,ddt把测试数据转换为有效的python标识符,生成名称为更有意义的测试方法。结果如下:

Python Selenium 之数据驱动测试的实现

使用外部数据的数据驱动测试

如果外部已经存在了需要的测试数据,如一个文本文件、电子表格或者数据库,那也可以用ddt来直接获取数据并传入测试方法进行测试。

下面将借助外部的CSV(逗号分隔值)文件和EXCLE表格数据来实现ddt。

通过CSV获取数据

同上在@data装饰符使用解析外部的CSV(testdata.csv)来作为测试数据(代替之前的测试数据)。其中数据如下:

Python Selenium 之数据驱动测试的实现

接下来,先要创建一个get_data()方法,其中包括路径(这里默认使用当前路径)、CSV文件名。调用CSV库去读取文件并返回一行数据。再使用@ddt及@data实现外部数据驱动测试百度搜索,代码如下:

import csv, unittest
from selenium import webdriver
from ddt import ddt, data, unpack

def get_data(file_name):
  # create an empty list to store rows
  rows = []
  # open the CSV file
  data_file = open(file_name, "r")
  # create a CSV Reader from CSV file
  reader = csv.reader(data_file)
  # skip the headers
  next(reader, None)
  # add rows from reader to list
  for row in reader:
    rows.append(row)
  return rows

@ddt
class SearchCSVDDT(unittest.TestCase):
  def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.implicitly_wait(30)
    self.driver.maximize_window()
    self.driver.get("https://www.baidu.com")

  # get test data from specified csv file by using the get_data funcion
  @data(*get_data('testdata.csv'))
  @unpack
  def test_search(self, search_value, expected_result):
    search_text = self.driver.find_element_by_id('kw')
    search_text.clear()
    search_text.send_keys(search_value)

    search_button = self.driver.find_element_by_id('su')
    search_button.click()

    tag = self.driver.find_element_by_link_text("PyPI").text
    self.assertEqual(expected_result, tag)

  def tearDown(self):
    self.driver.quit()

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

测试执行时,@data将调用get_data()方法读取外部数据文件,并将数据逐行返回给@data。执行的结果也同上~

通过Excel获取数据

测试中经常用Excle存放测试数据,同上在也可以使用@data装饰符来解析外部的CSV(testdata.csv)来作为测试数据(代替之前的测试数据)。其中数据如下:

Python Selenium 之数据驱动测试的实现

接下来,先要创建一个get_data()方法,其中包括路径(这里默认使用当前路径)、EXCEL文件名。调用xlrd库去读取文件并返回数据。再使用@ddt及@data实现外部数据驱动测试百度搜索,代码如下:

import xlrd, unittest
from selenium import webdriver
from ddt import ddt, data, unpack

def get_data(file_name):
  # create an empty list to store rows
  rows = []
  # open the CSV file
  book = xlrd.open_workbook(file_name)
  # get the frist sheet
  sheet = book.sheet_by_index(0)
  # iterate through the sheet and get data from rows in list
  for row_idx in range(1, sheet.nrows): #iterate 1 to maxrows
    rows.append(list(sheet.row_values(row_idx, 0, sheet.ncols)))
  return rows

@ddt
class SearchEXCLEDDT(unittest.TestCase):
  def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.implicitly_wait(30)
    self.driver.maximize_window()
    self.driver.get("https://www.baidu.com")

  # get test data from specified excle spreadsheet by using the get_data funcion
  @data(*get_data('TestData.xlsx'))
  @unpack
  def test_search(self, search_value, expected_result):
    search_text = self.driver.find_element_by_id('kw')
    search_text.clear()
    search_text.send_keys(search_value)

    search_button = self.driver.find_element_by_id('su')
    search_button.click()

    tag = self.driver.find_element_by_link_text("PyPI").text
    self.assertEqual(expected_result, tag)

  def tearDown(self):
    self.driver.quit()

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

与上面读取CVS文件一样,测试执行时,@data将调用get_data()方法读取外部数据文件,并将数据逐行返回给@data。执行的结果也同上~

如果想从数据库的库表中获取数据,同样也需要一个get_data()方法,并且通过DB相关的库来连接数据库、SQL查询来获取测试数据。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用mysqldb连接数据库操作方法示例详解
Dec 03 Python
Python读写Redis数据库操作示例
Mar 18 Python
Python在Console下显示文本进度条的方法
Feb 14 Python
Python基于QRCode实现生成二维码的方法【下载,安装,调用等】
Jul 11 Python
完美解决安装完tensorflow后pip无法使用的问题
Jun 11 Python
实例分析python3实现并发访问水平切分表
Sep 29 Python
python实现flappy bird游戏
Dec 24 Python
Python动态赋值的陷阱知识点总结
Mar 17 Python
为什么python比较流行
Jun 19 Python
详解scrapy内置中间件的顺序
Sep 28 Python
python爬虫请求头的使用
Dec 01 Python
python基于openpyxl生成excel文件
Dec 23 Python
Python 一键获取百度网盘提取码的方法
Aug 01 #Python
Django中的静态文件管理过程解析
Aug 01 #Python
pycharm 批量修改变量名称的方法
Aug 01 #Python
python腾讯语音合成实现过程解析
Aug 01 #Python
Python空间数据处理之GDAL读写遥感图像
Aug 01 #Python
Python生命游戏实现原理及过程解析(附源代码)
Aug 01 #Python
Python中list的交、并、差集获取方法示例
Aug 01 #Python
You might like
phpcms模块开发之swfupload的使用介绍
2013/04/28 PHP
PHP网页游戏学习之Xnova(ogame)源码解读(十三)
2014/06/26 PHP
PHP处理会话函数大总结
2015/08/05 PHP
PHP获取访问页面HTTP状态码的实现代码
2016/11/03 PHP
JavaScript 检测浏览器和操作系统的脚本
2008/12/26 Javascript
jQuery EasyUI API 中文文档 搜索框
2011/09/29 Javascript
jquery通过a标签删除table中的一行的代码
2013/12/02 Javascript
jquery鼠标停止移动事件
2013/12/21 Javascript
jQuery的animate函数学习记录
2014/08/08 Javascript
node.js实现BigPipe详解
2014/12/05 Javascript
Javascript核心读书有感之表达式和运算符
2015/02/11 Javascript
JavaScript通过join函数连接数组里所有元素的方法
2015/03/20 Javascript
javascript html5 canvas实现可拖动省份的中国地图
2016/03/11 Javascript
javascript宿主对象之window.navigator详解
2016/09/07 Javascript
node.js路径处理方法以及绝对路径详解
2021/03/04 Javascript
求js数组的最大值和最小值的四种方法
2017/03/03 Javascript
解决vue build打包之后首页白屏的问题
2018/03/06 Javascript
JavaScript实现仿Clock ISO时钟
2018/06/29 Javascript
Vue.js项目实战之多语种网站的功能实现(租车)
2019/08/07 Javascript
Python中的模块导入和读取键盘输入的方法
2015/10/16 Python
Python中使用支持向量机(SVM)算法
2017/12/26 Python
解决pyttsx3无法封装的问题
2018/12/24 Python
解决Python plt.savefig 保存图片时一片空白的问题
2019/01/10 Python
详解Python locals()的陷阱
2019/03/26 Python
Python vtk读取并显示dicom文件示例
2020/01/13 Python
荷兰包包购物网站:The Little Green Bag
2018/03/17 全球购物
.net C#面试题
2012/08/28 面试题
工程师求职简历的自我评价分享
2013/10/10 职场文书
机械电子工程专业推荐信范文
2013/11/20 职场文书
工程力学专业自荐信范文
2014/03/17 职场文书
数学教育专业求职信
2014/07/22 职场文书
2014物价局群众路线对照检查材料思想汇报
2014/09/21 职场文书
先进典型事迹材料
2014/12/29 职场文书
2015年敬老院工作总结
2015/05/18 职场文书
后天观后感
2015/06/08 职场文书
2015年教师节新闻稿
2015/07/17 职场文书