数据驱动+PageObject模式

练习要求：

126 邮箱登录 答案：

from selenium import webdriver
from selenium.webdriver.common.by import By
from time import sleep
class Page(object):
    login_url='http://www.126.com'
    
    def __int__(self,selenium_driver,base_url=login_url):
        self.base_url=base_url
        self.driver=selenium_driver
        self.timeout=30

    def on_page(self):
        return self.driver.current_url ==(self.base_url+self.url)
    def _open(self,url):
        url=self.base_url+url
        self.driver.get(url)
        assert self.on_page(),'Did not land on %s' % url
    def open(self):
        self._open(self.url)
    def find_element(self,*loc):
    return self.driver.find_element(*loc)

class LoginPage(Page):
    url='/'
    username_loc=(By.ID,"idInput")
    password_loc=(By.ID,"pwdInput")
    submit_loc=(By.ID,"loginBtn")
    def type_password(self,password):
        self.find_element(*self.username_loc).send_keys(username)
    def type_password(self,password):
        self.find_element(*self.password_loc).send_keys(password)
    def type_username(self):
        self.find_element(*self.submit_loc).click()

def test_user_login(driver,username,password):
    login_page=LoginPage(driver)
    login_page.open()
    login_page.type_username(username)
    login_page.type_password(password)
    login_page.submit()

def main():
    try:
        driver=webdriver.Firefox()
        username='username'
        password='123456'
        test_user_login(driver,username,password)
        sleep(3)
        text=driver.find_element_by_xpath("//span[@id='spnUid'").text
        assert(text=='username@126.com'),"用户名称不匹配，登录失败！"
        username='username2'
        password='123456'
        test_user_login(driver,username,password)
        sleep(3)
        text=driver.find_element_by_xpath("//span[@id='spnUid'").text
        assert(text=='username2@126.com'),"用户名称不匹配，登录失败！"
    finally:
        #关闭浏览器窗口
        driver.close()
if __name__='__main__':
    main()