Python Selenium设计模式-POM

前言

本文就python selenium自动化测试实践中所需要的POM设计模式进行分享，以便大家在实践中对POM的特点、应用场景和核心思想有一定的理解和掌握。

为什么要用POM

基于python selenium2开始UI级自动化测试并不是多么艰巨的任务。**只需要定位到元素，执行对应的操作即可。**下面我们看一下这个简单的脚本实现百度搜索。

from selenium import webdriver
import time

driver = webdriver.Firefox()
driver.implicitly_wait(30)

# 启动浏览器，访问百度
driver.get("http://www.baidu.com")

# 定位 百度搜索框，并输入selenium
driver.find_element_by_id("kw").send_keys("selenium")

# 定位 百度一下 按钮并单击进行搜索
driver.find_element_by_id("su").click()

time.sleep(5)driver.quit()

从上述代码来看，我们所能做的就是定位到元素，然后进行键盘输入或鼠标动作。就这个小程序而已，维护起来看起来是很容易的。但随着时间的迁移，测试套件将持续的增长。脚本也将变得越来越臃肿庞大。如果变成我们需要维护10个页面，100个页面，甚至1000个呢？那页面元素的任何改变都会让我们的脚本维护变得繁琐复杂，而且变得耗时易出错。

那怎么解决呢？ 在自动化测试中，引入了Page Object Model（POM）：页面对象模式来解决，POM能让我们的测试代码变得可读性更好，高可维护性，高复用性。

下图为非POM和POM对比图：

POM是什么

页面对象模型(POM)是一种设计模式，用来管理维护一组web元素集的对象库

在POM下，应用程序的每一个页面都有一个对应的page class

每一个page class维护着该web页的元素集和操作这些元素的方法

page class中的方法命名最好根据其对应的业务场景进行，例如通常登录后我们需要等待几秒中，我们可以这样命名该方法: waitingForLoginSuccess().

下面我们看看POM的代码目录组织示例：

POM的优势

POM提供了一种在UI层操作、业务流程与验证分离的模式，这使得测试代码变得更加清晰和高可读性

对象库与用例分离，使得我们更好的复用对象，甚至能与不同的工具进行深度结合应用

可复用的页面方法代码会变得更加优化

更加有效的命名方式使得我们更加清晰的知道方法所操作的UI元素。例如我们要回到首页，方法名命名为: gotoHomePage(),通过方法名即可清晰的知道具体的功能实现。

POM实现示例

下面我们看下使用POM百度搜索 POM代码示例： 看先下代码组织结构如下：

# basePage.py代码如下

# _*_ coding:utf-8 _*_

__author__ = '苦叶子'

import sys
reload(sys)
sys.setdefaultencoding("utf-8")

# pages基类
class Page(object):
   """
       Page基类，所有page都应该继承该类
   """
   def __init__(self, driver, base_url=u"http://www.baidu.com"):
       self.driver = driver
       self.base_url = base_url
       self.timeout = 30
   
   def find_element(self, *loc):
       return self.driver.find_element(*loc)  
 
   def input_text(self, loc, text):
       self.find_element(*loc).send_keys(text)
           
   def click(self, loc):
       self.find_element(*loc).click()
         
   def get_title(self):
       return self.driver.title

# searchPage.py 代码如下
# _*_ coding:utf-8 _*_

__author__ = '苦叶子'

import sys
from selenium.webdriver.common.by import By
from pages.basePage import Page

reload(sys)
sys.setdefaultencoding("utf-8")

# 百度搜索page
class SearchPage(Page):
   # 元素集
   
   # 搜索输入框
   search_input = (By.ID, u'kw')    
   # 百度一下 按钮
   search_button = (By.ID, u'su')  
 
   def __init__(self, driver, base_url=u"http://www.baidu.com"):
       Page.__init__(self, driver, base_url)  
     
   def gotoBaiduHomePage(self):
       print u"打开首页: ", self.base_url
       self.driver.get(self.base_url)          
      
   def input_search_text(self, text=u"开源优测"):
       print u"输入搜索关键字： 开源优测 "
       self.input_text(self.search_input, text)  
     
   def click_search_btn(self):
       print u"点击 百度一下  按钮"
       self.click(self.search_button)

# testSearchPage.py代码如下

# _*_ coding:utf-8 _*_

__author__ = '苦叶子'
import unittest
import sys
from selenium import webdriver
from pages.searchPage import SearchPage

reload(sys)
sys.setdefaultencoding("utf-8")

# 百度搜索测试
class TestSearchPage(unittest.TestCase):
   
   def setUp(self):
       self.driver = webdriver.Ie()    
   
   def testSearch(self):
       driver = self.driver        
       # 百度网址
       url = u"http://www.baidu.com"
       # 搜索文本
       text = u"开源优测"
       # 期望验证的标题
       assert_title = u"开源优测_百度搜索"
       print assert_title
       
       search_Page = SearchPage(driver, url)  
         
       # 启动浏览器，访问百度首页
       search_Page.gotoBaiduHomePage()   
             
       # 输入 搜索词
       search_Page.input_search_text(text)
                     
       # 单击 百度一下 按钮进行搜索
       search_Page.click_search_btn()    
                        
       # 验证标题
       self.assertEqual(search_Page.get_title(), assert_title)  
                             
   def tearDown(self):
       self.driver.quit()

# 主入口程序代码如下
# _*_ coding:utf-8 _*_
__author__ = '苦叶子'

import unittest
import sys
from common import HTMLTestRunner
from testcase.testSearchPage
import TestSearchPage

reload(sys)
sys.setdefaultencoding("utf-8")
if __name__ == '__main__':
   testunit = unittest.TestSuite()
   testunit.addTest(TestSearchPage('testSearch'))  
 
   # 定义报告输出路径
   htmlPath = u"page_demo_Report.html"
   fp = file(htmlPath, "wb")
   
   runner = HTMLTestRunner.HTMLTestRunner(stream=fp,
          title=u"百度测试",
          description=u"测试用例结果")
   
   runner.run(testunit)
   
   fp.close()

按照如图所示组织代码结构，输入如上代码，执行以下命令运行，会在当前目录生成测试报告：

python main.py

总结

最后做个总结，所有代码请手动输入，不要直接拷贝。再次对POM进行小结

POM是selenium webdriver自动化测试实践对象库设计模式

POM使得测试脚本更易于维护

POM通过对象库方式进一步优化了元素、用例、数据的维护组织