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 ....
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页面中的对象组件定义进行更改即可。
联系客服