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

转载：http://www.cnblogs.com/tsbc/p/4080301.html

        研究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.  1 ....
    2 #测试用例 
    3 def test_login_mail(self):
    4  driver = self.driver 
    5  driver.get("http://mail.126.com")
    6  driver.find_element_by_id("idInput").clear()
    7  driver.find_element_by_id("idInput").send_keys("auto_tester")
    8  driver.find_element_by_id("pwdInput").clear()
    9  driver.find_element_by_id("pwdInput").send_keys("123456")
   10  driver.find_element_by_id("loginBtn").click()
   11 ....

    

这段脚本实现，是126邮箱登录功能的一个测试用例，单纯从这段代码来看，并不能直观知道他实现了什么业务，而且对于同一个元素，都出现重复的定位代码，只一个简单的登录功能，一个用例只操作3个元素，就暴露出来了一些代码冗余和可读性的问题。那么如果有更多功能和更多用例，那么用例代码更加厚重，可读性也是大大降低，我们需要想办法对此代码进行优化，通过POM实现登录功能。

 

首先，我们要分离测试对象（元素对象）和测试脚本（用例脚本），那么我们分别创建两个脚本文件， LoginPage.py 用于定义页面元素对象，每一个元素都封装成组件（可以看做存放页面元素对象的仓库）  CaseLoginTest.py 测试用例脚本。我们的实现思想，一切元素和元素的操作组件化定义在Page页面，用例脚本页面，通过调用Page中的组件对象，进行拼凑成一个登录脚本。

在写这两个脚本之前，我先对WebDriver中的一些方法进行重定义，以方便我们在写PO的时候，更简洁，快速。

BasePage.py：

1.  1 #-*- coding: utf-8 -*-
    2 __author__ ='tsbc'
    3 from selenium.webdriver.support.wait importWebDriverWait
    4 from selenium import webdriver
    5 classAction(object):
    6 """
    7  BasePage封装所有页面都公用的方法，例如driver, url ，FindElement等
    8 """
    9 #初始化driver、url、等
   10 def __init__(self, selenium_driver, base_url, pagetitle):
   11  self.base_url = base_url
   12  self.pagetitle = pagetitle
   13  self.driver = selenium_driver
   14 
   15 #打开页面，校验页面链接是否加载正确
   16 def _open(self, url, pagetitle):
   17 #使用get打开访问链接地址
   18 self.driver.get(url)
   19  self.driver.maximize_window()
   20 #使用assert进行校验，打开的链接地址是否与配置的地址一致。调用on_page()方法
   21 assert self.on_page(pagetitle), u"打开开页面失败 %s"% url
   22 
   23 #重写元素定位方法
   24 def find_element(self,*loc):
   25 #return self.driver.find_element(*loc)
   26 try:
   27 WebDriverWait(self.driver,10).until(lambda driver: driver.find_element(*loc).is_displayed())
   28 return self.driver.find_element(*loc)
   29 except:
   30 print u"%s 页面中未能找到 %s 元素"%(self, loc)
   31 
   32 #重写switch_frame方法
   33 def switch_frame(self, loc):
   34 return self.driver.switch_to_frame(loc)
   35 #定义open方法，调用_open()进行打开链接
   36 
   37 def open(self):
   38  self._open(self.base_url, self.pagetitle)
   39 
   40 #使用current_url获取当前窗口Url地址，进行与配置地址作比较，返回比较结果（True False）
   41 def on_page(self, pagetitle):
   42 return pagetitle in self.driver.title
   43 
   44 #定义script方法，用于执行js脚本，范围执行结果
   45 def script(self, src):
   46  self.driver.execute_script(src)
   47 
   48 #重写定义send_keys方法
   49 def send_keys(self, loc, vaule, clear_first=True, click_first=True):
   50 try:
   51  loc = getattr(self,"_%s"% loc)
   52 if click_first:
   53  self.find_element(*loc).click()
   54 if clear_first:
   55  self.find_element(*loc).clear()
   56  self.find_element(*loc).send_keys(vaule)
   57 exceptAttributeError:
   58 print u"%s 页面中未能找到 %s 元素"%(self, loc)

   LoginPage.py：

2.  1 #-*- coding: utf-8 -*-
    2 __author__ ='tsbc'
    3 
    4 from selenium.webdriver.common.by importBy
    5 importBasePage
    6 
    7 #继承BasePage类
    8 classLoginPage(BasePage.Action):
    9 
   10 #定位器，通过元素属性定位元素对象
   11 username_loc =(By.ID,"idInput")
   12  password_loc =(By.ID,"pwdInput")
   13  submit_loc =(By.ID,"loginBtn")
   14  span_loc =(By.CSS_SELECTOR,"div.error-tt>p")
   15  dynpw_loc =(By.ID,"lbDynPw")
   16  userid_loc =(By.ID,"spnUid")
   17 
   18 #Action
   19 def open(self):
   20 #调用page中的_open打开连接
   21 self._open(self.base_url, self.pagetitle)
   22 #调用send_keys对象，输入用户名
   23 def input_username(self, username):
   24  self.find_element(*self.username_loc).send_keys(username)
   25 #调用send_keys对象，输入密码
   26 def input_password(self, password):
   27  self.find_element(*self.password_loc).send_keys(password)
   28 #调用send_keys对象，点击登录
   29 def click_submit(self):
   30  self.find_element(*self.submit_loc).click()
   31 #用户名或密码不合理是Tip框内容展示
   32 def show_span(self):
   33 return self.find_element(*self.span_loc).text
   34 #切换登录模式为动态密码登录（IE下有效）
   35 def swich_DynPw(self):
   36  self.find_element(*self.dynpw_loc).click()
   37 #登录成功页面中的用户ID查找
   38 def show_userid(self):
   39 return self.find_element(*self.userid_loc).text
   40 
   41 CaseLoginTest.py：
   42 #-*- coding: utf-8 -*-
   43 __author__ ='tsbc'
   44 import sys
   45 reload(sys)
   46 sys.setdefaultencoding('utf-8')
   47 import unittest 
   48 from PO importLoginPage
   49 from selenium import webdriver
   50 
   51 classCaselogin126mail(unittest.TestCase):
   52 """
   53 登录126邮箱的case
   54  """
   55 @classmethod
   56 def setUpClass(cls):
   57  cls.driver = webdriver.Chrome()
   58  cls.driver.implicitly_wait(30)
   59 
   60  cls.url ="http://mail.126.com"
   61  cls.username ="auto_tester"
   62  cls.password ="123456"
   63 
   64 #用例执行体
   65 def test_login_mail(self):
   66 #声明LoginPage类对象
   67 login_page =LoginPage.LoginPage(self.driver, self.url, u"网易")
   68 
   69 #调用打开页面组件
   70 login_page.open()
   71 #调用用户名输入组件
   72 login_page.input_username(self.username)
   73 #调用密码输入组件
   74 login_page.input_password(self.password)
   75 #调用点击登录按钮组件
   76 login_page.click_submit()
   77 @classmethod
   78 def tearDownClass(cls):
   79  cls.driver.quit()
   80 
   81 if __name__ =="__main__":
   82  unittest.main()

    

通过使用POM进行重新构造代码结构后，发现代码测试用例代码的可读性提高很多，元素写成组件的方式，不需要每次都写findElement直接在脚本中调用组件就可以使用。在CaseLoginTest脚本用例执行体中，一旦我们输入 login_page并敲入一个点时，LoginPage页面中的元素对象组件都显示出来。并且定义好的PageObject组件可以重复在其它的脚本中进行使用，减少了代码的工作量，也方便对脚本进行后期的维护管理，当元素属性发生变化时，我们只需要对一个PageObaject页面中的对象组件定义进行更改即可。

<简单对Page Object Model使用实例进行描述，希望能够对Selenium自动化的POM理解有所帮助>