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源码:(源码会随着结构调整不断更新,但是整体结构不会变动)
Page Object Model 简称POM
下面通过一个实例演示POM的实现
在此之前,先看一个段比较普通的测试用例代码片段:
-
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 #-*- 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:
-
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理解有所帮助>