class TestWebElement(ElementTestCase, TestCase):
def get_elements(self):
self.root = WebElement('div')
self.tag = WebElement('span', parent=self.root)
self.df = DocumentFragment()
self.c1 = WebElement('c1')
self.c2 = WebElement('c2')
self.c3 = WebElement('c3')
self.c4 = WebElement('c4')
self.c1.textContent = 'child1'
self.c2.textContent = 'child2'
self.c3.textContent = 'child3'
self.c4.textContent = 'child4'
return self.root
def test_connection(self):
self.wait_until(lambda: self.root.connected)
self.assertTrue(self.root.connected)
self.browser.get('http://example.com/')
self.assertIsFalse(self.root.connected)
def test_text_content(self):
self.set_element(self.tag)
self.assertEqual(self.element.text, '')
self.tag.textContent = 'text'
self.wait_until(lambda: self.element.text == 'text')
self.assertEqual(self.element.text, 'text')
self.c1.textContent = 'child'
self.tag.appendChild(self.c1)
self.wait_until(lambda: self.element.text == 'textchild')
self.assertEqual(self.element.text, 'textchild')
self.tag.textContent = 'NewText'
self.wait_until(lambda: self.element.text == 'NewText')
self.assertEqual(self.element.text, 'NewText')
with self.assertRaises(NoSuchElementException):
self.set_element(self.c1, self.wait_time * 10)
self.set_element(self.tag)
t_node = Text('TextNode')
self.tag.replaceChild(t_node, self.tag.childNodes[0])
self.wait_until(lambda: self.element.text == 'TextNode')
self.assertEqual(self.element.text, 'TextNode')
self.tag.removeChild(self.tag.childNodes[0])
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == '')
self.assertEqual(self.element.text, '')
def test_attr(self):
self.set_element(self.tag)
self.assertIsNone(self.element.get_attribute('src'))
self.tag.setAttribute('src', 'a')
self.wait_until(lambda: self.element.get_attribute('src') == 'a')
self.assertEqual(self.element.get_attribute('src'), 'a')
self.tag.removeAttribute('src')
self.wait_until(lambda: self.element.get_attribute('src') is None)
self.assertIsNone(self.element.get_attribute('src'))
def test_addremove_child(self):
self.tag.appendChild(self.c1)
self.set_element(self.c1)
self.assertEqual(self.element.text, 'child1')
self.c1.textContent = 'Child'
self.wait_until(lambda: self.element.text == 'Child')
self.assertEqual(self.element.text, 'Child')
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == 'Child')
self.assertEqual(self.element.text, 'Child')
self.tag.removeChild(self.c1)
with self.assertRaises(NoSuchElementException):
self.wait(0.1)
self.set_element(self.c1, self.wait_time * 10)
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == '')
self.assertEqual(self.element.text, '')
def test_insert_child(self):
self.set_element(self.tag)
# test parent in constructor
self.c1 = WebElement('c1', parent=self.tag)
self.c1.textContent = 'child1'
self.assertIsTrue(self.set_element(self.c1))
with self.assertRaises(NoSuchElementException):
self.wait(0.1)
self.set_element(self.c2, self.wait_time * 10)
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == 'child1')
self.assertEqual(self.element.text, 'child1')
self.tag.insertBefore(self.c2, self.c1)
self.assertIsTrue(self.set_element(self.c2))
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == 'child2child1')
self.assertEqual(self.element.text, 'child2child1')
self.tag.empty()
self.wait_until(lambda: self.element.text == '')
self.assertEqual(self.element.text, '')
with self.assertRaises(NoSuchElementException):
self.set_element(self.c1, self.wait_time * 10)
with self.assertRaises(NoSuchElementException):
self.set_element(self.c2, self.wait_time * 10)
def test_add_df(self):
self.set_element(self.tag)
self.df.append(self.c1, self.c2, 'text')
self.tag.appendChild(self.df)
self.wait_until(lambda: self.element.text == 'child1child2text')
self.assertEqual(self.element.text, 'child1child2text')
df = DocumentFragment()
df.append(self.c3, 'text2')
self.tag.appendChild(df)
self.wait_until(
lambda: self.element.text == 'child1child2textchild3text2')
self.assertEqual(self.element.text, 'child1child2textchild3text2')
def test_insert_df(self):
self.set_element(self.tag)
self.tag.appendChild(self.c1)
self.df.append(self.c2, self.c3, 'text')
self.tag.insertBefore(self.df, self.c1)
self.wait_until(lambda: self.element.text == 'child2child3textchild1')
self.assertEqual(self.element.text, 'child2child3textchild1')
df = DocumentFragment()
df.append(self.c4, 'text2')
self.tag.insertBefore(df, self.c3)
self.wait_until(
lambda: self.element.text == 'child2child4text2child3textchild1')
self.assertEqual(self.element.text,
'child2child4text2child3textchild1')
def test_replace_child(self):
self.set_element(self.tag)
self.tag.appendChild(self.c1)
with self.assertRaises(NoSuchElementException):
self.set_element(self.c2, self.wait_time * 10)
self.assertIsTrue(self.set_element(self.c1))
self.wait_until(lambda: self.element.text == 'child1')
self.assertEqual(self.element.text, 'child1')
self.tag.replaceChild(self.c2, self.c1)
with self.assertRaises(NoSuchElementException):
self.wait(0.1)
self.set_element(self.c1, self.wait_time * 10)
self.assertIsTrue(self.set_element(self.c2))
self.wait_until(lambda: self.element.text == 'child2')
self.assertEqual(self.element.text, 'child2')
self.set_element(self.tag)
self.wait_until(lambda: self.element.text == 'child2')
self.assertEqual(self.element.text, 'child2')
def test_append(self):
self.set_element(self.tag)
self.tag.append(self.c1)
self.wait_until(lambda: self.element.text == 'child1')
self.assertEqual(self.element.text, 'child1')
self.tag.append(self.c2, self.c3)
self.wait_until(lambda: self.element.text == 'child1child2child3')
self.assertEqual(self.element.text, 'child1child2child3')
self.tag.append(self.c4, self.c1)
self.wait_until(
lambda: self.element.text == 'child2child3child4child1')
self.assertEqual(self.element.text, 'child2child3child4child1')
self.tag.append('t1', 't2')
self.wait_until(
lambda: self.element.text == 'child2child3child4child1t1t2')
self.assertEqual(self.element.text, 'child2child3child4child1t1t2')
def test_prepend(self):
self.set_element(self.tag)
self.tag.prepend(self.c1)
self.wait_until(lambda: self.element.text == 'child1')
self.assertEqual(self.element.text, 'child1')
self.tag.prepend(self.c2, self.c3)
self.wait_until(lambda: self.element.text == 'child2child3child1')
self.assertEqual(self.element.text, 'child2child3child1')
self.tag.prepend(self.c4, self.c1)
self.wait_until(
lambda: self.element.text == 'child4child1child2child3')
self.assertEqual(self.element.text, 'child4child1child2child3')
self.tag.prepend('t1', 't2')
self.wait_until(
lambda: self.element.text == 't1t2child4child1child2child3')
self.assertEqual(self.element.text, 't1t2child4child1child2child3')
def test_prepend_append_text(self):
self.set_element(self.tag)
self.tag.append('t1')
self.wait_until(lambda: self.element.text == 't1')
self.assertEqual(self.element.text, 't1')
self.tag.firstChild.remove()
self.wait_until(lambda: self.element.text == '')
self.assertEqual(self.element.text, '')
self.tag.prepend('t2')
self.wait_until(lambda: self.element.text == 't2')
self.assertEqual(self.element.text, 't2')
self.tag.append('t3', 't4')
self.wait_until(lambda: self.element.text == 't2t3t4')
self.assertEqual(self.element.text, 't2t3t4')
self.tag.prepend('t5', 't6')
self.wait_until(lambda: self.element.text == 't5t6t2t3t4')
self.assertEqual(self.element.text, 't5t6t2t3t4')
def test_after(self):
self.set_element(self.tag)
self.tag.append(self.c1)
self.c1.after(self.c2)
self.wait_until(lambda: self.element.text == 'child1child2')
self.assertEqual(self.element.text, 'child1child2')
self.c1.after(self.c3, self.c4)
self.wait_until(
lambda: self.element.text == 'child1child3child4child2')
self.assertEqual(self.element.text, 'child1child3child4child2')
self.c1.after(self.c2, 'text')
self.wait_until(
lambda: self.element.text == 'child1child2textchild3child4')
self.assertEqual(self.element.text, 'child1child2textchild3child4')
def test_before(self):
self.set_element(self.tag)
self.tag.append(self.c1)
self.c1.before(self.c2)
self.wait_until(lambda: self.element.text == 'child2child1')
self.assertEqual(self.element.text, 'child2child1')
self.c1.before(self.c3, self.c4)
self.wait_until(
lambda: self.element.text == 'child2child3child4child1')
self.assertEqual(self.element.text, 'child2child3child4child1')
self.c1.before(self.c2, 'text')
self.wait_until(
lambda: self.element.text == 'child3child4child2textchild1')
self.assertEqual(self.element.text, 'child3child4child2textchild1')
def test_after_before_text(self):
self.set_element(self.tag)
self.tag.append('a')
t = self.tag.firstChild
t.after('b')
self.wait_until(lambda: self.element.text == 'ab')
self.assertEqual(self.element.text, 'ab')
t.after('c', 'd')
self.wait_until(lambda: self.element.text == 'acdb')
self.assertEqual(self.element.text, 'acdb')
t.before('e')
self.wait_until(lambda: self.element.text == 'eacdb')
self.assertEqual(self.element.text, 'eacdb')
t.before('f', 'g')
self.wait_until(lambda: self.element.text == 'efgacdb')
self.assertEqual(self.element.text, 'efgacdb')
def test_shortcut_attr(self):
self.tag.textContent = 'TAG'
self.set_element(self.tag)
self.wait_until(lambda: self.element.is_displayed())
self.assertIsTrue(self.element.is_displayed())
self.tag.hidden = True
self.wait_until(lambda: not self.element.is_displayed())
self.assertIsFalse(self.element.is_displayed())
self.tag.hidden = False
self.wait_until(lambda: self.element.is_displayed())
self.assertIsTrue(self.element.is_displayed())
def test_style(self):
self.tag.textContent = 'Style'
self.set_element(self.tag)
self.wait_until(lambda: self.element.get_attribute('style') == '')
self.assertEqual(self.element.get_attribute('style'), '')
style = 'color: red;'
self.tag.style = style
self.wait_until(lambda: self.element.get_attribute('style') == style)
self.assertEqual(self.element.get_attribute('style'), style)
self.tag.style.color = 'black'
self.wait_until(
lambda: self.element.get_attribute('style') == 'color: black;')
self.assertEqual(self.element.get_attribute('style'), 'color: black;')
def test_classlist(self):
self.set_element(self.tag)
self.assertEqual(self.element.get_attribute('class'), '')
self.tag.classList.add('a')
self.wait_until(lambda: self.element.get_attribute('class') == 'a')
self.assertEqual(self.element.get_attribute('class'), 'a')
self.tag.classList.add('b', 'c', 'd')
self.wait_until(
lambda: self.element.get_attribute('class') == 'a b c d')
self.assertEqual(self.element.get_attribute('class'), 'a b c d')
self.tag.classList.remove('c')
self.wait_until(lambda: self.element.get_attribute('class') == 'a b d')
self.assertEqual(self.element.get_attribute('class'), 'a b d')
self.tag.classList.remove('a', 'd')
self.wait_until(lambda: self.element.get_attribute('class') == 'b')
self.assertEqual(self.element.get_attribute('class'), 'b')
self.tag.classList.toggle('b')
self.wait_until(lambda: self.element.get_attribute('class') == '')
self.assertEqual(self.element.get_attribute('class'), '')
self.tag.classList.toggle('b')
self.wait_until(lambda: self.element.get_attribute('class') == 'b')
self.assertEqual(self.element.get_attribute('class'), 'b')
def test_click(self):
mock = MagicMock(_is_coroutine=False)
self.tag.addEventListener('click', mock)
self.tag.click()
self.wait_until(lambda: mock.call_count == 1)
self.assertEqual(mock.call_count, 1)
@sync
@asyncio.coroutine
def test_get_rect(self):
rect = WebElement('div', style='width:200px;height:100px;')
self.tag.appendChild(rect)
yield from asyncio.sleep(self.wait_time)
data = yield from rect.getBoundingClientRect()
self.assertEqual(data['width'], 200)
self.assertEqual(data['height'], 100)
@sync
@asyncio.coroutine
def test_scroll(self):
rect = WebElement('div',
style='width:3000px;height:3000px;background:#eee;')
self.tag.appendChild(rect)
yield from asyncio.sleep(self.wait_time)
X = yield from rect.scrollX()
Y = yield from rect.scrollY()
self.assertEqual(X['x'], 0)
self.assertEqual(Y['y'], 0)
rect.scrollTo(200, 200)
yield from asyncio.sleep(self.wait_time)
X = yield from rect.scrollX()
Y = yield from rect.scrollY()
self.assertEqual(X['x'], 200)
self.assertEqual(Y['y'], 200)
def test_exec(self):
self.tag.exec('this.style = "color: red;"')
self.set_element(self.tag)
self.wait_until(
lambda: re.search(r'255,\s*0,\s* 0,\s*1\s*',
self.element.value_of_css_property('color')))
self.assertRegex(self.element.value_of_css_property('color'),
r'255,\s*0,\s* 0,\s*1\s*')
self.tag.exec('node.style = "color: blue;"')
self.wait_until(
lambda: re.search(r'0,\s*0,\s* 255,\s*1\s*',
self.element.value_of_css_property('color')))
self.assertRegex(self.element.value_of_css_property('color'),
r'0,\s*0,\s*255,\s*1\s*')
def test_exec_error(self):
with self.assertLogs('wdom.server', 'ERROR') as log:
self.tag.exec('a.b')
self.wait_until(lambda: len(log.output) > 0)
self.assertRegex(log.output[0], r'JS: ReferenceError')
