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())