快速教你如何搭建数据驱动自动化测试框架？

前言

说到数据驱动自动化测试，你会不会有这样的疑问：数据怎么管理？数据怎么才能驱动测试用例执行？到底怎么样才算数据驱动？那么本篇文章就教你如何进行数据驱动测试，相信你一定能对数据驱动自动化测试有一个不一样的理解，希望这篇文章能帮助还在懵懵懂懂的你了解数据驱动测试，如何来做数据驱动测试？那么就跟上我的脚步吧^_^

项目介绍

地址：https://www.126.com/

1.实现126邮箱登录功能的验证

2.实现126邮箱添加新联系人功能的验证

目录结构

项目源码

1.新建config目录存放和业务无关的数据文件，比如PageElementLocator.ini 用来存放我们每个页面的定位元素表达式。

[126mail_login]
loginPage.frame=xpath>//div[@id='loginDiv']/iframe
loginPage.username=xpath>//input[@name='email']
loginPage.password=xpath>//input[@name='password']
loginPage.loginBtn=xpath>//a[@id='dologin']
[126mail_homePage]
homePage.addressbook=id>_mail_tabitem_1_4text
[126mail_addContactPage]
addContactPage.newContact=xpath>//span[text()='新建联系人']
addContactPage.newName=id>input_N
addContactPage.newMail=xpath>//div[@id='iaddress_MAIL_wrap']//input[@class='nui-ipt-input']
addContactPage.newMark=xpath>//span[@class='nui-chk-text']/preceding-sibling::span/b
addContactPage.newPhone=xpath>//div[@id='iaddress_TEL_wrap']//input[@class='nui-ipt-input']
addContactPage.newComment=id>input_DETAIL
addContactPage.newCommit=xpath>//span[text()='确 定']

2.有了上面元素的表达式我们又该如何查找我们的元素呢？这时候我们需要封装一套无论何种定位方式都可以使用的通用的查找方法。新建util目录(util目录主要存放和我们业务无关的公共方法)，此目录下新建ObjectMap.py文件

from selenium.webdriver.support.wait import WebDriverWait

def getElement(driver, by, locate):
    '''
    查找单一个元素
    :param driver: 浏览器驱动
    :param by: 定位方式，id, name, class, xpath...
    :param locate: 定位表达式
    :return: 元素
    '''
    try:
        element = WebDriverWait(driver, 30).until(lambda x :x.find_element(by, locate))
    except Exception as e:
        raise e
    else:
        return element
def getElelments(driver, by, locate):
    '''
    查找一组元素
    :param driver: 浏览器驱动
    :param by: 定位方式
    :param locate: 定位表达式
    :return: 一组元祖组成的列表
    '''
    try:
        elements = WebDriverWait(driver, 30).until(lambda x :x.find_elements(by, locate))
    except Exception as e:
        raise e
    else:
        return elements
if __name__=='__main__':
    from selenium import webdriver
    driver = webdriver.Firefox()
    driver.get('http://www.baidu.com')
    inputBaidu = getElement(driver, 'id', 'kw')
    inputBaidu.send_keys('python')

3.定位表达式有了，查找元素的方法封装好了，下面我们需要解析配置文件，获取配置文件中我们需要的定位方式和定位表达式。同样在util目下新建ParseConfigurationFile.py文件用来解析配置文件

from config.varCondig import pageElementLocatorPath
import configparser

class ParseConfigFile(object):
    '''
    解析ini配置文件
    '''
    def __init__(self):
        try:
            self.cf = configparser.ConfigParser() # 获取配置文件对象
            self.cf.read(pageElementLocatorPath, encoding='utf-8') # 加载配置文件到内存中
        except Exception as e:
            raise e

    def getItemsSection(self, sectionName):
        '''
        获取section下面所有section的键值
        :param sectionName:
        :return:
        '''
        try:
            vlaues = dict(self.cf.items(sectionName))
        except Exception as e:
            raise e
        else:
            return vlaues

    def getElementValue(self, sectionName, optionName):
        try:
            locator = self.cf.get(sectionName, optionName).split('>')
        except Exception as e:
            raise e
        else:
            return locator # 获取option键对应的value

    def getAllSections(self):
        try:
            allsections = self.cf.sections()
        except Exception as e:
            raise e
        else:
            return allsections # 所有的sections返回值是个列表

    def getAllOptions(self, section):
        try:
            options = self.cf.options(section)
        except Exception as e:
            raise e
        else:
            return options # 某个section下面的键

if __name__=='__main__':
    cf = ParseConfigFile()
    locator = cf.getElementValue('126mail_login','loginPage.username')
    # print(locator)
    print(cf.getItemsSection('126mail_login'))
    print(cf.getAllSections())
    print(cf.getAllOptions('126mail_addContactPage'))

4.我们获取到了每一个元素的定位方式和定位表达式，接下来我们又该如何查找到我们的元素对象呢？这时候我们会想到PO（pageObject）设计模式，把我们每一个page页面的元素保存到对应页面的文件中。

新建pageObjects目录，在此目录下分别创建HomePage.py LoginPage.py NewContact.py 三个页面文件

LoginPage.py文件存放我们登录页面所需要的元素, HomePage.py 存放登录成功后首页所需要的页面元素， NewContact.py文件存放我们添加联系人页面所需要的元素。

from util.ObjectMap import *
from util.ParseConfigurationFile import ParseConfigFile
class LoginPage(object):
    '''
    登录页面所有的操作元素对象
    '''
    def __init__(self, driver):
        self.driver = driver
        self.cf = ParseConfigFile()

    def switchToFrame(self):
        '''
        切换到frame中
        :return:
        '''
        by, locator = self.cf.getElementValue('126mail_login', 'loginPage.frame')
        try:
            self.driver.switch_to.frame(getElement(self.driver, by,locator))
        except Exception as e:
            raise e

    def switchToDefaultFrame(self):
        '''
        跳出frame
        :return:
        '''
        try:
            self.driver.switch_to.default_content()
        except Exception as e:
            raise e

    def userNameObj(self): # 用户名输入框
        by, locator = self.cf.getElementValue('126mail_login', 'loginPage.username')

        username = getElement(self.driver, by, locator)
        return username

    def passwordObj(self): # 密码输入框
        by, locator = self.cf.getElementValue('126mail_login', 'loginPage.password')

        password = getElement(self.driver, by, locator)
        return password

    def loginBtnObj(self): # 登录按钮
        by, locator = self.cf.getElementValue('126mail_login', 'loginPage.loginBtn')

        loginbtn = getElement(self.driver, by, locator)
        return loginbtn

if __name__=='__main__':
    from selenium import webdriver
    import time
    driver = webdriver.Firefox()
    driver.get('https://mail.126.com')
    login = LoginPage(driver)
    time.sleep(5)
    login.switchToFrame()
    login.userNameObj().send_keys('linuxxiaochao')
    login.passwordObj().send_keys('xiaochao11520')
    login.loginBtnObj().click()
    login.switchToDefaultFrame()
    driver.quit()

from util.ObjectMap import * #查找元素的模块
from util.ParseConfigurationFile import ParseConfigFile

class HomePage(object):

    def __init__(self, driver):
        self.driver = driver
        self.cf = ParseConfigFile()

    def addressLink(self):
        '''
        通讯录菜单对象
        :return:
        '''
        by, locator = self.cf.getElementValue('126mail_homePage','homePage.addressbook')

        elementObj = getElement(self.driver, by, locator)
        return elementObj

if __name__=='__main__':
    from selenium import webdriver
    from pageObjects.LoginPage import LoginPage
    import time
    driver = webdriver.Firefox()
    driver.get('https://mail.126.com')
    login = LoginPage(driver)
    homePage = HomePage(driver)
    time.sleep(5)
    login.switchToFrame()
    login.userNameObj().send_keys('linuxxiaochao')
    login.passwordObj().send_keys('xiaochao11520')
    login.loginBtnObj().click()
    login.switchToDefaultFrame()
    time.sleep(3)
    homePage.addressLink().click()
    time.sleep(10)
    driver.quit()

from util.ParseConfigurationFile import ParseConfigFile
from util.ObjectMap import *

class AddContactPage(object):
    '''
    添加联系人页面所有操作元素对象
    '''
    def __init__(self, driver):
        self.driver = driver
        self.cf = ParseConfigFile()

    def newContact(self): # 新建联系人
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newContact')

        element = getElement(self.driver, by, locator)
        return element

    def addName(self): # 姓名输入框
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newName')

        element = getElement(self.driver, by, locator)
        return element

    def addMail(self): # 电子邮件输入框
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newMail')

        element = getElement(self.driver, by, locator)
        return element

    def markStar(self): # 设为星际联系人
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newMark')

        element = getElement(self.driver, by, locator)
        return element

    def addPhone(self): # 手机号码输入框
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newPhone')

        element = getElement(self.driver, by, locator)
        return element

    def addContent(self): # 备注
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newComment')

        element = getElement(self.driver, by, locator)
        return element

    def clickCommitBtn(self): # 确定按钮
        by, locator = self.cf.getElementValue('126mail_addContactPage', 'addContactPage.newCommit')

        element = getElement(self.driver, by, locator)
        return element

if __name__=='__main__':
    from selenium import webdriver
    import time
    from pageObjects.HomePage import HomePage
    from appModules.LoginAction import LoginAction

    driver = webdriver.Firefox()
    driver.get('https://mail.126.com')
    time.sleep(3)
    # 登录
    LoginAction.login(driver,'linuxxiaochao', 'xiaochao11520')
    # 主页面
    homepage = HomePage(driver)
    homepage.addressLink().click()
    time.sleep(5)
    # 添加联系人页面
    addcontact = AddContactPage(driver)
    addcontact.newContact().click()
    time.sleep(2)
    addcontact.addName().send_keys('test')
    addcontact.addMail().send_keys('13691579846@qq.com')
    addcontact.addPhone().send_keys('13691579846')
    addcontact.addContent().send_keys('ceshi')
    addcontact.markStar().click()
    time.sleep(3)
    addcontact.clickCommitBtn().click()

5.接下来考虑，我们的框架需要数据驱动？那么我们该怎么用数据驱动我们测试呢？是不是应该通过修改测试数据能够控制我们用例的执行还是不执行呢？没错，大概就是这个样子，那我们又该如何来设计我们的数据呢？大概最好的办法就是使用excel文件来存储或者如果你有能力可以考虑使用数据库来存储，看下我们的表格是什么样子。

新建testData目录存放我们测试过程中所有和业务数据相关的数据文件，并新建126MailContact.xlsx 文件，文件中分两个sheet 126account 和 126contact 分别存放登录和联系人数据

6.通过上面的表我们可以看到两列数据“是否执行”，没错了我们就是通过这来控制数据是否驱动用例执行的，用例执行后写入测试是否通过和测试执行时间（代码中我没加写入时间，感兴趣的自己在用例只加就行了）

7.有了这些数据我们又该如何读取到这些数据应用到我们的用例中呢？在util目录下新建ParseExcel.py 用来解析excel文件

from openpyxl import load_workbook
from config.varCondig import *
class ParseExcel(object):
    '''
    解析excel文件的封装
    '''
    def __init__(self):
        # 加载excel文件到内存
        self.wb = load_workbook(testExcelValuePath)

    def getRowValue(self, sheetName, rawNo):
        '''
        获取某一行的数据
        :param sheetName:
        :param rawNo:
        :return: 列表
        '''
        sh = self.wb[sheetName]
        rowValueList = []
        for y in range(2, sh.max_column+1):
            value = sh.cell(rawNo,y).value
            rowValueList.append(value)
        return rowValueList
    def getColumnValue(self, sheetName, colNo):
        '''
        获取某一列的数据
        :param sheetName:
        :param colNo:
        :return: 列表
        '''
        sh = self.wb[sheetName]
        colValueList = []
        for x in range(2, sh.max_row +1):
            value = sh.cell(x, colNo).value
            colValueList.append(value)
        return colValueList

    def getCellOfValue(self, sheetName, rowNo, colNo):
        '''
        获取某一个单元格的数据
        :param sheetName:
        :param rowNo:
        :param colNo:
        :return: 字符串
        '''
        sh = self.wb[sheetName]
        value = sh.cell(rowNo, colNo).value
        return value
    def writeCell(self, sheetName, rowNo, colNo, value):
        '''
        向某个单元格写入数据
        :param rowNo: 行号
        :param colNo: 列号
        :param value:
        :return: 无
        '''
        sh = self.wb[sheetName]
        sh.cell(rowNo, colNo).value = value
        self.wb.save(testExcelValuePath)
if __name__=='__main__':
    p = ParseExcel()
    print(p.getRowValue('126account',2))
    print(p.getColumnValue('126account',3))
    print(p.getCellOfValue('126account', 2, 3))

8.在config目录下新建varConfig.py文件来存储一些目录信息和数据表对应的列号

import os

# print(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
# 项目目录
parentDirPath = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# init文件路径
pageElementLocatorPath = parentDirPath+r'\config\PageElementLocator.ini'
# excel文件路径
testExcelValuePath = parentDirPath+r'\testData\126MailContact.xlsx'
# 日志文件存放路径
logPath = parentDirPath + r'\log'

# 126username 表，每列对用的序号
account_userName = 2
account_passWord = 3
account_dataBook = 4
account_isExecute = 5
account_testResult = 6

# 126联系人表，每列对应的序号
contact_contactName = 2
contact_contactMail = 3
contact_contactStar = 4
contact_contactPhone = 5
contact_contactComment = 6
contact_contactKeyWords = 7
contact_contactIsExcute = 8
contact_contactExcuteTime = 9
contact_contactTestResult = 10

if __name__=='__main__':

    print(pageElementLocatorPath)
    print(testExcelValuePath)
    print(logPath)

9.所有的数据已经准备完全，我们可以编写用例了把？等等！ 我们是不是要考虑把业务功能封装一下呢？这样也方便以后依赖这些功能的用例直接调用！新建目录appModules 此目录用来存放业务功能代码，

目录下新建LoginAction.py 和 AddContact.py 文件

# 封装登录方法

from pageObjects.LoginPage import LoginPage
class LoginAction(object):
    def __init__(self):
        pass

    @staticmethod #
    def login(driver, username, password):
        '''
        登录场景
        :param driver:
        :param username:
        :param password:
        :return:
        '''
        login = LoginPage(driver)
        login.switchToFrame()
        login.userNameObj().send_keys(username)
        login.passwordObj().send_keys(password)
        login.loginBtnObj().click()
        login.switchToDefaultFrame()

if __name__=='__main__':
    from selenium import webdriver
    driver = webdriver.Firefox()
    driver.get('https://mail.126.com')
    LoginAction.login(driver, 'linux', 'chao')

from pageObjects.HomePage import HomePage
from pageObjects.NewContact import AddContactPage
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from util.ParseConfigurationFile import ParseConfigFile

class NewContactPersonAction(object):
    def __init__(self):
        pass

    @staticmethod
    def addressLink(driver):
        '''
        点击通讯录按钮
        :param driver:
        :return:
        '''
        homePage = HomePage(driver)
        # 点击通讯录
        homePage.addressLink().click()
    @staticmethod
    def addContact(driver, contactName, contactMail, isSatr, contactPhone, contactComment):
        '''
        添加联系人场景
        :param driver:
        :param contactName:
        :param contactMail:
        :param isSatr:
        :param contactPhone:
        :param contactComment:
        :return:
        '''
        # 点击新建联系人
        addContact = AddContactPage(driver)
        # 调试的时候这边有时候会报错。点击不到[新建联系人]这个按钮，所以加了一个显示等待
        by, locator = ParseConfigFile().getElementValue('126mail_addContactPage', 'addContactPage.newContact')
        WebDriverWait(driver, 30).until(EC.element_to_be_clickable((by, locator)))
        addContact.newContact().click()
        if contactName:
            # 非必填项
            addContact.addName().send_keys(contactName)
        # 必填项
        addContact.addMail().send_keys(contactMail)
        if isSatr == '是':
            addContact.markStar().click()
        if contactPhone:
            addContact.addPhone().send_keys(contactPhone)
        if contactComment:
            addContact.addContent().send_keys(contactComment)
        addContact.clickCommitBtn().click()

if __name__=='__main__':
    from appModules.LoginAction import LoginAction
    import time
    from selenium import webdriver
    driver = webdriver.Firefox()
    driver.get('https://mail.126.com')
    time.sleep(5)
    LoginAction.login(driver, 'linux', 'chao')
    NewContactPersonAction.addressLink(driver)
    NewContactPersonAction.addContact(driver, '','123456@qq.com', '是', '','')
    time.sleep(5)
    driver.quit()

10.我们的大体框架就已经搭建完成了，貌似还少了个执行日志，我们在util目录下新建log.py封装日志模块, 加入到我们想加入的任何地方（用例中我就加了几条，可以自己任意加在想加的位置）

import logging
import time
from config.varCondig import *

class Logger(object):
    '''
    封装的日志模块
    '''
    def __init__(self, logger, CmdLevel=logging.INFO, FileLevel=logging.INFO):
        """

        :param logger:
        :param CmdLevel:
        :param FileLevel:
        """
        self.logger = logging.getLogger(logger)
        self.logger.setLevel(logging.DEBUG)  # 设置日志输出的默认级别
        # 日志输出格式
        fmt = logging.Formatter('%(asctime)s - %(filename)s:[%(lineno)s] - [%(levelname)s] - %(message)s')
        # 日志文件名称
        # self.LogFileName = os.path.join(conf.log_path, "{0}.log.txt".format(time.strftime("%Y-%m-%d")))# %H_%M_%S
        currTime = time.strftime("%Y-%m-%d")
        self.LogFileName = logPath+r'\log'+currTime+'.txt'
        # 设置控制台输出
        # sh = logging.StreamHandler()
        # sh.setFormatter(fmt)
        # sh.setLevel(CmdLevel)# 日志级别

        # 设置文件输出
        fh = logging.FileHandler(self.LogFileName)
        fh.setFormatter(fmt)
        fh.setLevel(FileLevel)# 日志级别

        # self.logger.addHandler(sh)
        self.logger.addHandler(fh)

if __name__ == '__main__':
    logger = Logger("fox",CmdLevel=logging.DEBUG, FileLevel=logging.DEBUG)
    logger.logger.debug("debug")
    logger.logger.log(logging.ERROR,'%(module)s %(info)s',{'module':'log日志','info':'error'}) #ERROR,log日志 error

11.目前为止我们的框架就算已经搭建完成了，接下来就是编写我们的测试用例了^_^,新建testCases目录存放测试用例，并新建TestMail126.py编写用例

from selenium import webdriver
import time
from appModules.LoginAction import LoginAction
from appModules.AddContact import NewContactPersonAction
from config.varCondig import *
from util.ParseExcel import ParseExcel
from util.Log import Logger
import logging
import traceback

log = Logger(__name__, CmdLevel=logging.INFO, FileLevel=logging.INFO)
p = ParseExcel()
sheetName = p.wb.sheetnames # 获取所有的sheetname 是个列表

# print(sheetName)
def bDriver():
    try:
        driver = webdriver.Firefox()
        driver.get('https://mail.126.com')
        driver.implicitly_wait(30)
    except Exception as e:
        raise e
    else:
        return driver

def testMailLogin(driver):
    '''
    测试用例
    :return:
    '''
    # 是否执行列数据
    isExcute = p.getColumnValue(sheetName=sheetName[0], colNo=account_isExecute)
    # print(isExcute)
    for idx,value in enumerate(isExcute[:]):
        # print(idx, value) # 获取是否执行列数据列表的索引和数据
        if value.lower() == 'y':
            userRowValue = p.getRowValue(sheetName[0], idx+2) # 获取执行状态为y所在行的数据
            userName = userRowValue[account_userName-2]
            passWord = userRowValue[account_passWord-2]
            # driver = bDriver()
            # 登录
            LoginAction.login(driver,userName, passWord)
            time.sleep(10) # 足够的时间加载登录成功的页面
            try:
                assert '通讯录' in driver.page_source
            except Exception as e:
                log.logger.info('断言"通讯录"失败,错误信息{}'.format(traceback.format_exc()))
                p.writeCell(sheetName[0], idx + 2, account_testResult, 'failed')
                # raise e
            else:
                log.logger.info('{},{}登录成功, 断言”通讯录“成功'.format(userName, passWord))
                p.writeCell(sheetName[0], idx + 2, account_testResult, 'pass')
            # 获取联系人数据表中是否执行列的数据
                isExcute = p.getColumnValue(sheetName=sheetName[1], colNo=contact_contactIsExcute)
                for idx, value in enumerate(isExcute):
                    if value.lower() == 'y':
                        # 获取y表示行的数据
                        contactPersonValue = p.getRowValue(sheetName[1], idx+2)
                        # 获取添加联系人所需的数据
                        # 联系人姓名
                        contactPersonName = contactPersonValue[contact_contactName-2]
                        # 联系人邮箱
                        contactPersonMail = contactPersonValue[contact_contactMail-2]
                        # 是否为星级联系人
                        contactPersonStar = contactPersonValue[contact_contactStar-2]
                        # 联系人手机号
                        contactPersonPhone = contactPersonValue[contact_contactPhone-2]
                        # 联系人备注
                        contactPersonComment = contactPersonValue[contact_contactComment-2]
                        # 验证页面包含的关键字
                        contactAssert = contactPersonValue[contact_contactKeyWords-2]
                        NewContactPersonAction.addressLink(driver)
                        NewContactPersonAction.addContact(driver, contactPersonName, contactPersonMail
                                                          , contactPersonStar, contactPersonPhone, contactPersonComment)
                        try:
                            assert contactAssert in driver.page_source
                        except Exception as e:
                            p.writeCell(sheetName[1], idx + 2, contact_contactTestResult, 'failed')
                            raise e
                        else:
                            p.writeCell(sheetName[1], idx+2, contact_contactTestResult, 'pass')
                        # 设置足够长的时间 让添加联系人成功后的提示框自动消失，当然可以自己写代码关闭
                        time.sleep(10)
            driver.quit()

if __name__=='__main__':
    driver = bDriver()
    testMailLogin(driver)

12.用例编写完成，我们再来编写一个统一执行用例的入口文件，在项目根目录下新建RunTest.py

'''
用例执行入口
'''

import sys
sys.path.append(r'.')

if __name__=='__main__':
    from testCases.TestMail126 import *

    driver = bDriver()
    testMailLogin(driver)

总结

以上就是整个数据驱动测试框架的完整代码，后续还会更新关键字驱动框架和混合测试框架的设计，大家敬请期待！

原代码在我的github https://github.com/13691579846/DataDriverTestFrameWork上大家可以下载学习， QQ交流群：878565760