Selenium PageObject设计模式

from  time import sleep

class Page():
    '''页面基础类'''

    #初始化
    def __init__(self, dirver):
        self.base_url = 'http://localhost'
        self.driver = dirver
        self.timeout = 30

    #打开不同的子页面
    def _open(self, url):
        url_ = self.base_url + url
        print("Test page is： %s" %url_)
        self.driver.maximize_window()
        self.driver.get(url_)
        sleep(2)
        assert self.driver.current_url == url_, 'Did ont land on %s' % url_

    def open(self):
        self._open(self.url)

    #元素定位方法封装
    def find_element(self,*loc):
        return self.driver.find_element(*loc)

from BasePage import *
from selenium.webdriver.common.by import By

class LoginPage(Page):
    '''首页登录页面'''

    url='/'

    #定位器
    username_loc=(By.NAME,'username')
    password_loc=(By.NAME,'password')
    submit_loc=(By.NAME,'Submit')

    #用户名输入框元素
    def type_username(self,username):
        self.find_element(*self.username_loc).clear()
        self.find_element(*self.username_loc).send_keys(username)

    #密码输入框元素
    def type_password(self,password):
        self.find_element(*self.password_loc).clear()
        self.find_element(*self.password_loc).send_keys(password)

    #登录按钮元素
    def type_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_username(username)
    login_page.type_password(password)
    login_page.type_submit()

from LoginPage import *
from selenium import webdriver

driver=webdriver.Firefox()

username = '51zxw'
password = '123456'
test_user_login(driver, username, password)

sleep(3)
driver.quit()