(一) 前言
通过使用数据驱动测试,实现对输入值和预期结果的参数化。(例如:输入数据和预期结果可以直接读取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个字符 |
(五) 示例
1 from selenium import webdriver
2 from ddt import ddt,data,unpack
3 import xlrd
4 import unittest
5 #读取excel文件的函数
6 def get_data(file_name):
7 rows = []
8 #读取excel的数据
9 book = xlrd.open_workbook(file_name)
10 #通过索引访问第一个sheet页
11 sheet = book.sheet_by_index(0)
12 #迭代读取excel第一个sheet页的数据,sheet.nrows指excel的行数
13 for r_idx in range(1, sheet.nrows):
14 #row_values读取第r_idx行的数据(0代表读取第1列及后面所有列的数据)
15 #读取数据的时候,我们一般说的第一行、第一列,索引都是0
16 #所以r_idx=1的时候,读取的其实是excel第二行的数据
17 rows.append(list(sheet.row_values(r_idx,0)))
18 #先将手机号删除并赋值给pthone,然后转换为字符串并添加回原来的位置
19 pthone = rows[r_idx - 1].pop(1)
20 rows[r_idx - 1].insert(1, str(int(pthone)))
21 return rows
22 @ddt
23 class RegisterNewUserDDT(unittest.TestCase):
24 @classmethod
25 def setUpClass(cls):
26 cls.driver = webdriver.Chrome()
27 cls.driver.implicitly_wait(20)
28 cls.driver.maximize_window()
29 cls.driver.get('https://www.cnblogs.com/')
30 login_area = cls.driver.find_element_by_css_selector('#login_area')
31 register = login_area.find_element_by_link_text('注册')
32 register.click()
33 #读取excel文件的数据作为参数
34 @data(*get_data('data/reTest.xlsx'))
35 @unpack
36 def test_register_new_user(self,email,phone,login_name,nickname,password,confirm_password,expected_result):
37 driver = self.driver
38 self.assertTrue('用户注册 - 博客园' == driver.title)
39 # 定位注册页面各个字段
40 user_email = driver.find_element_by_id('Email')
41 user_phone_country = driver.find_element_by_id('CountryCode')
42 user_phone = driver.find_element_by_id('PhoneNum')
43 user_login_name = driver.find_element_by_id('LoginName')
44 user_nickname = driver.find_element_by_id('DisplayName')
45 user_password = driver.find_element_by_id('Password')
46 user_confirm_password = driver.find_element_by_id('ConfirmPassword')
47 #清除各字段的值(如果有)
48 user_email.clear()
49 user_phone.clear()
50 user_login_name.clear()
51 user_nickname.clear()
52 user_password.clear()
53 user_confirm_password.clear()
54 #输入邮箱、手机号等信息
55 user_email.send_keys(email)
56 user_phone.send_keys(phone)
57 user_login_name.send_keys(login_name)
58 user_nickname.send_keys(nickname)
59 user_password.send_keys(password)
60 user_confirm_password.send_keys(confirm_password)
61 #判断提示是否正确(这边应该有办法可以获取当前是第几次执行,下面的写法太死板了)
62 if phone == '1':
63 phone_error = driver.find_element_by_id('PhoneNum-error')
64 self.assertTrue(phone_error.text == expected_result)
65 elif login_name == 'b':
66 loginName_error = driver.find_element_by_id('LoginName-error')
67 self.assertTrue(loginName_error.text == expected_result)
68
69 @classmethod
70 def tearDownClass(cls):
71 cls.driver.quit()