def test_parse(self):
parser = Parser()
parser.add_soup_getter("header", "h1")
parser.add_soup_getter("with_property", "div", {"class": "test"})
parser.add_soup_list_getter("lis", "li")
parser.add_soup_list_getter("lis_others", "li", {"class": "other"})
parser.add_getter_function("function", test_getter_function)
parser.add_soup_getter("a_href", "a", attribute="href")
parser.add_soup_getter("a_text", "a", attribute="text")
result = parser.parse(TEST_HTML_P1)
self.assertEqual(
result, {
"header": TEST_STRING,
"with_property": "some text",
"lis": ["a", "b", "c", "d", "e"],
"lis_others": ["d", "e"],
"function": TEST_STRING.upper(),
"a_href": TEST_DOMAINS[0],
"a_text": TEST_A_TEXT,
})
def test_set_internal_getters(self):
parser = Parser()
parser.parse("")
self.assertTrue(Parser.NEXT_PAGE in parser.supported_properties)