def test_can_find_warnings_on_page(self):
page = PageObject()
page.set_html(self.warning)
self.assert_true(page.has_warning('fnord warning'))
def test_can_remove_html_formatting(self):
page = PageObject()
warning_without_html = page.remove_html_and_whitespace(self.warning)
self.assert_equals("Warning: fnord warning Administration", warning_without_html)
def test_can_find_notice_on_page(self):
page = PageObject()
page.set_html(self.notice)
self.assert_true(page.has_notice('fnord notice one'))
self.assert_true(page.has_notice('fnord notice two'))