Python+Selenium笔记：数据驱动测试

Posted 2020-10-25 习惯形成性格，性格决定命运

(一)   前言

通过使用数据驱动测试，实现对输入值和预期结果的参数化。（例如：输入数据和预期结果可以直接读取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文件

邮箱	手机	登陆名称	昵称	密码	确认密码	预期结果
[e-mail]	1	bky_110	盘古	test>100	test>100	手机号码有误
[e-mail]	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()