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.driver

        login(driver,username,password)    #调用登录函数


        time.sleep(3)

        #加入断言  //*[@id="header_container"]/div[1]/div[2]/div

        # 验证有显示：Products

        UIResult = driver.find_element_by_xpath('//*[@id="header_container"]/div[2]/span').text

        self.assertEqual("Products", UIResult, "密码正确登录验证失败，fail")



    def tearDown(self):  #浏览器退出

        self.driver.quit()

        print("登录测试结束")

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

1）先将Excel数据准备好

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

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

#coding=utf-8

import 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的框架，每个案例的标准形式

@ddt

class 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.driver

        print("此次测试的用户名%s" % dic.get("username"))

        login(driver,dic.get("username"),dic.get("password"))    #调用登录函数,这里修改为调用Excel中数据

        time.sleep(3)

        #加入断言  //*[@id="header_container"]/div[1]/div[2]/div

        # 验证有显示：Products

        UIResult = driver.find_element_by_xpath('//*[@id="header_container"]/div[2]/span').text

        self.assertEqual("Products", UIResult, "密码正确登录验证失败，fail")

执行结果如下：

今天就先学习到这里吧。后面我们再做优化。

每天进步一点点，加油！