Python+Selenium框架版（五）- 框架內封裝基類和實現POM

　　目標：封裝一個基類和介紹如何實現POM。關於基類，是這樣定義的：把一些常見的頁面操作的selenium封裝到base_page.py這個類文件，以后每個POM中的頁面類，都繼承這個基類，這樣每個頁面類都有基類的方法。

　　1.在實現封裝基類里，我們實現了元素八大方式的定位和截圖類封裝。具體項目層級結構如下圖。

 

 

2.基類base_page.py的具體實現代碼，這里就封裝了幾個常用方法，其他方法，你自己去練習封裝下。

import time
from selenium.common.exceptions import NoSuchElementException
import os.path
from framework.logger import Logger

logger = Logger(logger="BasePage").getlog()

class BasePage(object):
    """
    定義一個頁面基類，讓所有頁面都能繼承這個類，封裝一些常用的頁面操作方法到這個類
    """
    def __init__(self,driver):
        self.driver = driver

    def quit_browser(self):
        """
        退出瀏覽器
        :return:
        """
        self.driver.quit()
        logger.info("退出瀏覽器。")

    def forward(self):
        """
        單擊當前頁上的【前進】按鈕
        :return:
        """
        self.driver.forward()
        logger.info("單擊當前頁上的【前進】按鈕。 ")

    def back(self):
        """
        單擊當前頁上的【后退】按鈕。
        :return:
        """
        self.driver.back()
        logger.info("單擊當前頁上的【后退】按鈕。")

    def wait(self,seconds):
        """
        隱式等待
        :return:
        """
        self.driver.implicitly_wait(seconds)
        logger.info("隱式等待 %d 秒。"%seconds)

    def close(self):
        """
        點擊關閉當前窗口
        :return:
        """
        try:
            self.driver.close()
            logger.info("點擊關閉當前窗口。")
        except NameError as e:
            logger.error("點擊關閉當前窗口失敗：%s"%e)


    def get_window_img(self):
        """
        在這里我們把file_path這個參數寫死，直接保存到我們項目根目錄的一個文件夾.\screenshots下
        :return:
        """
        file_path = os.path.dirname(os.path.abspath('.')) + '/screenshots/'
        rq = time.strftime('%Y%m%d%H%M',time.localtime(time.time()))
        screen_name = file_path + rq + '.png'
        try:
            self.driver.get_screenshot_as_file(screen_name)
            logger.info("成功截圖並保存到這里：/screenshots")
        except NameError as e:
            logger.error("截圖失敗！ %s"%e)
            self.get_window_img()

    def find_element(self,selector):
        """
        這個地方為什么是根據=>來切割字符串，請看頁面里定位元素的方法
         submit_btn = "id=>su"
         login_lnk = "xpath => //*[@id='u1']/a[7]"  # 百度首頁登錄鏈接定位
         如果采用等號，結果很多xpath表達式中包含一個=，這樣會造成切割不准確，影響元素定位
        :param selector:
        :return:
        """
        element = ''
        if '=>' not in selector:
            return self.driver.find_element_by_id(selector)
        selector_by = selector.split('=>')[0]
        selector_value = selector.split('=>')[1]

        if selector_by == "i" or selector_by == 'id':
            try:
                element = self.driver.find_element_by_id(selector_value)
                logger.info("Had find the element \' %s \' successful"
                            "by %s via value:%s " % (element.text,selector_by,selector_value))
            except NoSuchElementException as e:
                logger.error("NoSuchElementException: %s" %e)
                self.get_window_img()
        elif selector_by == "n" or selector_by == 'name':
            element = self.driver.find_element_by_name(selector_value)
        elif selector_by == "c" or selector_by == 'class_name':
            element = self.driver.find_element_by_class_mane(selector_value)
        elif selector_by == "l" or selector_by == 'link_text':
            element = self.driver.find_element_by_link_text(selector_value)
        elif selector_by == "p" or selector_by == 'partial_link_text':
            element = self.driver.find_element_by_partial_link_text(selector_value)
        elif selector_by == "t" or selector_by == 'tag_name':
            element = self.driver.find_element_by_tag_name(selector_value)
        elif selector_by == "x" or selector_by == 'xpath':
            try:
                element = self.driver.find_element_by_xpath(selector_value)
                logger.info("Had find the element \' %s \' successful"
                                "by %s via value: %s "% (element.text,selector_by,selector_value))
            except NoSuchElementException as e:
                logger.error("NoSuchElementException: %s" %e)
                self.get_window_img()
        elif selector_by == "s" or selector_by == 'selector_selector':
            element = self.driver.find_element_by_css_selector(selector_value)
        else:
            raise NameError("Please enter a valid type of targeting elements.")

        return element

    def type(self,selector,text):
        """
        輸入
        :param selector:
        :param text:
        :return:
        """
        el = self.find_element(selector)
        el.clear()
        try:
            el.send_keys(text)
            logger.info("Had type \' %s \' in inputBox" %text)
        except NameError as e:
            logger.error("Failed to type in input box with %s"%e)
            self.get_window_img()

    def clear(self,selector):
        """
        清除文本框
        :param selector:
        :return:
        """
        el = self.find_element(selector)
        try:
            el.clear()
            logger.info("Clear text in input box before typing.")
        except NameError as e:
            logger.error("Fail to clear in input box with %s"%e)
            self.get_window_img()

    def click(self,selector):
        """
        點擊元素
        :param selector:
        :return:
        """
        el = self.find_element(selector)
        try:
            el.click()
            logger.info("The element \' %s \' was clicked." %el.text)
        except NameError as e:
            logger.error("Fail to click the element with %s" %e)

    def get_pape_title(self):
        """
        獲取網頁標題
        :return:
        """
        logger.info("Current page title is %s" %self.driver.title)
        return self.driver.title

    @staticmethod
    def sleep(seconds):
        time.sleep(seconds)
        logger.info("Sleep for %d seconds"%seconds)

　　

3. 頁面對象中，百度主頁的元素定位和簡單的操作函數，頁面類主要是元素定位和頁面操作寫成函數，供測試類調用。

baidu_homepage.py

from framework.base_page import BasePage

class HomePage(BasePage):
    input_box = "id=>kw"
    search_submit_btn = "xpath=>//*[@id='su']"

    def type_search(self,text):
        self.type(self.input_box,text)

    def send_submit_btn(self):
        self.click(self.search_submit_btn)

　　

這里注意下元素定位寫法，=>和base_page.py中find_element()方法元素定位切割有關系，網上有些人寫根據逗號切割或者等號切割，在實際使用xpath定位，發現單獨逗號或者單獨等號切割都不精確，造成元素定位失敗。

4. 測試類的寫法舉例

import time
import unittest
from framework.browser_engine import BrowserEngine
from pageobjects.baidu_homepage import HomePage

class BaiduSearch(unittest.TestCase):
    def setUp(self):
        """
        測試固件的setUp()代碼，主要是測試的前提工作
        :return:
        """
        browse = BrowserEngine(self)
        self.driver = browse.open_browser(self)

    def tearDown(self):
        """
        測試結束后的操作，這里基本上都是關閉瀏覽器
        :return:
        """
        self.driver.quit()

    def test_baidu_search(self):
        """
        這里一定要test開頭，
        把測試邏輯代碼封裝到一個test開頭的方法里
        :return:
        """
        homepage = HomePage(self.driver)
        homepage.type_search('selenium')# 調用頁面對象中的方法
        homepage.send_submit_btn() #調用頁面對象類中的點擊搜索按鈕方法
        time.sleep(2)
        homepage.get_window_img() # 調用基類截圖方法
        try:
            assert 'selenium' in homepage.get_pape_title()
            # 調用頁面對象繼承基類中的獲取頁面標題方法
            print('Test Pass.')
        except Exception as e:
            print('Test Fail.',format(e))

if __name__ == '__main__':
    unittest.main()

　　

注意：

homepage = HomePage(self.driver)

　　

  上面這行代碼要注意，意思是，到一個頁面，第一件事情是初始化這個頁面的一個頁面對象實例。注意，一定要帶self.driver，不然會報錯，這個self.driver，可以這樣理解：在當前測試類里面，self.driver是來自瀏覽器引擎類中方法得到的，在初始化一個頁面對象的時候，也把這個來自瀏覽器引擎類的driver給賦值給當前的頁面對象，這樣，才能執行頁面對象或者基類里面的相關driver方法。寫多了selenium的自動化腳本，你會明白，最重要的是保持前后driver的唯一性。
測試結果：會在logs文件夾生成一個日志文件，也會在screenshots文件夾生成一個png圖片。

 

參考文章：https://blog.csdn.net/u011541946/article/details/70269965