81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
# TODO(dkorn): add handle_popup method
|
|
|
|
import selenium.common.exceptions as Exceptions
|
|
import selenium.webdriver.support.ui as Support
|
|
|
|
|
|
class PageObject(object):
|
|
"""Base class for page objects."""
|
|
def __init__(self, driver, conf):
|
|
"""Constructor."""
|
|
self.driver = driver
|
|
self.conf = conf
|
|
self.login_url = self.conf.dashboard.login_url
|
|
self._page_title = None
|
|
|
|
@property
|
|
def page_title(self):
|
|
return self.driver.title
|
|
|
|
def is_the_current_page(self):
|
|
if self._page_title not in self.page_title:
|
|
raise AssertionError(
|
|
"Expected to find %s in page title, instead found: %s"
|
|
% (self._page_title, self.page_title))
|
|
return True
|
|
|
|
def get_url_current_page(self):
|
|
return self.driver.current_url()
|
|
|
|
def close_window(self):
|
|
return self.driver.close()
|
|
|
|
def go_to_login_page(self):
|
|
self.driver.get(self.login_url)
|
|
self.is_the_current_page()
|
|
|
|
def is_element_present(self, *locator):
|
|
try:
|
|
self.driver.find_element(*locator)
|
|
return True
|
|
except Exceptions.NoSuchElementException:
|
|
return False
|
|
|
|
def is_element_visible(self, *locator):
|
|
try:
|
|
return self.driver.find_element(*locator).is_displayed()
|
|
except (Exceptions.NoSuchElementException,
|
|
Exceptions.ElementNotVisibleException):
|
|
return False
|
|
|
|
def return_to_previous_page(self):
|
|
self.driver.back()
|
|
|
|
def get_element(self, *locator):
|
|
return self.driver.find_element(*locator)
|
|
|
|
def fill_field_element(self, data, field_element):
|
|
field_element.clear()
|
|
field_element.send_keys(data)
|
|
return field_element
|
|
|
|
def select_dropdown(self, value, element):
|
|
select = Support.Select(element)
|
|
select.select_by_visible_text(value)
|
|
|
|
def select_dropdown_by_value(self, value, element):
|
|
select = Support.Select(element)
|
|
select.select_by_value(value)
|