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个字符

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

(五) 示例

 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()

秒客网

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

相关文章