Selenium + Python 自动化测试22（PO+数据驱动）

本文主要是介绍Selenium + Python 自动化测试22（PO+数据驱动），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

我们的目标是：按照这一套资料学习下来，大家可以独立完成自动化测试的任务。

上一篇我们讨论了PO模式和unittest框架结合起来使用。

本篇文章我们综合一下之前学习的内容，如先将PO模式、数据驱动思想和我们生成HTML报告融合起来，综合的灵活的使用之前学习的内容。

1、先回忆一下之前的脚本：

如下图所示：我们有四个脚本文件：page_object、login_object、测试用例以及执行入口文件。

我们这次加入数据驱动的思想，主要改测试用例层就可以了。这样我们的代码就可以大致分为：基础页面层、数据层、业务页面层、测试用例层。使用分层的思想开始看起来有点麻烦，但是正是设计好了以后，后面使用起来相当方便、灵活。

2、测试用例层分离出数据模块

开始我们的测试用例脚本如下：

def login(driver,username,password):#登录的函数，基本是按流程login_page=LoginPage(driver)login_page.open()login_page.enter_username(username)login_page.enter_password(password)login_page.submit()#使用unittest的框架，每个案例的标准形式class LoginTest(unittest.TestCase):def setUp(self):self.driver=webdriver.Chrome("C:\\Users\\dewi\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe")print("登录测试开始")#输入正确的密码登录成功测试def test_login_Y(self):username = "standard_user"password = "secret_sauce"print("登录的用户名是：%s，密码是：%s"%(username,password))driver=self.driverlogin(driver,username,password)    #调用登录函数time.sleep(3)#加入断言  //*[@id="header_container"]/div[1]/div[2]/div# 验证有显示：ProductsUIResult = driver.find_element_by_xpath('//*[@id="header_container"]/div[2]/span').textself.assertEqual("Products", UIResult, "密码正确登录验证失败，fail")def tearDown(self):  #浏览器退出self.driver.quit()print("登录测试结束")

加入ddt模式，就是黄色部分修改：

1）先将Excel数据准备好

数据名称login_data.xlsx,放到本项目下面

2）参考之前编写获取Excel数据的方法：

#coding=utf-8import xlrd#通过Excel来获取数据,希望返回字典组成的列表：[{"username":" standard_user","password":"secret_sauce"},# {"username":" visual_user","password":"secret_sauce"},{"username":" error_user","password":"password_NG"}]def get_excel_data(filename,sheetnum):path = 'login_data.xlsx'book_data = xlrd.open_workbook(path) #打开文档book_sheet=book_data.sheet_by_index(0) #打开Excel中第一个表rows_num = book_sheet.nrows  #sheet1 行数,row:行rows0 = book_sheet.row_values(0) #第一行的各个名称作为字典的键rows0_num = len(rows0)   #第一行列表的长度即为数据的列数data_list = [ ]  #存放读取的数据,列表的每一项为一个字典for i in range(1,rows_num):rows_data = book_sheet.row_values(i) #取出每一行的值作为列表print(rows_data)  #看下数据，是列表吧rows_dir ={ }for y in range(0,rows0_num):           #将每一列的值与每一行对应起来rows_dir[rows0[y]] = rows_data[y]    #键值对应data_list.append(rows_dir)               #每个字典插入到列表中print(rows_dir.items())   #看下数据，字典有哪些return data_list

3）改写测试用例脚本

下面所示黄色的部分有所改写，其它只是之前的部分；

主要是读取Excel数据>unittest 框架中测试用例引用Excel数据>用户名和密码引用Excel数据>所有数据循环执行一遍

#读取Excel文件excel_data=get_excel_data("",0)
def login(driver,username,password):#登录的函数，基本是按流程login_page=LoginPage(driver)login_page.open()login_page.max_window()login_page.enter_username(username)login_page.enter_password(password)login_page.submit()#使用unittest的框架，每个案例的标准形式@ddtclass LoginTest(unittest.TestCase):def setUp(self):self.driver=webdriver.Chrome("C:\\Users\\dewi\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe")print("登录测试开始")#输入正确的密码登录成功测试@data(*excel_data)       #数据参数化，使用读取的数据，列表中的数据分别执行一次，每次是一个字典def test_login_Y(self,dic):#username = "standard_user"   #不再使用固定数据#password = "secret_sauce"    #不再使用固定数据#print("登录的用户名是：%s，密码是：%s"%(username,password))driver=self.driverprint("此次测试的用户名%s" % dic.get("username"))login(driver,dic.get("username"),dic.get("password"))    #调用登录函数,这里修改为调用Excel中数据time.sleep(3)#加入断言  //*[@id="header_container"]/div[1]/div[2]/div# 验证有显示：ProductsUIResult = driver.find_element_by_xpath('//*[@id="header_container"]/div[2]/span').textself.assertEqual("Products", UIResult, "密码正确登录验证失败，fail")