from selenium.webdriver.support.ui import WebDriverWait
classBasePageElement(object): """Base page class that is initialized on every page object class."""
def__set__(self, obj, value): """Sets the text to the value supplied""" 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): """Gets the text of the specified object""" 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")
defclick_go_button(self): """触发搜索""" element = self.driver.find_element(*MainPageLocators.GO_BUTTON) element.click()
classSearchResultsPage(BasePage): """Search results page action methods come here"""
defis_results_found(self): # Probably should search for this text in the specific page # element, but as for now it works fine return"No results found."notin self.driver.page_source