Selenium的PO模式（Page Object Model）|(Selenium Webdriver For Python)

        研究Selenium + python 自動化測試有近兩個月了，不能說非常熟練，起碼對selenium自動化的執行有了深入的認識。

從最初無結構的代碼，到類的使用，方法封裝，從原始函數調用，到重定義函數。從變量驅動，到數據驅動，再到關鍵字驅動，一步一步的默默走向自動化框架的構建。 雖然還有沒有投入使用，只是寫幾個demo，就慢慢的發現了 selenium自動用例腳本，相似功能地方，代碼基本都是一樣的，界面元素換個查找方式，把原來的使用 xpath方式，改為使用 id 查找，需要對每個用例腳本都要改，雖然幾個用例看不出什么工作量，但是重復findElement的代碼，已經讓我們感到了代碼的笨重。如果某些定位發生了改變，我們就得貫穿整個測試代碼進行調整元素定位，這樣就會導致我們的腳本在后期，難以維護。

        因此通過Page Object Model 我們可以創建更加健壯代碼，並減少或者消除重復的測試代碼，從而也能夠提高代碼的可讀性，減少編寫腳本的工作量。Page Object Model的實現，就是通過分離測試對象和測試腳本的抽象來實現的。

PO項目實戰演示Demo源碼：（源碼會隨着結構調整不斷更新，但是整體結構不會變動）

https://coding.net/u/tsbc/p/PySelenium_PO/git

 Page Object Model 簡稱POM 

下面通過一個實例演示POM的實現

在此之前，先看一個段比較普通的測試用例代碼片段：

1. ....
   #測試用例 
   def test_login_mail(self):
    driver = self.driver 
    driver.get("http://mail.126.com")
    driver.find_element_by_id("idInput").clear()
    driver.find_element_by_id("idInput").send_keys("auto_tester")
    driver.find_element_by_id("pwdInput").clear()
    driver.find_element_by_id("pwdInput").send_keys("123456")
    driver.find_element_by_id("loginBtn").click()
   ....

    

這段腳本實現，是126郵箱登錄功能的一個測試用例，單純從這段代碼來看，並不能直觀知道他實現了什么業務，而且對於同一個元素，都出現重復的定位代碼，只一個簡單的登錄功能，一個用例只操作3個元素，就暴露出來了一些代碼冗余和可讀性的問題。那么如果有更多功能和更多用例，那么用例代碼更加厚重，可讀性也是大大降低，我們需要想辦法對此代碼進行優化，通過POM實現登錄功能。

 

首先，我們要分離測試對象（元素對象）和測試腳本（用例腳本），那么我們分別創建兩個腳本文件， LoginPage.py 用於定義頁面元素對象，每一個元素都封裝成組件（可以看做存放頁面元素對象的倉庫）  CaseLoginTest.py 測試用例腳本。我們的實現思想，一切元素和元素的操作組件化定義在Page頁面，用例腳本頁面，通過調用Page中的組件對象，進行拼湊成一個登錄腳本。

在寫這兩個腳本之前，我先對WebDriver中的一些方法進行重定義，以方便我們在寫PO的時候，更簡潔，快速。

BasePage.py：

1. #-*- coding: utf-8 -*-
   __author__ ='tsbc'
   from selenium.webdriver.support.wait importWebDriverWait
   from selenium import webdriver
   classAction(object):
   """
    BasePage封裝所有頁面都公用的方法，例如driver, url ，FindElement等
   """
   #初始化driver、url、等
   def __init__(self, selenium_driver, base_url, pagetitle):
    self.base_url = base_url
    self.pagetitle = pagetitle
    self.driver = selenium_driver
   
   #打開頁面，校驗頁面鏈接是否加載正確
   def _open(self, url, pagetitle):
   #使用get打開訪問鏈接地址
   self.driver.get(url)
    self.driver.maximize_window()
   #使用assert進行校驗，打開的鏈接地址是否與配置的地址一致。調用on_page()方法
   assert self.on_page(pagetitle), u"打開開頁面失敗 %s"% url
   
   #重寫元素定位方法
   def find_element(self,*loc):
   #return self.driver.find_element(*loc)
   try:
   WebDriverWait(self.driver,10).until(lambda driver: driver.find_element(*loc).is_displayed())
   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)
   #定義open方法，調用_open()進行打開鏈接
   
   def open(self):
    self._open(self.base_url, self.pagetitle)
   
   #使用current_url獲取當前窗口Url地址，進行與配置地址作比較，返回比較結果（True False）
   def on_page(self, pagetitle):
   return pagetitle in self.driver.title
   
   #定義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)
   if click_first:
    self.find_element(*loc).click()
   if clear_first:
    self.find_element(*loc).clear()
    self.find_element(*loc).send_keys(vaule)
   exceptAttributeError:
   print u"%s 頁面中未能找到 %s 元素"%(self, loc)

   LoginPage.py：

2. #-*- coding: utf-8 -*-
   __author__ ='tsbc'
   
   from selenium.webdriver.common.by importBy
   importBasePage
   
   #繼承BasePage類
   classLoginPage(BasePage.Action):
   
   #定位器，通過元素屬性定位元素對象
   username_loc =(By.ID,"idInput")
    password_loc =(By.ID,"pwdInput")
    submit_loc =(By.ID,"loginBtn")
    span_loc =(By.CSS_SELECTOR,"div.error-tt>p")
    dynpw_loc =(By.ID,"lbDynPw")
    userid_loc =(By.ID,"spnUid")
   
   #Action
   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).send_keys(username)
   #調用send_keys對象，輸入密碼
   def input_password(self, password):
    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
   
   CaseLoginTest.py：
   #-*- coding: utf-8 -*-
   __author__ ='tsbc'
   import sys
   reload(sys)
   sys.setdefaultencoding('utf-8')
   import unittest 
   from PO importLoginPage
   from selenium import webdriver
   
   classCaselogin126mail(unittest.TestCase):
   """
   登錄126郵箱的case
    """
   @classmethod
   def setUpClass(cls):
    cls.driver = webdriver.Chrome()
    cls.driver.implicitly_wait(30)
   
    cls.url ="http://mail.126.com"
    cls.username ="auto_tester"
    cls.password ="123456"
   
   #用例執行體
   def test_login_mail(self):
   #聲明LoginPage類對象
   login_page =LoginPage.LoginPage(self.driver, self.url, u"網易")
   
   #調用打開頁面組件
   login_page.open()
   #調用用戶名輸入組件
   login_page.input_username(self.username)
   #調用密碼輸入組件
   login_page.input_password(self.password)
   #調用點擊登錄按鈕組件
   login_page.click_submit()
   @classmethod
   def tearDownClass(cls):
    cls.driver.quit()
   
   if __name__ =="__main__":
    unittest.main()

    

通過使用POM進行重新構造代碼結構后，發現代碼測試用例代碼的可讀性提高很多，元素寫成組件的方式，不需要每次都寫findElement直接在腳本中調用組件就可以使用。在CaseLoginTest腳本用例執行體中，一旦我們輸入 login_page並敲入一個點時，LoginPage頁面中的元素對象組件都顯示出來。並且定義好的PageObject組件可以重復在其它的腳本中進行使用，減少了代碼的工作量，也方便對腳本進行后期的維護管理，當元素屬性發生變化時，我們只需要對一個PageObaject頁面中的對象組件定義進行更改即可。

<簡單對Page Object Model使用實例進行描述，希望能夠對Selenium自動化的POM理解有所幫助>

來自為知筆記(Wiz)