Page Object Model 简称POM
普通的测试用例代码:
....#测试用例 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("liuke01") driver.find_element_by_id("pwdInput").clear() driver.find_element_by_id("pwdInput").send_keys("liuke123") driver.find_element_by_id("loginBtn").click() ....
改造:
首先,我们要分离测试对象(元素对象)和测试脚本(用例脚本),那么我们分别创建两个脚本文件, LoginPage.py 用于定义页面元素对象,每一个元素都封装成组件(可以看做存放页面元素对象的仓库) CaseLoginTest.py 测试用例脚本。我们的实现思想,一切元素和元素的操作组件化定义在Page页面,用例脚本页面,通过调用Page中的组件对象,进行拼凑成一个登录脚本。
在写这两个脚本之前,我先对WebDriver中的一些方法进行重定义,以方便我们在写PO的时候,更简洁,快速。
BasePage.py:
#-*- coding: utf-8 -*-from selenium.webdriver.support.wait importWebDriverWaitfrom 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:
#-*- coding: utf-8 -*-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")#Actiondef 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 -*-import sys reload(sys) sys.setdefaultencoding('utf-8')import unittest from PO importLoginPagefrom selenium import webdriver classCaselogin126mail(unittest.TestCase):"""登录126邮箱的case """@classmethoddef setUpClass(cls): cls.driver = webdriver.Chrome() cls.driver.implicitly_wait(30) cls.url ="http://mail.126.com" cls.username ="liuke01" cls.password ="liuke123"#用例执行体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() @classmethoddef tearDownClass(cls): cls.driver.quit()if __name__ =="__main__": unittest.main()
通过使用POM进行重新构造代码结构后,发现代码测试用例代码的可读性提高很多,元素写成组件的方式,不需要每次都写findElement直接在脚本中调用组件就可以使用。在CaseLoginTest脚本用例执行体中,一旦我们输入 login_page并敲入一个点时,LoginPage页面中的元素对象组件都显示出来。并且定义好的PageObject组件可以重复在其它的脚本中进行使用,减少了代码的工作量,也方便对脚本进行后期的维护管理,当元素属性发生变化时,我们只需要对一个PageObaject页面中的对象组件定义进行更改即可。