def test_take_multiple_element_screenshots_with_same_name(self):
"""
The expectation is that they overwrite each other
"""
page = GooglePage(driver=self.driver, url="http://www.google.com")
page.open().wait_for_page_load()
page.element = Element(self.driver, Locator.by_xpath("//foo"))
num_files_before = self.count_files(seleniumconfig.screenshot_dir)
time.sleep(1)
file1 = page.element.take_screenshot()
time.sleep(2)
file2 = page.element.take_screenshot(screenshot_name=file1)
file3 = page.element.take_screenshot(screenshot_name=file1)
filepath = "{}/{}.png".format(seleniumconfig.screenshot_dir, file1)
time.sleep(3)
num_files_after = self.count_files(seleniumconfig.screenshot_dir)
assert file1 == file2 and file2 == file3, "I'm expecting the files to the same if taken really fast - file1: {} - file2: {} - file3: {}".format(
file1, file2, file3)
assert os.path.exists(
filepath
), "Expecting that a screenshot was taken and saved here: {}".format(
filepath)
assert num_files_after == (
num_files_before +
1), "Expecting there to be 1-more screenshot taken"
def test_take_explicit_webpage_screenshot(self):
page = GooglePage(driver=self.driver, url="http://www.google.com")
page.open().wait_for_page_load()
file = page.take_screenshot()
filepath = "{}/{}.png".format(seleniumconfig.screenshot_dir, file)
assert os.path.exists(
filepath
), "Expecting that a screenshot was taken and saved here: {}".format(
filepath)
def test_webpage_with_multiple_elements(self):
google = GooglePage(self.driver, "http://www.google.com")
wikipedia = Wikipedia(self.driver,
"https://en.wikipedia.org/wiki/Selenium")
elements = google.get_element_attr()
self.assert_elements(elements, expected_count=1)
elements = wikipedia.get_element_attr()
self.assert_elements(elements, expected_count=4)
elements = wikipedia.get_element_attr(type=Button)
self.assert_elements(elements, expected_count=0)
elements = wikipedia.get_element_attr(type=TextElement)
self.assert_elements(elements, expected_count=3)
def test_take_implicit_element_screenshot_enabled_is_visible(self):
page = GooglePage(driver=self.driver, url="http://www.google.com")
page.open().wait_for_page_load()
page.element = Element(self.driver, Locator.by_xpath("//foo"))
num_files_before = self.count_files(seleniumconfig.screenshot_dir)
try:
page.element.is_visible(timeout=1)
except TimeoutException:
pass
except NoSuchElementException:
pass
num_files_after = self.count_files(seleniumconfig.screenshot_dir)
assert num_files_after == num_files_before, "Expecting there to be the same number of screenshots taken"
def test_take_implicit_element_screenshot_enabled_get_web_element(self):
page = GooglePage(driver=self.driver, url="http://www.google.com")
page.open().wait_for_page_load()
page.element = Element(self.driver, Locator.by_xpath("//foo"))
num_files_before = self.count_files(seleniumconfig.screenshot_dir)
try:
page.element.get_webelement()
except TimeoutException:
pass
except NoSuchElementException:
pass
num_files_after = self.count_files(seleniumconfig.screenshot_dir)
assert num_files_after == (
num_files_before +
1), "Expecting there to be 1-more screenshot taken"
def test_take_implicit_element_screenshot_disabled(self):
page = GooglePage(driver=self.driver, url="http://www.google.com")
page.open().wait_for_page_load()
page.element = Element(self.driver, Locator.by_xpath("//foo"))
num_files_before = self.count_files(seleniumconfig.screenshot_dir)
# Disable taking screenshot globally
seleniumconfig.screenshot_enabled = False
try:
page.element.get_webelement()
except TimeoutException:
pass
except NoSuchElementException:
pass
finally:
seleniumconfig.screenshot_enabled = True
num_files_after = self.count_files(seleniumconfig.screenshot_dir)
assert num_files_after == num_files_before, "Expecting there to be the same number of screenshots taken"
def setup_class(self):
super(TestGoogle, self).setup_class()
# Capability for Firefox
# self.capabilities = webdriver.DesiredCapabilities.FIREFOX
# self.capabilities['marionette'] = True
# Capability for HtmlUnit
# TODO : There is a potential issue reported with HTMLUnit. See this bug as of January 2017 - https://sourceforge.net/p/htmlunit/bugs/1846/
# self.capabilities = webdriver.DesiredCapabilities.HTMLUNIT
# self.capabilities = webdriver.DesiredCapabilities.HTMLUNITWITHJS
# self.capabilities['driverName'] = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.11; rv:51.0) Gecko/20100101 Firefox/51.0"
# Capability for PhantomJS
# self.capabilities = webdriver.DesiredCapabilities.PHANTOMJS
try:
self.google = GooglePage(self.driver, url=self.google_url)
self.wikipedia = Wikipedia(self.driver, url=self.wikipedia_url)
except:
if self.driver:
self.driver.quit()
class TestGoogle(UiTestWrapper):
"""
These are proof-of-concept Selenium programs that leverage the Selenium PageModel library. All these programs
are based around performing searches on Google.
"""
google_url = 'https://www.google.com'
wikipedia_url = 'https://en.wikipedia.org/wiki/Selenium'
capabilities = None
google = None
wikipedia = None
@classmethod
def setup_class(self):
super(TestGoogle, self).setup_class()
# Capability for Firefox
# self.capabilities = webdriver.DesiredCapabilities.FIREFOX
# self.capabilities['marionette'] = True
# Capability for HtmlUnit
# TODO : There is a potential issue reported with HTMLUnit. See this bug as of January 2017 - https://sourceforge.net/p/htmlunit/bugs/1846/
# self.capabilities = webdriver.DesiredCapabilities.HTMLUNIT
# self.capabilities = webdriver.DesiredCapabilities.HTMLUNITWITHJS
# self.capabilities['driverName'] = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.11; rv:51.0) Gecko/20100101 Firefox/51.0"
# Capability for PhantomJS
# self.capabilities = webdriver.DesiredCapabilities.PHANTOMJS
try:
self.google = GooglePage(self.driver, url=self.google_url)
self.wikipedia = Wikipedia(self.driver, url=self.wikipedia_url)
except:
if self.driver:
self.driver.quit()
def test_search(self):
"""This is the traditional 'HelloWorld' program you see in Selenium documentation, but implemented
Using the Selenium PageModel library"""
search_term = 'Cheese!'
# And now use this to visit Google
self.google.open().wait_for_page_load().validate()
# Check duration
assert self.google.page_timeout > 0, "Expecting page_timeout > 0 - actual: {}".format(
self.google.page_timeout)
# Check the title of the page
title = self.google.get_title()
url = self.google.get_current_url()
expected_url = 'https://www.google.com/'
# Should see: "cheese! - Google Search"
print "Page title is: {}".format(title)
print "Url: '{}'".format(url)
assert url == expected_url, "Expecting url to be '{}'".format(
expected_url)
# Enter something to search for
inner_html = self.google.search_field.get_html()
print "search_field inner-html: '{}'".format(inner_html)
assert len(
inner_html
) == 0, "Expecting there to be no html in the search_field element"
self.google.search_field.clear().type(search_term)
# Now submit the form. WebDriver will find the form for us from the element
self.google.search_field.submit()
# Check the title of the page
title = self.google.wait_for_title(search_term).get_title()
url = self.google.current_url
expected_url = 'https://www.google.com/search?hl=en&source=hp&biw=&bih=&q=Cheese%21&gbv=2&oq=&gs_l='
print "Page title is: {}".format(title)
print "Url: '{}'".format(url)
assert search_term in title, "Expected '{}' in '{}'".format(
search_term, title)
assert url == expected_url, "Expecting url to be '{}'".format(
expected_url)
self.google.validate()
def test_get_result_links(self):
"""This simply verifies that we are able to grab the links from a google search
"""
search_term = "Selenium"
self.google.open().wait_for_page_load().validate()
self.google.search_field.clear().type(search_term)
self.google.search_field.submit()
self.google.wait_for_title(search_term)
links = self.google.get_result_links()
assert len(links) > 5
print "found {} links".format(len(links))
for link in links:
print "- '{}'".format(link)
def test_search_for_10_words_from_wikipedia(self):
"""
This test does the following:
1) Go to Wikipedia and grab 10 random words
2) For each word, go to google and grab the top-5 links
"""
self.wikipedia.open().wait_for_page_load().validate()
# Check duration
assert self.wikipedia.page_timeout > 0, "Expecting page_timeout > 0 - actual: {}".format(
self.wikipedia.page_timeout)
# Testing inner-html
inner_html = self.wikipedia.bodyText.get_html()
# print "bodyText inner-html: '{}'".format(inner_html)
assert inner_html and len(inner_html) > 100, "Expecting there to be quite a bit " \
"of html code in inner_html"
wikipedia_text = self.wikipedia.get_text()
assert wikipedia_text
word_list = wikipedia_text.split()
random_words = []
# Get a list of 10 random words
while len(random_words) < 5:
randInt = random.randint(0, len(word_list) - 1)
word = word_list[randInt].lower()
# Basic check to ensure that word is not already in the list
if word in random_words:
next
random_words.append(word)
# Go to Google and perform search on each word
for search_term in random_words:
self.google.open().wait_for_page_load().validate()
self.google.search_field.clear().type(search_term)
self.google.search_field.submit()
self.google.wait_for_title(search_term)
# Grab the first 5 urls
links = self.google.get_result_links()
assert len(links) >= 5
for i, link in enumerate(links[0:5]):
print "[{}] '{}' - {}".format(i, search_term, link)