Page Object设计模式

Page Object设计模式的优点如下：

1、减少代码的重复

2、提高测试用例的可读性

3、提高测试用例的可维护性，特别是针对UI频繁变化的项目

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 __init__(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_username(self,username):
        self.find_element(*self.username_loc).send_keys(username)

    def type_password(self,password):
        self.find_element(*self.password_loc).send_keys(password)

    def submit(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_password(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='spanUid']").text
        assert(text == 'username@126.com'),"用户名称登录不匹配，登录失败"
    finally:
        driver.close()


if __name__ == '__main__':
    main()