Python+Selenium笔记（十二）：数据驱动测试

(一)   前言

通过使用数据驱动测试，实现对输入值和预期结果的参数化。（例如：输入数据和预期结果可以直接读取Excel文档的数据）

(二)   ddt

使用ddt执行数据驱动测试，ddt库可以将测试中的变量参数化。使用ddt的时候，在测试类上使用@ddt装饰符，在测试方法上使用@data装饰符。@data装饰符将参数当作测试数据，参数可以是单个值、列表、元组、字典。对于列表和元组，需要用@unpack装饰符把列表和元组解析成多个参数。

使用下面的命令安装ddt库

pip install ddt

(三)   通过Excel获取数据

读取Excel文件，需要用到xlrd库。

安装xlrd的库

pip install xlrd

如果要往Excel表格写数据，需要用到xlwt库

pip install xlwt

(四)   示例中用到的excel文件

邮箱	手机	登陆名称	昵称	密码	确认密码	预期结果
test@outlook.com	1	bky_110	盘古	test>100	test>100	手机号码有误
test@outlook.com	18898989878	b	盘古	test>100	test>100	不合要求，至少2个字符，最多30个字符

(五)   示例

 from selenium import webdriver
 from ddt import ddt,data,unpack
 import xlrd
 import unittest
 #读取excel文件的函数
 def get_data(file_name):
     rows = []
     #读取excel的数据
     book = xlrd.open_workbook(file_name)
     #通过索引访问第一个sheet页
     sheet = book.sheet_by_index(0)
     #迭代读取excel第一个sheet页的数据，sheet.nrows指excel的行数
     for r_idx in range(1, sheet.nrows):
         #row_values读取第r_idx行的数据（0代表读取第1列及后面所有列的数据）
         #读取数据的时候，我们一般说的第一行、第一列，索引都是0
         #所以r_idx=1的时候，读取的其实是excel第二行的数据
         rows.append(list(sheet.row_values(r_idx,0)))
         #先将手机号删除并赋值给pthone,然后转换为字符串并添加回原来的位置
         pthone = rows[r_idx - 1].pop(1)
         rows[r_idx - 1].insert(1, str(int(pthone)))
     return rows
 @ddt
 class RegisterNewUserDDT(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.driver = webdriver.Chrome()
         cls.driver.implicitly_wait(20)
         cls.driver.maximize_window()
         cls.driver.get('https://www.cnblogs.com/')
         login_area = cls.driver.find_element_by_css_selector('#login_area')
         register = login_area.find_element_by_link_text('注册')
         register.click()
     #读取excel文件的数据作为参数
     @data(*get_data('data/reTest.xlsx'))
     @unpack
     def test_register_new_user(self,email,phone,login_name,nickname,password,confirm_password,expected_result):
         driver = self.driver
         self.assertTrue('用户注册 - 博客园' == driver.title)
         # 定位注册页面各个字段
         user_email = driver.find_element_by_id('Email')
         user_phone_country = driver.find_element_by_id('CountryCode')
         user_phone = driver.find_element_by_id('PhoneNum')
         user_login_name = driver.find_element_by_id('LoginName')
         user_nickname = driver.find_element_by_id('DisplayName')
         user_password = driver.find_element_by_id('Password')
         user_confirm_password = driver.find_element_by_id('ConfirmPassword')
         #清除各字段的值（如果有）
         user_email.clear()
         user_phone.clear()
         user_login_name.clear()
         user_nickname.clear()
         user_password.clear()
         user_confirm_password.clear()
         #输入邮箱、手机号等信息
         user_email.send_keys(email)
         user_phone.send_keys(phone)
         user_login_name.send_keys(login_name)
         user_nickname.send_keys(nickname)
         user_password.send_keys(password)
         user_confirm_password.send_keys(confirm_password)
         #判断提示是否正确（这边应该有办法可以获取当前是第几次执行，下面的写法太死板了）
         if phone == '':
             phone_error = driver.find_element_by_id('PhoneNum-error')
             self.assertTrue(phone_error.text == expected_result)
         elif login_name == 'b':
             loginName_error = driver.find_element_by_id('LoginName-error')
             self.assertTrue(loginName_error.text == expected_result)

     @classmethod
     def tearDownClass(cls):
         cls.driver.quit()