Python自動化測試系列[v1.0.0][數據驅動DDTXML]

數據文件XML

<?xml version = "1.0" encoding = "utf-8"?>
<bookList type = "technology">
    <book>
        <name>selenium xml datadriven</name>
        <author>davieyang</author>
    </book>
    <book>
        <name>selenium excel datadriven</name>
        <author>davieyang</author>
    </book>
    <book>
        <name>selenium ddt data list</name>
        <author>davieyang</author>
    </book>
</bookList>

解析XML方法

# encoding = utf-8
"""
__title__ = ''
__author__ = 'davieyang'
__mtime__ = '2018/4/21'
"""
from xml.etree import ElementTree
 
 
class ParseXML(object):
    def __init__(self, xmlPath):
        self.xmlPath = xmlPath
 
    def getRoot(self):
        # 打開將要解析的XML文件
        tree = ElementTree.parse(self.xmlPath)
        # 獲取XML文件的根節點對象，然後返回給調用者
        return tree.getroot()
 
    def findNodeByName(self, parentNode, nodeName):
        # 通過節點的名字獲取節點對象
        nodes = parentNode.findall(nodeName)
        return nodes
 
    def getNodeofChildText(self, node):
        # 獲取節點node下所有子節點的節點名作爲key，本節點作爲value組成的字典對象
        childrenTextDict = {i.tag: i.text for i in list(node.iter())[1:]}
        # 上面代碼等價於
        '''
        childrenTextDict = {}
        for i in list(node.iter())[1:]:
            fhildrenTextDict[i.tag] = i.text
        '''
        return childrenTextDict
 
    def getDataFromXml(self):
        # 獲取XML文檔的根節點對象
        root = self.getRoot()
        # 獲取根節點下所有名爲book的節點對象
        books = self.findNodeByName(root, "book")
        dataList = []
        # 遍歷獲取到的所有book節點對象
        # 取得需要的測試數據
        for book in books:
            childrenText = self.getNodeofChildText(book)
            dataList.append(childrenText)
        return dataList
 
 
if __name__ == "__main__":
    xml = ParseXML(r"F:\seleniumWithPython\TestData\TestData.xml")
    datas = xml.getDataFromXml()
    for i in datas:
        print(i["name"], i["author"])

測試腳本

# encoding = utf-8
"""
__title__ = ''
__author__ = 'davieyang'
__mtime__ = '2018/4/21'
"""
from selenium import webdriver
import unittest
import time
import logging
import traceback
import ddt
from Util.ParseXMLUtil import ParseXML
from selenium.common.exceptions import NoSuchElementException
 
 
# 初始化日誌對象
logging.basicConfig(
    # 日誌級別
    level=logging.INFO,
    # 時間、代碼所在文件名、代碼行號、日誌級別名字、日誌信息
    format='%(asctime)s %(filename)s[line: %(lineno)d] %(levelname)s %(message)s',
    # 打印日誌的時間
    datefmt='%a, %d %b %Y %H:%M:%S',
    # 日誌文件存放的目錄及日誌文件名
    filename='D:\\Programs\\Python\\PythonUnittest\\Reports\\TestResults.TestResults',
    # 打開日誌的方式
    filemode='w'
)
 
# currentPath = os.path.dirname(os.path.abspath(__file__))
# dataFilePath = os.path.join(currentPath, "TestData.xml")
dataFilePath = "D:\\Programs\\Python\\PythonUnittest\\TestData\\TestData.xml"
print(dataFilePath)
 
# 創建ParseXML類實例對象
xml = ParseXML(dataFilePath)
 
 
@ddt.ddt
class DataDrivenTestByXML(unittest.TestCase):
 
    def setUp(self):
        self.driver = webdriver.Chrome(executable_path=r"F:\automation\webdriver\chromedriver.exe")
 
    @ddt.data(* xml.getDataFromXml())
    def test_dataDrivenByXML(self, data):
        testData, expectData = data["name"], data["author"]
        url = "http://www.baidu.com"
        self.driver.get(url)
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)
 
        try:
            self.driver.find_element_by_id("kw").send_keys(testData)
            self.driver.find_element_by_id("su").click()
            time.sleep(3)
            self.assertTrue(expectData in self.driver.page_source)
        except NoSuchElementException as e:
            logging.error(u"查找的頁面元素不存在，異常堆棧信息爲：" + str(traceback.format_exc()))
        except AssertionError as e:
            logging.info(u"搜索 ‘%s’,期望 ‘%s’ ,失敗" % (testData, expectData))
        except Exception as e:
            logging.error(u"未知錯誤，錯誤信息：" + str(traceback.format_exc()))
        else:
            logging.info(u"搜索 ‘%s’,期望 ‘%s’ ,通過" % (testData, expectData))
 
    def tearDown(self):
        self.driver.quit()
 
 
if __name__ == "__main__":
    unittest.main()