class WebDriver:
log = Log.func_logger()
def init_driver(self):
self.log.info("Opening in Chrome")
return webdriver.Chrome("/Users/edgarnav/Documents/chromedriver")
class ResultsPage(ElementsInteractions):
def __init__(self, driver):
super().__init__(driver)
self.driver = driver
log = Logger.func_logger()
url_batman = "https://www.tvmaze.com/shows/975/batman"
class_url = "white-text"
class_back_btn = "btn-primary"
def find_url_batman(self):
elements = self.get_all_elements(self.class_url, "class")
links = [element.get_attribute('href') for element in elements]
index = links.index(self.url_batman)
elements[index].click()
def return_page(self):
self.back_page()
def click_back_btn(self):
self.click_element(self.class_back_btn, "class")
class ElementsInteractions:
log = Log.func_logger()
def __init__(self, driver):
self.driver = driver
def locator(self, locator_type):
if locator_type == "id":
return By.ID
elif locator_type == "name":
return By.NAME
elif locator_type == "class":
return By.CLASS_NAME
elif locator_type == "xpath":
return By.XPATH
elif locator_type == "tag":
return By.TAG_NAME
else:
self.log.error("Locator Type : " + locator_type +
" entered is not found")
return False
def verify_activity(self, activity_name):
cont_max_time = 0
while cont_max_time > 10:
time.sleep(1)
cont_max_time += 1
if activity_name == self.driver.current_activity:
self.log.info("Activity name match with: " + activity_name)
break
elif activity_name != self.driver.current_activity and cont_max_time == 10:
self.take_screenshot(self.driver.activity_name)
self.log.info("Activity name expected: " + activity_name)
assert False
def explicit_wait(self, locator_value, locator_type, max_time):
try:
locator_by_type = self.locator(locator_type)
WebDriverWait(self.driver, max_time).until(
ec.presence_of_all_elements_located(
(locator_by_type, locator_value)))
self.log.info("Element found with locator " + locator_value +
" using locatorType " + locator_by_type)
return True
except Exception:
self.log.error("Element not found with locator " + locator_value +
" using locatorType " + locator_type)
return False
def get_element(self, locator_value, locator_type):
element = None
try:
locator_by_type = self.locator(locator_type)
element = self.driver.find_element(locator_by_type, locator_value)
self.log.info("Element found with locator " + locator_value +
" using locatorType " + locator_by_type)
except Exception:
self.log.error("Element not found with locator " + locator_value +
" using locatorType " + locator_type)
print_stack()
return element
def wait_element(self, locator_value, locator_type):
try:
locator_by_type = self.locator(locator_type)
wait = WebDriverWait(self.driver,
25,
poll_frequency=1,
ignored_exceptions=[
ElementNotVisibleException,
NoSuchElementException
])
element = wait.until(
ec.presence_of_element_located(
(locator_by_type, locator_value)))
self.log.info("Element found with locator value " + locator_value +
" using locatorType " + locator_type)
except Exception:
self.log.error("Element not found with locator value " +
locator_value + " using locatorType " +
locator_type)
print_stack()
self.take_screenshot(locator_type)
assert False
return element
def press_element(self, locator_value, locator_type):
try:
element = self.wait_element(locator_value, locator_type)
element.click()
self.log.info("Clicked on element with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error("Unable to Click on element with locator value " +
locator_value + " using locatorType " +
locator_type)
print_stack()
assert False
def send_text(self, locator_value, locator_type, text):
try:
element = self.wait_element(locator_value, locator_type)
element.send_keys(text)
self.log.info("Sent the text " + text +
" in element with locator value " + locator_value +
" using locatorType " + locator_type)
except Exception:
self.log.error("Unable to Sent the text " + text +
" in element with locator value " + locator_value +
"using locatorType " + locator_type)
print_stack()
self.take_screenshot(locator_type)
assert False
def get_text(self, locator_value, locator_type):
element_text = None
try:
element = self.wait_element(locator_value, locator_type)
element_text = element.text
self.log.info("Got the text " + element_text +
" from element with locator value " + locator_value +
" using locatorType " + locator_type)
except Exception:
self.log.error(
"Unable to get the text from element with locator value " +
locator_value + "using locatorType " + locator_type)
print_stack()
return element_text
def get_attribute(self, locator_value, locator_type, attribute_name):
attribute = None
try:
element = self.wait_element(locator_value, locator_type)
attribute = element.get_attribute(attribute_name)
self.log.info("Got the attribute " + attribute_name + " -> " +
attribute + " from element with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error("Unable to get the attribute " + attribute_name +
" from element with locator value " +
locator_value + "using locatorType " + locator_type)
print_stack()
return attribute
def is_element_displayed(self, locator_value, locator_type):
element_displayed = None
try:
element = self.wait_element(locator_value, locator_type)
element_displayed = element.is_displayed()
self.log.info(
"Element is Displayed on web page with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error(
"Element is not Displayed on web page with locator value " +
locator_value + " using locatorType " + locator_type)
print_stack()
return element_displayed
def take_screenshot(self, text):
pass
from driver_interactions.ElementsInteractions import ElementsInteractions
from driver_interactions.WebDriver import WebDriver
import utilities.Logger as Logger
import config.ConfigFile as ConfigFile
import time
log = Logger.func_logger()
def before_all(context):
log.info("Script started")
context.prepare_driver = WebDriver()
context.driver = context.prepare_driver.init_driver()
context.bp = ElementsInteractions(context.driver)
context.bp.launch_web_page(ConfigFile.url)
def after_all(context):
time.sleep(2)
context.driver.quit()
log.info("Script ended")
class ElementsInteractions:
log = Log.func_logger()
def __init__(self, driver):
self.driver = driver
def locator(self, locator_type):
if locator_type == "id":
return By.ID
elif locator_type == "name":
return By.NAME
elif locator_type == "class":
return By.CLASS_NAME
elif locator_type == "xpath":
return By.XPATH
elif locator_type == "css":
return By.CSS_SELECTOR
elif locator_type == "tag":
return By.TAG_NAME
elif locator_type == "link":
return By.LINK_TEXT
elif locator_type == "plink":
return By.PARTIAL_LINK_TEXT
else:
self.log.error("Locator Type : " + locator_type +
" entered is not found")
return False
def launch_web_page(self, url):
try:
self.driver.get(url)
self.log.info("Web Page Launched with URL : " + url)
except Exception:
self.log.info("Web Page not Launched with URL : " + url)
def go_to_url(self, url):
self.driver.get(url)
def verify_page(self, page_name):
if page_name != self.driver.title:
self.take_screenshot(self.driver.title)
assert False
def back_page(self):
self.driver.back()
def explicit_wait(self, locator_value, locator_type, time):
try:
locator_by_type = self.locator(locator_type)
WebDriverWait(self.driver, time).until(
ec.presence_of_all_elements_located(
(locator_by_type, locator_value)))
self.log.info("Element found with locator " + locator_value +
" using locatorType " + locator_by_type)
except Exception:
self.log.error("Element not found with locator " + locator_value +
" using locatorType " + locator_type)
print_stack()
self.take_screenshot(locator_type)
assert False
def get_element(self, locator_value, locator_type):
element = None
try:
locator_by_type = self.locator(locator_type)
element = self.driver.find_element(locator_by_type, locator_value)
self.log.info("Element found with locator " + locator_value +
" using locatorType " + locator_by_type)
except Exception:
self.log.error("Element not found with locator " + locator_value +
" using locatorType " + locator_type)
print_stack()
return element
def get_all_elements(self, locator_value, locator_type):
elements = None
try:
locator_by_type = self.locator(locator_type)
elements = self.driver.find_elements(locator_by_type,
locator_value)
self.log.info("Elements found with locator " + locator_value +
" using locatorType " + locator_by_type)
except Exception:
self.log.error("Elements not found with locator " + locator_value +
" using locatorType " + locator_type)
print_stack()
return elements
def wait_element(self, locator_value, locator_type):
try:
locator_by_type = self.locator(locator_type)
wait = WebDriverWait(self.driver,
25,
poll_frequency=1,
ignored_exceptions=[
ElementNotVisibleException,
NoSuchElementException
])
element = wait.until(
ec.presence_of_element_located(
(locator_by_type, locator_value)))
self.log.info("WebElement found with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error("WebElement not found with locator value " +
locator_value + " using locatorType " +
locator_type)
print_stack()
self.take_screenshot(locator_type)
assert False
return element
def click_element(self, locator_value, locator_type):
try:
element = self.wait_element(locator_value, locator_type)
element.click()
self.log.info("Clicked on WebElement with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error(
"Unable to Click on WebElement with locator value " +
locator_value + " using locatorType " + locator_type)
print_stack()
assert False
def send_text(self, text, locator_value, locator_type):
try:
element = self.wait_element(locator_value, locator_type)
element.send_keys(text)
self.log.info("Sent the text " + text +
" in WebElement with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error("Unable to Sent the text " + text +
" in WebElement with locator value " +
locator_value + "using locatorType " + locator_type)
print_stack()
self.take_screenshot(locator_type)
assert False
def get_text(self, locator_value, locator_type):
element_text = None
try:
element = self.wait_element(locator_value, locator_type)
element_text = element.text
self.log.info("Got the text " + element_text +
" from WebElement with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error("Unable to get the text " + element_text +
" from WebElement with locator value " +
locator_value + "using locatorType " + locator_type)
print_stack()
return element_text
def is_element_displayed(self, locator_value, locator_type):
element_displayed = None
try:
element = self.wait_element(locator_value, locator_type)
element_displayed = element.is_displayed()
self.log.info(
"WebElement is Displayed on web page with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error(
"WebElement is not Displayed on web page with locator value " +
locator_value + " using locatorType " + locator_type)
print_stack()
return element_displayed
def scroll(self, locator_value, locator_type):
actions = ActionChains(self.driver)
try:
element = self.wait_element(locator_value, locator_type)
actions.move_to_element(element).perform()
self.log.info("Scrolled to WebElement with locator value " +
locator_value + " using locatorType " + locator_type)
except Exception:
self.log.error(
"Unable to scroll to WebElement with locator value " +
locator_value + "using locatorType " + locator_type)
print_stack()
def take_screenshot(self, text):
allure.attach(self.driver.get_screenshot_as_png(),
name=text,
attachment_type=AttachmentType.PNG)
