Page Object Model | POM (Selenium Webdriver For Python)

 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)

 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()