Selenium关于Page Objects

import unittest
from selenium import webdriver
import page

class BaiduSearch(uinttest.TestCase):
    
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.get("http://www.baidu.com")

    def test_search_in_baidu(self):
        main_page = page.MainPage(self.driver)
        #断言标题关键字
        assert main_page.is_title_matches(), "baidu title doesn‘t match."
        #设置搜索的关键字
        main_page.search_text_element = "selenium"
        main_page.click_go_button()
        search_results_page = page.SearchResultsPage(self.driver)
        #断言结果页面不是空的
            assert search_results_page.is_result_found(), "No result found."

    def tearDown(self):
        self.driver.close()

if __name__ == "__main__":
    unittest.main()

from element import BasePageElement
from locators import MainPageLocators

class SearchTextElement(BasePageElement):
    locator = ‘kw‘

class BasePage(object):
    """被所有页面调用的页面初始化的基础类"""
    
    def __int__(self, driver):
        self.driver = driver

class MainPage(BasePage):
    """主页面的方法"""

    search_text_element = searchTextElement()

    def is_title_match(self):
        return "selenium" in self.driver.title

    def click_go_button(self):
        elemnt = self.driver.find_element(*MainPageLocators.GO_BUTTON)
    element.click()

class SearchsultsPage(BasePage):
    """搜索结果页面的方法"""
    
    def is_results_found(self):
        return "No results found." not  in self.driver.page_source

from selenium.webdriver.support.ui import WebDriverWait

class BasePageElement(object):
    def __set__(self, obj, vaule):
        driver = obj.driver
        WebDriverWait(driver, 100).until(
            lambda driver: driver.find_element_by_name(self.locator))
        driver.find_element_by_name(self.locator).send_keys(value)

    def __get__(self, obj, owner):
        driver = obj.driver
           WebDriverWait(driver, 100).until(
            lambda driver: driver.find_element_by_name(self.locator))
        element = driver.find_element_by_name(self.locator)
        return element.get_attribute("value")

from selenium.webdriver.common.by import By

class MainPageLocators(object): 
    """所有主页面定位到的元素"""
    GO_BUTTON = (By.ID, ‘su‘)

class SearchResultsPageLoactors(object):
    """所有搜索结果定位的元素"""
    pass