selenium +python之Page Obiect设计模式

 1 from selenium import webdriver
 2 from selenium.webdriver.common.by import By
 3 from time import sleep
 4 
 5 
 6 class Page(object):
 7     """
 8     基础类，用于页面对象类的继承
 9     """
10     login_url = ‘http://mail.163.com/‘
11 
12     def __init__(self, selenium_driver, base_url=login_url):
13         self.base_url = base_url
14         self.driver = selenium_driver
15         self.timeout = 30
16 
17     def on_page(self):
18         return self.driver.current_url == (self.base_url + self.url)
19 
20     def _open(self, url):
21         url = self.base_url + url
22         self.driver.get(url)
23         assert self.on_page(), ‘Did not land on %s‘ % url
24 
25     def open(self):
26         self._open(self.url)
27 
28     def find_element(self, *loc):
29         return self.driver.find_element(*loc)
30 
31 
32 class LoginPage(Page):
33     """
34     126邮箱登陆页面模型
35     """
36     url = ‘/‘
37     # 定位器
38     username_loc = (By.ID, "idInput")
39     password_loc = (By.ID, "pwdInput")
40     submit_loc = (By.ID, "loginBtn")
41 
42     # Action
43     def type_username(self, username):
44         self.find_element(*self.username_loc).send_keys(username)
45 
46     def type_password(self, password):
47         self.find_element(*self.password_loc).send_keys(password)
48 
49     def submit(self):
50         self.find_element(*self.submit_loc).click()
51 
52 
53 def test_user_login(driver, username, password):
54     """
55      测试获取的用户名/密码是否可以登陆
56      """
57     login_page = LoginPage(driver)
58     login_page.open()
59     login_page.type_username(username)
60     login_page.type_password(password)
61     login_page.submit()
62 
63 
64 def main():
65     try:
66         driver = webdriver.Chrome()
67         username = ‘fengyiru6369@163.com‘
68         password = ‘fyr19910915‘
69         test_user_login(driver, username, password)
70         sleep(3)
71         text = driver.find_element_by_xpath("//span[@id = ‘spnUid‘]").text
72         assert (text == ‘fengyiru6369@163.com‘), "用户名称不匹配，登陆失败！"
73     finally:
74         # 关闭浏览器窗口
75         driver.close()
76 
77 
78 if __name__ == ‘__main__‘:
79     main()