Python+Selenium使用Page Object實現頁面自動化測試

    Page Object模式是Selenium中的一種測試設計模式，主要是將每一個頁面設計為一個Class，其中包含頁面中需要測試的元素（按鈕，輸入框，標題 等），這樣在Selenium測試頁面中可以通過調用頁面類來獲取頁面元素，這樣巧妙的避免了當頁面元素id或者位置變化時，需要改測試頁面代碼的情況。 當頁面元素id變化時，只需要更改測試頁Class中頁面的屬性即可。

    Page Object模式是一種自動化測試設計模式，將頁面定位和業務操作分開，分離測試對象（元素對象）和測試腳本（用例腳本），提高用例的可維護性。

    unittest是一種單元測試框架，用於設計各式各樣的測試用例，可調用PageObject設計的頁面類（對象），設計出更加可維護的用例。它提供用例組織與執行，提供豐富的比較（斷言）方法，提供豐富的日志，統一適用於web自動化用例的開發與執行。

 

使用PO模式設計思路如下：

1.定義頁面基礎類，封裝所有頁面公用的方法。

命名為test_8_3_2_BasePage.py

# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:基礎類BasePage，封裝所有頁面都公用的方法，
定義open函數，重定義find_element，switch_frame，send_keys等函數。
在初始化方法中定義驅動driver，基本url，title
WebDriverWait提供了顯式等待方式。
'''
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class BasePage(object):
    """
    BasePage封裝所有頁面都公用的方法，例如driver, url ,FindElement等
    """
    #初始化driver、url、pagetitle等
    #實例化BasePage類時，最先執行的就是__init__方法，該方法的入參，其實就是BasePage類的入參。
    #__init__方法不能有返回值，只能返回None
    #self只實例本身，相較於類Page而言。
    def __init__(self, selenium_driver, base_url, pagetitle):
        self.driver = selenium_driver
        self.base_url = base_url
        self.pagetitle = pagetitle
          
    #通過title斷言進入的頁面是否正確。
    #使用title獲取當前窗口title，檢查輸入的title是否在當前title中，返回比較結果（True 或 False）
    def on_page(self, pagetitle):
        return pagetitle in self.driver.title
    
    #打開頁面，並校驗頁面鏈接是否加載正確
    #以單下划線_開頭的方法，在使用import *時，該方法不會被導入，保證該方法為類私有的。
    def _open(self, url, pagetitle):
        #使用get打開訪問鏈接地址
        self.driver.get(url)
        self.driver.maximize_window()
        #使用assert進行校驗，打開的窗口title是否與配置的title一致。調用on_page()方法
        assert self.on_page(pagetitle), u"打開開頁面失敗 %s"%url
    
    #定義open方法，調用_open()進行打開鏈接
    def open(self):
        self._open(self.base_url, self.pagetitle)
    
    #重寫元素定位方法
    def find_element(self,*loc):
#        return self.driver.find_element(*loc)
        try:
            #確保元素是可見的。
            #注意：以下入參為元組的元素，需要加*。Python存在這種特性，就是將入參放在元組里。
#            WebDriverWait(self.driver,10).until(lambda driver: driver.find_element(*loc).is_displayed())
            #注意：以下入參本身是元組，不需要加*
            WebDriverWait(self.driver,10).until(EC.visibility_of_element_located(loc))
            return self.driver.find_element(*loc)
        except:
            print u"%s 頁面中未能找到 %s 元素"%(self, loc)
    
    #重寫switch_frame方法
    def switch_frame(self, loc):
        return self.driver.switch_to_frame(loc)
    
    #定義script方法，用於執行js腳本，范圍執行結果
    def script(self, src):
        self.driver.execute_script(src)
    
    #重寫定義send_keys方法
    def send_keys(self, loc, vaule, clear_first=True, click_first=True):
        try:
            loc = getattr(self,"_%s"% loc)  #getattr相當於實現self.loc
            if click_first:
                self.find_element(*loc).click()
            if clear_first:
                self.find_element(*loc).clear()
                self.find_element(*loc).send_keys(vaule)
        except AttributeError:
            print u"%s 頁面中未能找到 %s 元素"%(self, loc)

2.定義登錄頁面的基本操作方法。

所有頁面元素定位都在此層定義，UI一旦有更改，只需在修改這一層頁面對象屬性即可。

命名為test_8_3_2_LoginPage.py

# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:頁面基本操作方法：如open，input_username，input_password，click_submit
'''
from selenium.webdriver.common.by import By
from test_8_3_2_BasePage import BasePage

#繼承BasePage類
class LoginPage(BasePage):
    #定位器，通過元素屬性定位元素對象
    username_loc =(By.NAME,'email')
    password_loc =(By.NAME,'password')
    submit_loc =(By.ID,'dologin')
    span_loc =(By.CSS_SELECTOR,"div.error-tt>p")
    dynpw_loc =(By.ID,"lbDynPw")
    userid_loc =(By.ID,"spnUid")
    
    #操作
    #通過繼承覆蓋（Overriding）方法：如果子類和父類的方法名相同，優先用子類自己的方法。
    #打開網頁
    def open(self):
    #調用page中的_open打開連接
        self._open(self.base_url, self.pagetitle)
    
    #輸入用戶名：調用send_keys對象，輸入用戶名
    def input_username(self, username):
#        self.find_element(*self.username_loc).clear()
        self.find_element(*self.username_loc).send_keys(username)
    
    #輸入密碼：調用send_keys對象，輸入密碼
    def input_password(self, password):
#        self.find_element(*self.password_loc).clear()
        self.find_element(*self.password_loc).send_keys(password)
        
    #點擊登錄：調用send_keys對象，點擊登錄
    def click_submit(self):
        self.find_element(*self.submit_loc).click()
    
    #用戶名或密碼不合理是Tip框內容展示
    def show_span(self):
        return self.find_element(*self.span_loc).text
    
    #切換登錄模式為動態密碼登錄（IE下有效）
    def swich_DynPw(self):
        self.find_element(*self.dynpw_loc).click()
    
    #登錄成功頁面中的用戶ID查找
    def show_userid(self):
        return self.find_element(*self.userid_loc).text

3.使用unittest框架編寫測試用例

# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:使用unittest框架編寫測試用例。
'''
import unittest 
from test_8_3_2_LoginPage import LoginPage
from selenium import webdriver

class Caselogin126mail(unittest.TestCase):
    """
          登錄126郵箱的case
    """
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.url ="http://mail.126.com"
        self.username ="XXX"
        self.password ="XXX"
    
    #用例執行體
    def test_login_mail(self):
        #聲明LoginPage類對象
        login_page = LoginPage(self.driver, self.url, u"網易")
        #調用打開頁面組件
        login_page.open()
        #切換到登錄框Frame
        login_page.switch_frame('x-URS-iframe')
        #調用用戶名輸入組件
        login_page.input_username(self.username)    
        #調用密碼輸入組件
        login_page.input_password(self.password)       
        #調用點擊登錄按鈕組件
        login_page.click_submit()

    def tearDown(self):
        self.driver.quit()
        
if __name__ == "__main__":
    unittest.main()

 

結語：

這樣分層的好處是，不同層關心不同的問題。頁面對象層只關心元素定位問題，測試用例只關心測試的數據。