class TestAssertSelect(object):
def setup(self):
self.full_page = Page(filename="test/test.html")
self.fragment = Page(filename="test/fragment.html")
with file("test/fragment.html") as f:
self.string = Page(content=f.read())
def test_page_init_full_page(self):
assert self.full_page
def test_page_init_fragment(self):
assert self.fragment
def test_css_select(self):
nodes = self.fragment.css_select("div.secondary strong")
assert_equals(len(nodes), 2)
for node in nodes:
print type(node)
assert_equals(node.name, "strong")
def test_assert_select_string_defaults(self):
strongs = self.string.assert_select("strong")
assert_equals(len(strongs), 2)
def test_assert_select_defaults(self):
strongs = self.fragment.assert_select("strong")
assert_equals(len(strongs), 2)
def test_as_count(self):
self.fragment.assert_select("strong", 2)
try:
self.fragment.assert_select("strong", 3)
except AssertionError, e:
assert_in("3", e.message)
assert_in("found 2", e.message)
def setup(self):
self.full_page = Page(filename="test/test.html")
self.fragment = Page(filename="test/fragment.html")
with file("test/fragment.html") as f:
self.string = Page(content=f.read())
