def test_find_with_link_synonym(self):
finder = ElementFinder()
browser = mock()
when(browser).get_current_url().thenReturn("http://localhost/mypage.html")
finder.find(browser, "test1", tag='link')
verify(browser).find_elements_by_xpath(
"//a[(@id='test1' or @name='test1' or @href='test1' or normalize-space(descendant-or-self::text())='test1' or @href='http://localhost/test1')]")
def test_find_with_locator_with_quote_and_apos(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test \"1\" and '2'")
verify(browser).find_elements_by_xpath(
"//*[(@id=concat('test \"1\" and ', \"'\", '2', \"'\", '') or @name=concat('test \"1\" and ', \"'\", '2', \"'\", ''))]"
)
def test_find_by_id_with_synonym_and_constraints(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'input', 'span', 'input',
'a', 'input', 'div', 'input')
elements[1].set_attribute('type', 'radio')
elements[3].set_attribute('type', 'checkbox')
elements[5].set_attribute('type', 'text')
elements[7].set_attribute('type', 'file')
when(browser).find_elements_by_id("test1").thenReturn(elements)
result = finder.find(browser, "id=test1")
self.assertEqual(result, elements)
result = finder.find(browser, "id=test1", tag='input')
self.assertEqual(result,
[elements[1], elements[3], elements[5], elements[7]])
result = finder.find(browser, "id=test1", tag='radio button')
self.assertEqual(result, [elements[1]])
result = finder.find(browser, "id=test1", tag='checkbox')
self.assertEqual(result, [elements[3]])
result = finder.find(browser, "id=test1", tag='text field')
self.assertEqual(result, [elements[5]])
result = finder.find(browser, "id=test1", tag='file upload')
self.assertEqual(result, [elements[7]])
def test_find_with_text_field_synonym(self):
finder = ElementFinder()
browser = mock()
when(browser).get_current_url().thenReturn("http://localhost/mypage.html")
finder.find(browser, "test1", tag='text field')
verify(browser).find_elements_by_xpath(
"//input[@type='text' and (@id='test1' or @name='test1' or @value='test1' or @src='test1' or @src='http://localhost/test1')]")
def test_find_with_invalid_prefix(self):
finder = ElementFinder()
browser = mock()
assert_raises_with_msg(ValueError, "Element locator with prefix 'something' is not supported",
finder.find, browser, "something=test1")
assert_raises_with_msg(ValueError, "Element locator with prefix ' by ID ' is not supported",
finder.find, browser, " by ID =test1")
def test_find_with_button(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test1", tag='button')
verify(browser).find_elements_by_xpath(
"//button[(@id='test1' or @name='test1' or @value='test1' or normalize-space(descendant-or-self::text())='test1')]"
)
def test_find_with_img(self):
finder = ElementFinder()
browser = mock()
when(browser).get_current_url().thenReturn("http://localhost/mypage.html")
finder.find(browser, "test1", tag='img')
verify(browser).find_elements_by_xpath(
"//img[(@id='test1' or @name='test1' or @src='test1' or @alt='test1' or @src='http://localhost/test1')]")
def _find_by_model(self, browser, criteria, tag, constraints):
prefixes = ['ng-', 'ng_', 'data-ng-', 'x-ng-']#, 'ng\\:']
for prefix in prefixes:
selector = '[%smodel="%s"]' % (prefix, criteria)
elements = browser.execute_script("""return document.querySelectorAll('%s');""" % selector);
if len(elements):
return ElementFinder()._filter_elements(elements, tag, constraints)
raise ValueError("Element locator '" + criteria + "' did not match any elements.")
def test_find_with_input(self):
finder = ElementFinder()
browser = mock()
browser.current_url = "http://localhost/mypage.html"
finder.find(browser, "test1", tag='input')
verify(browser).find_elements_by_xpath(
"//input[(@id='test1' or @name='test1' or @value='test1' or @src='test1' or @src='http://localhost/test1')]"
)
def test_find_with_file_upload_synonym(self):
finder = ElementFinder()
browser = mock()
browser.current_url = "http://localhost/mypage.html"
finder.find(browser, "test1", tag='file upload')
verify(browser).find_elements_by_xpath(
"//input[@type='file' and (@id='test1' or @name='test1' or @value='test1' or @src='test1' or @src='http://localhost/test1')]"
)
def test_find_with_a(self):
finder = ElementFinder()
browser = mock()
browser.current_url = "http://localhost/mypage.html"
finder.find(browser, "test1", tag='a')
verify(browser).find_elements_by_xpath(
"//a[(@id='test1' or @name='test1' or @href='test1' or normalize-space(descendant-or-self::text())='test1' or @href='http://localhost/test1')]"
)
def test_find_with_sloppy_criteria(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_id("test1").thenReturn(elements)
result = finder.find(browser, "id= test1 ")
self.assertEqual(result, elements)
def test_find_with_explicit_default_strategy_and_equals(self):
finder = ElementFinder()
browser = mock()
when(browser).get_current_url().thenReturn("http://localhost/mypage.html")
finder.find(browser, "default=page.do?foo=bar", tag='a')
verify(browser).find_elements_by_xpath(
"//a[(@id='page.do?foo=bar' or @name='page.do?foo=bar' or @href='page.do?foo=bar' or " +
"normalize-space(descendant-or-self::text())='page.do?foo=bar' or " +
"@href='http://localhost/page.do?foo=bar')]")
def test_find_by_dom(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).execute_script("return document.getElementsByTagName('a');").thenReturn(
[elements[1], elements[3]])
result = finder.find(browser, "dom=document.getElementsByTagName('a')")
self.assertEqual(result, [elements[1], elements[3]])
def test_find_with_invalid_prefix(self):
finder = ElementFinder()
browser = mock()
try:
self.assertRaises(ValueError, finder.find, browser,
"something=test1")
except ValueError as e:
self.assertEqual(
e.args[0],
"Element locator with prefix 'something' is not supported")
def test_find_by_css_selector(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_css_selector("#test1").thenReturn(elements)
result = finder.find(browser, "css=#test1")
self.assertEqual(result, elements)
result = finder.find(browser, "css=#test1", tag='a')
self.assertEqual(result, [elements[1], elements[3]])
def test_find_by_tag_name(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_tag_name("div").thenReturn(elements)
result = finder.find(browser, "tag=div")
self.assertEqual(result, elements)
result = finder.find(browser, "tag=div", tag='a')
self.assertEqual(result, [elements[1], elements[3]])
def test_find_by_partial_link_text(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_partial_link_text("my link").thenReturn(elements)
result = finder.find(browser, "partial link=my link")
self.assertEqual(result, elements)
result = finder.find(browser, "partial link=my link", tag='a')
self.assertEqual(result, [elements[1], elements[3]])
def test_find_by_xpath(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_xpath("//*[(@test='1')]").thenReturn(elements)
result = finder.find(browser, "xpath=//*[(@test='1')]")
self.assertEqual(result, elements)
result = finder.find(browser, "xpath=//*[(@test='1')]", tag='a')
self.assertEqual(result, [elements[1], elements[3]])
def _find_by_ng_repeater(self, browser, criteria, tag, constraints):
repeater_row_col = self._parse_ng_repeat_locator(criteria)
js_repeater_str = self._reconstruct_js_locator(repeater_row_col)
elements = browser.execute_script(
js_repeater_min +
"""var ng_repeat = new byRepeaterInner(true);""" +
"""return ng_repeat%s.getElements();""" % (js_repeater_str), arg0,
arg1)
if len(elements):
return ElementFinder()._filter_elements(elements, tag, constraints)
else:
raise ValueError("Element locator '" + criteria +
"' did not match any elements.")
def test_find_by_identifier(self):
finder = ElementFinder()
browser = mock()
id_elements = self._make_mock_elements('div', 'a')
name_elements = self._make_mock_elements('span', 'a')
when(browser).find_elements_by_id("test1").thenReturn(list(id_elements)).thenReturn(list(id_elements))
when(browser).find_elements_by_name("test1").thenReturn(list(name_elements)).thenReturn(list(name_elements))
all_elements = list(id_elements)
all_elements.extend(name_elements)
result = finder.find(browser, "identifier=test1")
self.assertEqual(result, all_elements)
result = finder.find(browser, "identifier=test1", tag='a')
self.assertEqual(result, [id_elements[1], name_elements[1]])
def test_find_returns_bad_values(self):
finder = ElementFinder()
browser = mock()
# selenium.webdriver.ie.webdriver.WebDriver sometimes returns these
for bad_value in (None, {'': None}):
for func_name in ('find_elements_by_id', 'find_elements_by_name',
'find_elements_by_xpath', 'find_elements_by_link_text',
'find_elements_by_css_selector', 'find_elements_by_tag_name'):
when_find_func = getattr(when(browser), func_name)
when_find_func(any()).thenReturn(bad_value)
for locator in ("identifier=it", "id=it", "name=it", "xpath=//div",
"link=it", "css=div.it", "tag=div", "default"):
result = finder.find(browser, locator)
self.assertEqual(result, [])
result = finder.find(browser, locator, tag='div')
self.assertEqual(result, [])
def test_find_with_sloppy_prefix(self):
finder = ElementFinder()
browser = mock()
elements = self._make_mock_elements('div', 'a', 'span', 'a')
when(browser).find_elements_by_id("test1").thenReturn(elements)
when(browser).find_elements_by_partial_link_text("test1").thenReturn(elements)
result = finder.find(browser, "ID=test1")
self.assertEqual(result, elements)
result = finder.find(browser, "iD=test1")
self.assertEqual(result, elements)
result = finder.find(browser, "id=test1")
self.assertEqual(result, elements)
result = finder.find(browser, " id =test1")
self.assertEqual(result, elements)
result = finder.find(browser, " partiallink =test1")
self.assertEqual(result, elements)
result = finder.find(browser, " p art iallin k =test1")
self.assertEqual(result, elements)
def test_find_with_list_synonym(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test1", tag='list')
verify(browser).find_elements_by_xpath(
"//select[(@id='test1' or @name='test1')]")
def test_find_with_locator_with_quote(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test \"1\"")
verify(browser).find_elements_by_xpath(
"//*[(@id='test \"1\"' or @name='test \"1\"')]")
def test_find_with_locator_with_apos(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test '1'")
verify(browser).find_elements_by_xpath(
"//*[(@id=\"test '1'\" or @name=\"test '1'\")]")
def test_find_with_tag(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "test1", tag='div')
verify(browser).find_elements_by_xpath(
"//div[(@id='test1' or @name='test1')]")
def test_find_with_explicit_default_strategy(self):
finder = ElementFinder()
browser = mock()
finder.find(browser, "default=test1")
verify(browser).find_elements_by_xpath(
"//*[(@id='test1' or @name='test1')]")
def test_find_with_empty_locator(self):
finder = ElementFinder()
browser = mock()
self.assertRaises(AssertionError, finder.find, browser, "")
def test_find_with_null_browser(self):
finder = ElementFinder()
self.assertRaises(AssertionError, finder.find, None, "id=test1")
