在完成了日誌類封裝之後，那我們就要對測試基類進行實現，在其中對一些請求再次封裝，在專案下新建一個framework資料夾，在資料夾下新建Base_Page.py檔案，這是用來寫測試基類的檔案。在專案下新建screenshots檔案件，用來存放截圖。專案結構如下。

具體怎麼封裝還是要看被測試需要什麼，只能提供一個大致思路，我專案能用的封裝，你專案並不一定能用

測試基類的思路就是把所有能公用的方法全部封裝，並加上日誌輸出。報錯截圖等。

看下Base_Page.py下具體程式碼

from Logs.log import log1
from selenium.common.exceptions import NoSuchElementException
import getcwd
import os
import time
class BasePage():
    '''測試基類'''
    def __init__(self,driver):
        self.driver = driver


    def get_img(self):
        '''截圖'''
        path = os.path.join(getcwd.get_cwd(),'screenshots/')#拼接截圖儲存路徑
        rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))#按格式獲取當前時間
        screen_name = path + rq + '.png'#拼接截圖檔名
        try:
            self.driver.get_screenshot_as_file(screen_name)
            log1.info("截圖儲存成功")
        except BaseException:
            log1.error("截圖失敗",exc_info = 1)

    def find_element(self,selector):
        '''定位元素'''
        by = selector[0]
        value = selector[1]
        element = None
        if by =='id' or by =='name' or by == 'class' or by == 'tag' or by =='link' or by =='plink' or by =='css' or by == 'xpath':
            try:
                if by == 'id':
                    element = self.driver.find_element_by_id(value)
                elif by == 'name':
                    element = self.driver.find_element_by_name(value)
                elif by == 'class':
                    element = self.driver.find_element_by_class_name(value)
                elif by == 'tag':
                    element = self.driver.find_element_by_tag_name(value)
                elif by == 'link':
                    element = self.driver.find_element_by_link_text(value)
                elif by == 'plink':
                    element = self.driver.find_element_by_partial_link_text(value)
                elif by == 'css':
                    element = self.driver.find_element_by_css_selector(value)
                elif by == 'xpath':
                    element = self.driver.find_element_by_xpath(value)
                else:
                    log1.error('沒有找到元素')
                log1.info('元素定位成功。定位方式：%s，使用的值%s：' % (by,value))
                return element
            except NoSuchElementException:
                log1.error("報錯資訊：",exc_info = 1)
                self.get_img()#呼叫截圖
        else:
            log1.error('輸入的元素定位方式錯誤')


    def type(self,selector,value):
        '''輸入內容'''
        element=self.find_element(selector)
        element.clear()
        log1.info('清空輸入內容')
        try:
            element.send_keys(value)
            log1.info('輸入的內容：%s' % value)
        except BaseException:
            log1.error('內容輸入報錯',exc_info = 1)
            self.get_img()


    def click(self,selector):
        '''點選元素'''
        element = self.find_element(selector)
        try:
            element.click()
            log1.info('點選元素成功')
        except BaseException:
            log1.error('點選元素報錯',exc_info = 1)
            self.get_img()


    def sleep(self,secondes):
        '''強制暫停'''
        time.sleep(secondes)
        log1.info('暫停%d秒' % secondes)


    def get_title(self):
        '''獲取title'''
        title = self.driver.title
        log1.info('當前視窗的title是:%s' % title)
        return  title

    def quit(self):
        self.driver.quit()
        log1.info('關閉瀏覽器')
 

只寫了一部分，提供一個思路，還有很多方法可以繼續封裝來完善框架，比如多選單的切換，視窗切換等。

接下來測試一下封裝的程式碼

在test_base.py寫如下程式碼並執行

from selenium import webdriver
from framework.Base_Page import BasePage
dr = webdriver.Firefox()
dr.get('https:www.baidu.com')
s = dr.window_handles
driver = BasePage(dr)
kw = ['id','kw']
driver.type(kw,'selenium+python')
driver.sleep(3)
driver.type(kw,'selenium')
su = ['id','su']
driver.click(su)
driver.get_img()
driver.sleep(2)
driver.get_title()