import pytest
import webview
from .util import run_test
html = """
Heading
Content 1
Content 2
"""
@pytest.fixture
def window():
return webview.create_window('Get elements test', html=html)
def test_single(window):
run_test(webview, window, single_test)
def test_multiple(window):
run_test(webview, window, multiple_test)
def test_none(window):
run_test(webview, window, none_test)
def single_test(window):
try:
elements = window.dom.get_elements('#heading')
assert len(elements) == 1
assert elements[0].node['innerHTML'] == 'Heading'
except NotImplementedError:
pass
def multiple_test(window):
try:
elements = window.dom.get_elements('.content')
assert len(elements) == 2
assert elements[0].node['innerHTML'] == 'Content 1'
assert elements[1].node['innerHTML'] == 'Content 2'
except NotImplementedError:
pass
def none_test(window):
try:
elements = window.dom.get_elements('.adgdfg')
assert len(elements) == 0
except NotImplementedError:
pass