selenium自动化测试框架之PO设计模式

测试积点老人

作者作品链接 点击链接>>>

面向对象的特性：封装、继承、多态。在自动化中一样适用，Selenium自动化测试中有一个名字常常被提及PageObject（思想与面向对象的特性相同），通过PO模式可以大大提高测试用例的维护效率。

##传统测试脚本的弊端

• 测试脚本分离，维护成本高
• 可扩展性差
• 复用性低等
  

PageObject设计模式

PO的核心要素：

• 在PO模式中抽象封装成一个BasePage类，该基类应该拥有一个只实现webdriver实例的属性。
• 每个一个page都继承BasePage，通过driver来管理本page中元素，将page中的操作封装成一个个的方法。
• TestCase继承unittest.Testcase类，并且依赖page类，从而实现相应的测试步骤。
  

案例基础案例

前面基础场景选取的是baidu搜索页面（baidu页面简单，不需要搭建测试环境）baidu.py

1. from selenium import webdriver
   
2. from time import sleep
   
3. 
   
4. driver = webdriver.Firefox()
   
5. driver.get("http://www.baidu.com")
   
6. 
   
7. driver.find_element_by_xpath("//input[@id='kw']").send_keys("Bela")
   
8. driver.find_element_by_xpath("//input[@id='su']").click()
   
9. sleep(5)
   
10. driver.quit()

复制代码

将上面的脚本放在baidu.py文件中。

分析

通过对baidu.py脚本的分析，可以提取到：

• 不同的运行脚本环境，浏览器不同：驱动webdriver.Firefox()可以剥离，
• 请求地址的变化（生产环境与测试环境）：url==http://www.baidu.com可以剥离
• 操作元素时，常常需要等待元素加载完毕后方可进行操作：是否可以把webdriver提供的findelement* 方法封装下，才操作元素前，先判断元素的是否可操作。
  

===================================================

• 实际测试场景中，可能有多个测试场景，如果每个测试场景都需要维护url、浏览器驱动、元素定位等，效率会非常低。
• 因此基于对上面的分析，是否可以设计一个所有测试页面（selenium本身是对B/S系统开展测试）的基类，来维护一些公共的方法。此处先定义个名字哦BasePage.py，用于存放页面公共方法及webdriver原有方法二次封装等。
  BasePage.py内容如下：
  

1. from selenium.webdriver.support.wait import WebDriverWait
   
2. from selenium import  webdriver
   
3. from selenium.webdriver.support import expected_conditions as EC
   
4. 
   
5. class BasePage(object):
   
6.     """
   
7.     BasePage封装所有页面都公用的方法，例如driver, Find_Element等
   
8.     """
   
9.     # 实例化BasePage类时，最先执行的就是__init__方法，该方法的入参，其实就是BasePage类的入参。
   
10.     # __init__方法不能有返回值，只能返回None
    
11. 
    
12.     def __init__(self,selenium_driver,base_url):
    
13.         self.driver = selenium_driver
    
14.         self.base_url = base_url
    
15.         # self.pagetitle = pagetitle
    
16. 
    
17.     def on_page(self,pagetitle):
    
18.         return pagetitle in self.driver.title
    
19. 
    
20.     def _open(self,url):
    
21.         self.driver.get(url)
    
22.         self.driver.maximize_window()
    
23. 
    
24.     def open(self):
    
25.         self._open(self.base_url,self.pagetitle)
    
26. 
    
27.     def find_element(self,*loc):  #*loc任意数量的位置参数（带单个星号参数）
    
28.         # return self.driver.find_element(*loc)
    
29.         try:
    
30.             WebDriverWait(self.driver,10).until(EC.visibility_of_element_located(loc))
    
31.             return self.driver.find_element(*loc)
    
32.         except:
    
33. 
    
34.             print("%s 页面未能找到 %s 元素"%(self,loc))
    
35. 
    
36.     def script(self,src):
    
37.         self.driver.excute_script(src)
    
38. 
    
39.     def send_keys(self, loc, vaule, clear_first=True, click_first=True):
    
40.         try:
    
41.             loc = getattr(self, "_%s" % loc)  # getattr相当于实现self.loc
    
42.             if click_first:
    
43.                 self.find_element(*loc).click()
    
44.             if clear_first:
    
45.                 self.find_element(*loc).clear()
    
46.                 self.find_element(*loc).send_keys(vaule)
    
47.         except AttributeError:
    
48.             print("%s 页面中未能找到 %s 元素" % (self, loc))

复制代码

测试脚本的优化

BasePage.py提取完毕，其中设计了BasePage类，对一些webdriver的方法进行了二次封装。

baidu.py基于BasePage.py进行优化（充分体现PO的设计思想，封装、继承）

1. # 基本测试场景
   
2. # from selenium import  webdriver
   
3. # from time import sleep
   
4. #
   
5. # driver = webdriver.Firefox()
   
6. # driver.get("http://www.baidu.com")
   
7. #
   
8. # driver.find_element_by_xpath("//input[@id='kw']").send_keys("Bela") #输入框
   
9. # driver.find_element_by_xpath("//input[@id='su']").click() #百度一下按钮
   
10. #
    
11. # sleep(3)
    
12. # driver.quit()
    
13. 
    
14. # 优化后的测试场景
    
15. from selenium.webdriver.common.by import By
    
16. from PODemo.BasePage import BasePage  #假设baidu.py、BasePage.py均在PODemo.BasePage目录下
    
17. from selenium import webdriver
    
18. 
    
19. class SearchPage(BasePage):
    
20. 
    
21.     # 定位元素
    
22.     search_loc = (By.ID,"kw")
    
23.     btn_loc = (By.ID,"su")
    
24. 
    
25.     def open(self):
    
26.         self._open(self.base_url)
    
27. 
    
28.     def search_content(self,content):
    
29.         BaiduContent = self.find_element(*self.search_loc)
    
30.         BaiduContent.send_keys(content)
    
31. 
    
32.     def btn_click(self):
    
33.         BaiduBtn = self.find_element(*self.btn_loc)
    
34.         BaiduBtn.click()

复制代码

PageObject总结

• PO设计模式中的BasePage基类对应案例中的BasePage.py文件。
• PO模式中的page1或pageN对应案例中的Search.py
• PO设计模式中TestCase对应案例中的TestCase.py